Upload myinfer_latest.py

myinfer_latest.py

@@ -27,6 +27,8 @@ from threading import Semaphore
 from threading import Lock
 from multiprocessing import Process, SimpleQueue, set_start_method,get_context
 from queue import Empty
+from pydub import AudioSegment
+import io
 
 
 
@@ -170,6 +172,19 @@ def api_convert_voice():
         file = request.files['file']
         if file.filename == '':
             return jsonify({"error": "No selected file"}), 400
+        
+        if file.content_length > 6 * 1024 * 1024:
+            return jsonify({"error": "File size exceeds 6 MB"}), 400
+            
+        audio = AudioSegment.from_file(io.BytesIO(file.read()), format="wav")  # Adjust format as necessary
+        file.seek(0)  # Reset file pointer after reading
+    
+    # Calculate audio length in minutes
+        audio_length_minutes = len(audio) / 60000.0  # pydub returns length in milliseconds
+    
+        if audio_length_minutes > 3:
+            return jsonify({"error": "Audio length exceeds 3 minutes"}), 400    
+            
         #created_files = []
         # Save the file to a temporary path
         unique_id = str(uuid.uuid4())