Upload myinfer_latest.py

myinfer_latest.py

@@ -21,6 +21,9 @@ import os
 import werkzeug
 from pydub import AudioSegment
 import uuid
+from threading import Semaphore
+
+
 
 
 app = Flask(__name__)
@@ -31,6 +34,14 @@ shutil.rmtree(tmp, ignore_errors=True)
 os.makedirs(tmp, exist_ok=True)
 os.environ["TEMP"] = tmp
 split_model="htdemucs"
+
+# Define the maximum number of concurrent requests
+MAX_CONCURRENT_REQUESTS = 1  # Adjust this number as needed
+
+# Initialize the semaphore with the maximum number of concurrent requests
+request_semaphore = Semaphore(MAX_CONCURRENT_REQUESTS)
+
+
 from lib.infer_pack.models import (
     SynthesizerTrnMs256NSFsid,
     SynthesizerTrnMs256NSFsid_nono,
@@ -116,51 +127,57 @@ def cleanup_files(file_paths):
 processed_audio_storage = {}
 @app.route('/convert_voice', methods=['POST'])
 def api_convert_voice():
-    if session.get('submitted'):
-        return jsonify({"error": "Form already submitted"}), 400
-
-    # Process the form here...
-    # Set the flag indicating the form has been submitted
-    session['submitted'] = True
-    print(request.form)
-    print(request.files)
-    spk_id = request.form['spk_id']+'.pth'
-    voice_transform = request.form['voice_transform']
-
-    # The file part
-    if 'file' not in request.files:
-        return jsonify({"error": "No file part"}), 400
-    file = request.files['file']
-    if file.filename == '':
-        return jsonify({"error": "No selected file"}), 400
-    created_files = []
-    # Save the file to a temporary path
-    unique_id = str(uuid.uuid4())
-    print(unique_id)
-	
-    filename = werkzeug.utils.secure_filename(file.filename)
-    input_audio_path = os.path.join(tmp, f"{spk_id}_input_audio_{unique_id}.{filename.split('.')[-1]}")
-    file.save(input_audio_path)
-    
-    created_files.append(input_audio_path)
-    
-    #split audio
-    cut_vocal_and_inst(input_audio_path,spk_id)
-    print("audio splitting performed")
-    vocal_path = f"output/{split_model}/{spk_id}_input_audio_{unique_id}/vocals.wav"
-    inst = f"output/{split_model}/{spk_id}_input_audio_{unique_id}/no_vocals.wav"
-    
-    output_path = convert_voice(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
-    session['processed_audio_id'] = unique_id 
-    
-    print(output_path1)
+    acquired = request_semaphore.acquire(blocking=False)
+    if not acquired:
+        return jsonify({"error": "Too many requests, please try again later"}), 429
+    try:
     
-    created_files.extend([vocal_path, inst, output_path])
-    return jsonify({"message": "File processed successfully", "audio_id": unique_id}), 200
-
+        if session.get('submitted'):
+            return jsonify({"error": "Form already submitted"}), 400
+
+        # Process the form here...
+        # Set the flag indicating the form has been submitted
+        session['submitted'] = True
+        print(request.form)
+        print(request.files)
+        spk_id = request.form['spk_id']+'.pth'
+        voice_transform = request.form['voice_transform']
+
+        # The file part
+        if 'file' not in request.files:
+            return jsonify({"error": "No file part"}), 400
+        file = request.files['file']
+        if file.filename == '':
+            return jsonify({"error": "No selected file"}), 400
+        created_files = []
+        # Save the file to a temporary path
+        unique_id = str(uuid.uuid4())
+        print(unique_id)
+        
+        filename = werkzeug.utils.secure_filename(file.filename)
+        input_audio_path = os.path.join(tmp, f"{spk_id}_input_audio_{unique_id}.{filename.split('.')[-1]}")
+        file.save(input_audio_path)
+        
+        created_files.append(input_audio_path)
+        
+        #split audio
+        cut_vocal_and_inst(input_audio_path,spk_id)
+        print("audio splitting performed")
+        vocal_path = f"output/{split_model}/{spk_id}_input_audio_{unique_id}/vocals.wav"
+        inst = f"output/{split_model}/{spk_id}_input_audio_{unique_id}/no_vocals.wav"
+        
+        output_path = convert_voice(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
+        session['processed_audio_id'] = unique_id 
+        
+        print(output_path1)
+        
+        created_files.extend([vocal_path, inst, output_path])
+        return jsonify({"message": "File processed successfully", "audio_id": unique_id}), 200
+    finally:
+        semaphore.release()
     #if os.path.exists(output_path1):
         
     #    return send_file(output_path1, as_attachment=True)