Upload myinfer_latest.py

myinfer_latest.py

@@ -104,7 +104,14 @@ def clean():
         gr.Dropdown.update(value=""),
         gr.Slider.update(visible=False)
     )
-
+# Function to delete files
+def cleanup_files(file_paths):
+    for path in file_paths:
+        try:
+            os.remove(path)
+            print(f"Deleted {path}")
+        except Exception as e:
+            print(f"Error deleting {path}: {e}")
 
 
 @app.route('/convert_voice', methods=['POST'])
@@ -118,7 +125,7 @@ def api_convert_voice():
     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 = uuid.uuid4()
 	
@@ -126,6 +133,8 @@ def api_convert_voice():
     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")
@@ -136,8 +145,10 @@ def api_convert_voice():
     output_path1= combine_vocal_and_inst(output_path,inst)
     print(output_path1)
     
+    created_files.extend([vocal_path, inst_path, output_path])
     
     if os.path.exists(output_path1):
+        
         return send_file(output_path1, as_attachment=True)
     else:
         return jsonify({"error": "File not found."}), 404