Update app.py

app.py

@@ -4,8 +4,7 @@ import time
 import json
 from fastapi import FastAPI, Request
 from fastapi.responses import StreamingResponse
-# ייבוא מהחבילה שהותקנה מה-Git
-from phonikud import Phonikud, phonemize 
+from phonikud import Phonikud, phonemize # ייבוא רשמי מהספרייה
 from google import genai
 from google.genai import types
 
@@ -14,7 +13,7 @@ logger = logging.getLogger(__name__)
 
 app = FastAPI()
 
-# טעינת המודל (שימוש בנתיב המקומי שלך)
+# טעינת המודל (הקובץ חייב להיות בתיקייה הראשית של ה-Space)
 MODEL_PATH = "./phonikud-1.0.int8.onnx"
 pk = Phonikud(MODEL_PATH) if os.path.exists(MODEL_PATH) else None
 
@@ -26,7 +25,7 @@ async def chat_proxy(request: Request):
         body = await request.json()
         messages = body.get("messages", [])
         
-        # בניית הקונטקסט ל-Gemini 2.5 Flash
+        # בניית ההקשר לדניאל
         system_instruction = ""
         contents = []
         for msg in messages:
@@ -36,7 +35,7 @@ async def chat_proxy(request: Request):
                 role = "user" if msg["role"] == "user" else "model"
                 contents.append(types.Content(role=role, parts=[types.Part(text=msg["content"])]))
 
-        # 1. Gemini מייצר את התשובה
+        # 1. קבלת תשובה מ-Gemini
         response = client.models.generate_content(
             model='gemini-2.5-flash',
             contents=contents,
@@ -46,23 +45,21 @@ async def chat_proxy(request: Request):
         original_text = response.text or ""
         logger.info(f"Gemini: {original_text}")
 
-        # 2. הפיכה לפונמות (IPA) בפורמט קרטסיה
+        # 2. הפיכה לפונמות IPA (השיטה מהמאמר)
         final_output = original_text
         if pk and original_text:
             try:
-                # א) הוספת ניקוד
+                # הוספת ניקוד
                 vocalized = pk.add_diacritics(original_text)
-                # ב) המרה לפונמות IPA
-                phonemes = phonemize(vocalized)
-                # ג) עטיפה עבור קרטסיה (הסרת קווים מיותרים)
-                clean_ipa = phonemes.replace('|', '')
-                final_output = f"<<{clean_ipa}>>"
-                
-                logger.info(f"Cartesia Phonemes: {final_output}")
+                # המרה לפונמות IPA
+                ipa = phonemize(vocalized)
+                # עטיפה לקרטסיה (ניקוי תווים מבניים במידת הצורך)
+                final_output = f"<<{ipa.replace('|', '')}>>"
+                logger.info(f"IPA for Cartesia: {final_output}")
             except Exception as e:
                 logger.error(f"Phonemization failed: {e}")
 
-        # 3. הזרמה לוואפי (SSE)
+        # 3. הזרמה בפורמט שוואפי מבינה
         def generate_vapi_stream():
             chat_id = f"chatcmpl-{int(time.time())}"
             chunk = {
@@ -81,4 +78,8 @@ async def chat_proxy(request: Request):
 
     except Exception as e:
         logger.error(f"Error: {str(e)}")
-        return {"error": str(e)}, 500
+        return {"error": str(e)}, 500
+
+if __name__ == "__main__":
+    import uvicorn
+    uvicorn.run(app, host="0.0.0.0", port=7860)