Update app.py

app.py

@@ -1,23 +1,20 @@
 import gradio as gr
 import requests
-import json
 import time
 
 def monster_chat(message, history):
-    # שימוש במודל Llama-3-8B - הכי יציב ומהיר ב-API
+    # מודל Llama-3 היציב
     API_URL = "https://api-inference.huggingface.co/models/meta-llama/Meta-Llama-3-8B-Instruct"
     
-    # בניית הפרומפט בצורה שהמודל יבין שהוא בצ'אט
     payload = {
-        "inputs": f"<|begin_of_text|><|start_header_id|>system<|end_header_id|>\n\nYou are a funny and charismatic monster AI. Answer in Hebrew only.<|eot_id|><|start_header_id|>user<|end_header_id|>\n\n{message}<|eot_id|><|start_header_id|>assistant<|end_header_id|>\n\n",
-        "parameters": {"max_new_tokens": 200, "temperature": 0.7, "return_full_text": False}
+        "inputs": f"<|begin_of_text|><|start_header_id|>system<|end_header_id|>\n\nYou are a funny AI. Answer in Hebrew.<|eot_id|><|start_header_id|>user<|end_header_id|>\n\n{message}<|eot_id|><|start_header_id|>assistant<|end_header_id|>\n\n",
+        "parameters": {"max_new_tokens": 150, "temperature": 0.7}
     }
     
-    # ניסיון ראשון לשלוח הודעה
     try:
         response = requests.post(API_URL, json=payload, timeout=15)
         
-        # אם המודל בטעינה, נחכה 5 שניות וננסה שוב אוטומטית
+        # אם המודל בטעינה (503), נחכה קצת
         if response.status_code == 503:
             time.sleep(5)
             response = requests.post(API_URL, json=payload, timeout=15)
@@ -25,17 +22,15 @@ def monster_chat(message, history):
         result = response.json()
         
         if isinstance(result, list) and len(result) > 0:
-            return result[0].get('generated_text', "וואלה, אין לי מילים!")
-        elif isinstance(result, dict) and "generated_text" in result:
-            return result["generated_text"]
-        else:
-            return "המפלצת מנסה להבין מה אמרת... תשלח שוב!"
-    except Exception as e:
-        return "יש תקלה בחיבור למפלצת, נסה שוב בעוד כמה שניות."
+            return result[0].get('generated_text', "המפלצת איבדה את הקול!")
+        return "המפלצת חושבת... נסה לשלוח שוב."
+    except:
+        return "תקלה בחיבור למפלצת."
 
-# עיצוב הממשק
-with gr.Blocks(theme=gr.themes.Default(primary_hue="purple", secondary_hue="indigo")) as demo:
-    gr.HTML("<div style='text-align:center'><h1>👾 THE REAL MONSTER AI</h1><p>עכשיו עם מנוע Llama-3 עוצמתי</p></div>")
-    gr.ChatInterface(fn=monster_chat, type="messages")
+# יצירת הממשק בצורה נקייה בלי פרמטרים שבורים
+with gr.Blocks() as demo:
+    gr.Markdown("# 👾 MONSTER CHAT")
+    gr.ChatInterface(fn=monster_chat)
 
+# העברנו את ה-theme לפה, בדיוק כמו שהשגיאה ביקשה
 demo.launch()