Update app.py

app.py

@@ -1,61 +1,110 @@
-from flask import Flask, request, jsonify, send_file
 import os
 import uuid
+import shutil
+import json
+from fastapi import FastAPI, HTTPException, Query
+from fastapi.responses import JSONResponse, FileResponse
+from pydantic import BaseModel
 import requests
-from process_interview import process_interview
-from urllib.parse import urlparse
+from process_interview import process_interview  # افترض ان ده ملف عندك فيه الدالة
+from fastapi.staticfiles import StaticFiles
 
-app = Flask(__name__)
+app = FastAPI()
 
-OUTPUT_DIR = "./processed_audio"
+app.mount("/static", StaticFiles(directory="static"), name="static")
+
+
+TEMP_DIR = "./temp_files"
+OUTPUT_DIR = "./static/outputs"
+
+os.makedirs(TEMP_DIR, exist_ok=True)
 os.makedirs(OUTPUT_DIR, exist_ok=True)
 
-@app.route("/process-audio", methods=["POST"])
-def process_audio():
+VALID_EXTENSIONS = ('.wav', '.mp3', '.m4a', '.flac')
+MAX_FILE_SIZE_MB = 300
+
+class ProcessResponse(BaseModel):
+    summary: str
+    json_url: str
+    pdf_url: str
+
+@app.get("/")
+async def root():
+    return {"message": "EvalBot API is running"}
+
+@app.post("/process-audio", response_model=ProcessResponse)
+async def process_audio(file_url: str = Query(..., description="URL of the audio file"),
+                        user_id: str = Query(..., description="User ID")):
     try:
-        data = request.json
-        user_id = data.get("user_id")
-        audio_url = data.get("audio_url")
+        # تحميل الملف مؤقتًا
+        file_ext = os.path.splitext(file_url)[1].lower()
+        if file_ext not in VALID_EXTENSIONS:
+            raise HTTPException(status_code=400, detail=f"Invalid file extension: {file_ext}")
 
-        if not user_id or not audio_url:
-            return jsonify({"error": "Missing user_id or audio_url"}), 400
+        local_filename = f"{user_id}_{uuid.uuid4().hex}{file_ext}"
+        local_path = os.path.join(TEMP_DIR, local_filename)
 
-        file_ext = os.path.splitext(urlparse(audio_url).path)[1]
-        filename = f"{user_id}_{uuid.uuid4()}{file_ext}"
-        local_file_path = os.path.join(OUTPUT_DIR, filename)
+        # تحميل الملف
+        resp = requests.get(file_url, stream=True, timeout=30)
+        if resp.status_code != 200:
+            raise HTTPException(status_code=400, detail="Could not download the file from URL")
 
-        response = requests.get(audio_url, stream=True)
-        if response.status_code != 200:
-            return jsonify({"error": "Failed to download audio file"}), 400
+        with open(local_path, "wb") as f:
+            for chunk in resp.iter_content(chunk_size=8192):
+                if chunk:
+                    f.write(chunk)
 
-        with open(local_file_path, "wb") as f:
-            for chunk in response.iter_content(chunk_size=8192):
-                f.write(chunk)
+        # تحقق من الحجم
+        file_size_mb = os.path.getsize(local_path) / (1024 * 1024)
+        if file_size_mb > MAX_FILE_SIZE_MB:
+            os.remove(local_path)
+            raise HTTPException(status_code=400, detail=f"File too large: {file_size_mb:.2f} MB")
 
-        result = process_interview(local_file_path)
+        # معالجة الملف (process_interview ترجع dict فيها json_path و pdf_path)
+        result = process_interview(local_path)
+        if not result:
+            os.remove(local_path)
+            raise HTTPException(status_code=500, detail="Processing failed")
 
-        if not result or 'pdf_path' not in result or 'json_path' not in result:
-            return jsonify({"error": "Processing failed"}), 500
+        # نسخ ملفات JSON و PDF لمجلد ثابت للاحتفاظ بهم
+        json_src = result['json_path']
+        pdf_src = result['pdf_path']
 
-        server_url = request.host_url.rstrip('/')
-        pdf_url = f"{server_url}/download?file={os.path.basename(result['pdf_path'])}"
-        json_url = f"{server_url}/download?file={os.path.basename(result['json_path'])}"
+        # أسماء جديدة مع UUID
+        json_dest_name = f"{user_id}_{uuid.uuid4().hex}.json"
+        pdf_dest_name = f"{user_id}_{uuid.uuid4().hex}.pdf"
 
-        return jsonify({
-            "message": "Processing complete",
-            "pdf_url": pdf_url,
-            "json_url": json_url
-        })
+        json_dest = os.path.join(OUTPUT_DIR, json_dest_name)
+        pdf_dest = os.path.join(OUTPUT_DIR, pdf_dest_name)
 
-    except Exception as e:
-        return jsonify({"error": str(e)}), 500
+        shutil.copyfile(json_src, json_dest)
+        shutil.copyfile(pdf_src, pdf_dest)
 
-@app.route("/download", methods=["GET"])
-def download_file():
-    filename = request.args.get("file")
-    file_path = os.path.join(OUTPUT_DIR, filename)
-    if not os.path.exists(file_path):
-        return jsonify({"error": "File not found"}), 404
-    return send_file(file_path, as_attachment=True)
+        # قراءة بيانات JSON للملخص
+        with open(json_src, "r") as jf:
+            analysis_data = json.load(jf)
 
-# لا تحتفظي بـ app.run لأن gunicorn هيشغله
+        voice = analysis_data.get('voice_analysis', {}).get('interpretation', {})
+        speakers = analysis_data.get('speakers', [])
+        total_duration = analysis_data.get('text_analysis', {}).get('total_duration', 0.0)
+
+        summary = (
+            f"User ID: {user_id}\n"
+            f"Speakers: {', '.join(speakers)}\n"
+            f"Duration: {total_duration:.2f} sec\n"
+            f"Confidence: {voice.get('confidence_level', 'N/A')}\n"
+            f"Anxiety: {voice.get('anxiety_level', 'N/A')}"
+        )
+
+        # بناء URLs للملفات
+        base_url = f"http://localhost:7860/static/outputs"
+        json_url = f"{base_url}/{json_dest_name}"
+        pdf_url = f"{base_url}/{pdf_dest_name}"
+
+        # تنظيف الملف الصوتي المؤقت
+        os.remove(local_path)
+
+        return ProcessResponse(summary=summary, json_url=json_url, pdf_url=pdf_url)
+
+    except Exception as e:
+        raise HTTPException(status_code=500, detail=str(e))