Update app.py

app.py

@@ -14,7 +14,8 @@ app.add_middleware(
 )
 
 HF_TOKEN = os.getenv("HF_TOKEN")
-MODEL_ID = "meta-llama/Llama-3.1-8B-Instruct"
+# පාවිච්චි කරන Model එක
+MODEL_ID = "Qwen/Qwen2.5-72B-Instruct"
 
 @app.post("/chat")
 async def chat(request: Request):
@@ -27,22 +28,30 @@ async def chat(request: Request):
             "Content-Type": "application/json"
         }
         
-        # 🛡️ මෙහි ඇති URL එක නව 'router' endpoint එකට මාරු කරන ලදී
+        # 🛡️ පිරිපහදු කළ Payload එක
+        payload = {
+            "model": MODEL_ID,
+            "messages": messages,
+            "max_tokens": 1024,
+            "temperature": 0.7
+        }
+        
+        # Hugging Face Router වෙත දත්ත යැවීම
         response = requests.post(
-            f"https://router.huggingface.co/openai/v1/chat/completions",
+            "https://router.huggingface.co/openai/v1/chat/completions",
             headers=headers,
-            json={
-                "model": MODEL_ID,
-                "messages": messages,
-                "max_tokens": 1024,
-                "temperature": 0.7,
-                "stream": False
-            }
+            json=payload,
+            timeout=30 # කාලය ඉක්මවා යාම වැළැක්වීමට
         )
         
+        # ⚠️ වැදගත්: ප්‍රතිචාරය JSON ද කියා පරීක්ෂා කිරීම
+        if response.status_code != 200:
+            return {"error": f"HF API Error {response.status_code}", "details": response.text}
+            
         return response.json()
+        
     except Exception as e:
-        return {"error": str(e)}
+        return {"error": "Internal System Failure", "details": str(e)}
 
 @app.get("/")
 def home():