Upload myinfer_latest.py

myinfer_latest.py

@@ -22,6 +22,9 @@ import werkzeug
 from pydub import AudioSegment
 import uuid
 from threading import Semaphore
+from threading import Lock
+
+
 
 
 
@@ -34,7 +37,7 @@ shutil.rmtree(tmp, ignore_errors=True)
 os.makedirs(tmp, exist_ok=True)
 os.environ["TEMP"] = tmp
 split_model="htdemucs"
-
+convert_voice_lock = Lock()
 # Define the maximum number of concurrent requests
 MAX_CONCURRENT_REQUESTS = 2  # Adjust this number as needed
 
@@ -166,7 +169,7 @@ def api_convert_voice():
         vocal_path = f"output/{spk_id}_{unique_id}/{split_model}/{spk_id}_input_audio_{unique_id}/vocals.wav"
         inst = f"output/{spk_id}_{unique_id}/{split_model}/{spk_id}_input_audio_{unique_id}/no_vocals.wav"
         print("*****before making call to convert ", unique_id)
-        output_path = convert_voice(spk_id, vocal_path, voice_transform,unique_id)
+        output_path = convert_voice_thread_safe(spk_id, vocal_path, voice_transform,unique_id)
         output_path1= combine_vocal_and_inst(output_path,inst,unique_id)
         
         processed_audio_storage[unique_id] = output_path1
@@ -184,6 +187,11 @@ def api_convert_voice():
     #else:
     #    return jsonify({"error": "File not found."}), 404
 
+def convert_voice_thread_safe(spk_id, vocal_path, voice_transform, unique_id):
+    with convert_voice_lock:
+        return convert_voice(spk_id, vocal_path, voice_transform, unique_id)
+
+
 @app.route('/')
 def upload_form():
     return render_template('ui.html')