Update app.py

app.py

@@ -1,51 +1,32 @@
 from fastapi import FastAPI, Request
-import requests
-import json
 from fastapi.middleware.cors import CORSMiddleware
-import uvicorn
+import requests
 
 app = FastAPI()
-app.add_middleware(CORSMiddleware, allow_origins=["*"], allow_methods=["*"], allow_headers=["*"])
 
-# 🔱 GLM-5.1 CONFIGURATION
+# 🔱 සියලුම සන්නිවේදන බාධක ඉවත් කිරීම
+app.add_middleware(
+    CORSMiddleware,
+    allow_origins=["*"],
+    allow_methods=["*"],
+    allow_headers=["*"],
+)
+
 API_URL = "https://api.us-west-2.modal.direct/v1/chat/completions"
 API_KEY = "modalresearch_0yKf_z9azDFpOXr-W1N1WsmBUgXf6Ux-yvpVsFQWzks"
-MODEL_ID = "zai-org/GLM-5.1-FP8"
 
 @app.post("/generate")
 async def generate(request: Request):
     try:
         data = await request.json()
-        user_prompt = data.get("prompt")
-
-        # 🔱 Payload for GLM-5.1
+        headers = {"Authorization": f"Bearer {API_KEY}", "Content-Type": "application/json"}
         payload = {
-            "model": MODEL_ID,
-            "messages": [
-                {"role": "system", "content": "You are Elephant AI Pro, a master-level coding assistant built by MINZO-PRIME. Provide highly detailed and functional code."},
-                {"role": "user", "content": user_prompt}
-            ],
-            "max_tokens": 1024,
-            "temperature": 0.6
-        }
-
-        headers = {
-            "Authorization": f"Bearer {API_KEY}",
-            "Content-Type": "application/json"
+            "model": "zai-org/GLM-5.1-FP8",
+            "messages": [{"role": "user", "content": data.get("prompt")}],
+            "max_tokens": 1024
         }
-
-        # 🔱 Fast-Inference Request
-        response = requests.post(API_URL, headers=headers, json=payload, timeout=60)
-        result = response.json()
-
-        if "choices" in result:
-            ai_response = result["choices"][0]["message"]["content"]
-            return {"response": ai_response}
-        else:
-            return {"response": "SYSTEM ERROR: API Node non-responsive. Check Token status."}
-
+        # සර්වර් එක හරහා යන නිසා බ්‍රවුසරයට බාධා කළ නොහැක
+        response = requests.post(API_URL, headers=headers, json=payload)
+        return response.json()
     except Exception as e:
-        return {"error": str(e)}
-
-if __name__ == "__main__":
-    uvicorn.run(app, host="0.0.0.0", port=7860)
+        return {"error": str(e)}