Upload 2 files

myinfer_latest.py

@@ -14,7 +14,7 @@ from scipy.io import wavfile
 from datetime import datetime
 from urllib.parse import urlparse
 from mega import Mega
-from flask import Flask, request, jsonify, send_file
+from flask import Flask, request, jsonify, send_file,session
 import base64
 import tempfile
 import os
@@ -113,10 +113,10 @@ def cleanup_files(file_paths):
         except Exception as e:
             print(f"Error deleting {path}: {e}")
 
-
+processed_audio_storage = {}
 @app.route('/convert_voice', methods=['POST'])
 def api_convert_voice():
-    spk_id = request.form['spk_id']
+    spk_id = request.form['spk_id']+'.pth'
     voice_transform = request.form['voice_transform']
 
     # The file part
@@ -143,16 +143,32 @@ def api_convert_voice():
     
     output_path = convert_voice(spk_id, vocal_path, voice_transform)
     output_path1= combine_vocal_and_inst(output_path,inst)
+    
+    processed_audio_storage[unique_id] = output_path1
+    session['processed_audio_id'] = unique_id 
+    
     print(output_path1)
     
     created_files.extend([vocal_path, inst, output_path])
-    
-    if os.path.exists(output_path1):
+    return jsonify({"message": "File processed successfully", "audio_id": unique_id}), 200
+
+    #if os.path.exists(output_path1):
         
-        return send_file(output_path1, as_attachment=True)
-    else:
-        return jsonify({"error": "File not found."}), 404
-    
+    #    return send_file(output_path1, as_attachment=True)
+    #else:
+    #    return jsonify({"error": "File not found."}), 404
+
+@app.route('/')
+def upload_form():
+    return render_template('ui.html')
+
+@app.route('/get_processed_audio/<audio_id>')
+def get_processed_audio(audio_id):
+    # Retrieve the path from temporary storage or session
+    if audio_id in processed_audio_storage:
+        file_path = processed_audio_storage[audio_id]
+        return send_file(file_path, as_attachment=True)
+    return jsonify({"error": "File not found."}), 404
 
 
 def convert_voice(spk_id, input_audio_path, voice_transform):

ui.html

@@ -0,0 +1,35 @@
+<!-- Include jQuery for simplicity -->
+<script src="https://ajax.googleapis.com/ajax/libs/jquery/3.5.1/jquery.min.js"></script>
+
+<script>
+$(document).ready(function() {
+    $('#uploadForm').submit(function(e) {
+        e.preventDefault();
+        var formData = new FormData(this);
+
+        $.ajax({
+            url: '/convert_voice',
+            type: 'POST',
+            data: formData,
+            success: function(data) {
+                if (data.audio_id) {
+                    // Update the source of the processed audio element
+                    $('#processedAudio source').attr('src', '/get_processed_audio/' + data.audio_id);
+                    $('#processedAudio')[0].load();
+                    $('#processedAudio')[0].play();
+                }
+            },
+            cache: false,
+            contentType: false,
+            processData: false
+        });
+    });
+});
+</script>
+
+<!-- Processed Audio Playback -->
+<h3>Processed Audio:</h3>
+<audio id="processedAudio" controls>
+    <source src="" type="audio/wav">
+    Your browser does not support the audio element.
+</audio>