Upload myinfer_latest.py

myinfer_latest.py

@@ -30,7 +30,7 @@ from queue import Empty
 from pydub import AudioSegment
 from flask_dance.contrib.google import make_google_blueprint, google
 import io
-from space import list_models
+from space import ensure_model_in_weights_dir,upload_to_do
 import boto3
 
 
@@ -275,8 +275,13 @@ def api_convert_voice():
         
         if file.content_length > 10 * 1024 * 1024:
             return jsonify({"error": "File size exceeds 6 MB"}), 400
+        
         print("after file access")
-
+        print("check if model is there in weights dir or not")
+        filename_without_extension = os.path.splitext(file.filename)[0]
+        unique_id = filename_without_extension
+        ensure_model_in_weights_dir(spk_id)
+        print("checking done for the model")
         content_type_format_map = {
             'audio/mpeg': 'mp3',
             'audio/wav': 'wav',
@@ -302,7 +307,7 @@ def api_convert_voice():
             
         #created_files = []
         # Save the file to a temporary path
-        unique_id = str(uuid.uuid4())
+        #unique_id = str(uuid.uuid4())
         print(unique_id)
         
         filename = werkzeug.utils.secure_filename(file.filename)
@@ -347,6 +352,8 @@ def api_convert_voice():
         
         #created_files.extend([vocal_path, inst, output_path])
         task_status_tracker[unique_id]["status"] = "Completed"
+        upload_to_do()
+        
         return jsonify({"message": "File processed successfully", "audio_id": unique_id}), 200
     finally:
         request_semaphore.release()