Upload myinfer_latest.py

myinfer_latest.py

@@ -49,6 +49,8 @@ MAX_CONCURRENT_REQUESTS = 2  # Adjust this number as needed
 # Initialize the semaphore with the maximum number of concurrent requests
 request_semaphore = Semaphore(MAX_CONCURRENT_REQUESTS)
 
+task_status_tracker = {}
+
 #set_start_method('spawn', force=True)
 from lib.infer_pack.models import (
     SynthesizerTrnMs256NSFsid,
@@ -147,12 +149,20 @@ def cleanup_files(file_paths):
         except Exception as e:
             print(f"Error deleting {path}: {e}")
 
+@app.route('/status/<audio_id>', methods=['GET'])
+def get_status(audio_id):
+    # Retrieve the task status using the unique ID
+    status_info = task_status_tracker.get(audio_id, {"status": "Unknown ID", "percentage": 0})
+    return jsonify({"audio_id": audio_id, "status": status_info["status"], "percentage": status_info["percentage"]})
+
 processed_audio_storage = {}
 @app.route('/convert_voice', methods=['POST'])
 def api_convert_voice():
     acquired = request_semaphore.acquire(blocking=False)
+    
     if not acquired:
         return jsonify({"error": "Too many requests, please try again later"}), 429
+    task_status_tracker[unique_id] = {"status": "Starting", "percentage": 0}
     try:
     
         #if session.get('submitted'):
@@ -212,11 +222,14 @@ def api_convert_voice():
         #created_files.append(input_audio_path)
         
         #split audio
+        task_status_tracker[unique_id] = {"status": "Processing: Step 1", "percentage": 30}
+
         cut_vocal_and_inst(input_audio_path,spk_id,unique_id)
         print("audio splitting performed")
         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)
+        task_status_tracker[unique_id] = "Processing: Step 2"
         #output_queue = SimpleQueue()
         ctx = get_context('spawn')
         output_queue = ctx.Queue()
@@ -229,6 +242,7 @@ def api_convert_voice():
         print("*******waiting for process to complete ")
         
         output_path = output_queue.get()
+        task_status_tracker[unique_id] = {"status": "Processing: Step 2", "percentage": 80}
         #if isinstance(output_path, Exception):
          #   print("Exception in worker:", output_path)
         #else:
@@ -238,10 +252,11 @@ def api_convert_voice():
         
         processed_audio_storage[unique_id] = output_path1
         session['processed_audio_id'] = unique_id 
-        
+        task_status_tracker[unique_id] = {"status": "Finalizing", "percentage": 100}
         print(output_path1)
         
         #created_files.extend([vocal_path, inst, output_path])
+        task_status_tracker[unique_id]["status"] = "Completed"
         return jsonify({"message": "File processed successfully", "audio_id": unique_id}), 200
     finally:
         request_semaphore.release()