change

ai_med_extract/api/routes.py

@@ -160,7 +160,7 @@ def register_routes(app, agents):
     PHIScrubberAgent = agents["phi_scrubber"]
     SummarizerAgent = agents["summarizer"]
     MedicalDataExtractorAgent = agents["medical_data_extractor"]
-    whisper_model = agents["whisper_model"]()
+    whisper_model = agents["whisper_model"]  # No longer needs to be called as a function
 
     @app.route("/upload", methods=["POST"])
     def upload_file():
@@ -362,19 +362,28 @@ def register_routes(app, agents):
 
     @app.route("/transcribe", methods=["POST"])
     def transcribe_audio():
-        if "file" not in request.files:
-            return jsonify({"error": "No file part"}), 400
-        file = request.files["file"]
-        if file.filename == "":
-            return jsonify({"error": "No selected file"}), 400
-        temp_path = os.path.join("/tmp", file.filename)
-        file.save(temp_path)
+        temp_path = None
         try:
-            whisper_model = agents["whisper_model"]()
+            if "file" not in request.files:
+                return jsonify({"error": "No file part"}), 400
+            file = request.files["file"]
+            if file.filename == "":
+                return jsonify({"error": "No selected file"}), 400
+
+            # Use secure filename
+            from werkzeug.utils import secure_filename
+            import uuid
+            temp_filename = f"{uuid.uuid4()}_{secure_filename(file.filename)}"
+            temp_path = os.path.join("/tmp", temp_filename)
+            
+            file.save(temp_path)
             result = whisper_model.transcribe(temp_path)
             os.remove(temp_path)
             return jsonify({"transcription": result["text"]}), 200
         except Exception as e:
+            if temp_path and os.path.exists(temp_path):
+                os.remove(temp_path)
+            logging.error(f"Transcription failed: {str(e)}", exc_info=True)
             return jsonify({"error": str(e)}), 500
 
     @app.route("/extract_medical_data", methods=["POST"])

ai_med_extract/utils/file_utils.py

@@ -6,8 +6,9 @@ from werkzeug.utils import secure_filename
 from flask import current_app
 
 ALLOWED_EXTENSIONS = {"pdf", "jpg", "jpeg", "png", "svg", "docx", "doc", "xlsx", "xls", "wav", "mp3", "m4a", "ogg"}
-MAX_SIZE_PDF_DOCS = 1 * 1024 * 1024 * 1024  # 1GB
-MAX_SIZE_IMAGES = 500 * 1024 * 1024  # 500MB
+MAX_SIZE_PDF_DOCS = 1 * 1024 * 1024 * 1024    # 1GB
+MAX_SIZE_IMAGES = 500 * 1024 * 1024           # 500MB
+MAX_SIZE_AUDIO = 100 * 1024 * 1024            # 100MB
 
 
 def allowed_file(filename):
@@ -31,6 +32,8 @@ def check_file_size(file):
             return False, f"File {file.filename} exceeds 1GB size limit"
         elif extension in {"jpg", "jpeg", "png"} and size > MAX_SIZE_IMAGES:
             return False, f"Image {file.filename} exceeds 500MB size limit"
+        elif extension in {"wav", "mp3", "m4a", "ogg"} and size > MAX_SIZE_AUDIO:
+            return False, f"Audio file {file.filename} exceeds 100MB size limit"
         return True, None
     except Exception as e:
         logging.error(f"Error checking file size: {e}", exc_info=True)