Update app.py

app.py

@@ -1,36 +1,37 @@
 import os
 from fastapi import FastAPI, Request
-from fastapi.responses import JSONResponse, FileResponse
+from fastapi.responses import JSONResponse, HTMLResponse
 from fastapi.staticfiles import StaticFiles
+from fastapi.templating import Jinja2Templates
 from huggingface_hub import InferenceClient
 
 app = FastAPI()
 
-# إعدادات المحرك
+# إعداد النموذج
 HF_TOKEN = os.getenv("HF_TOKEN")
-MODEL_ID = "uihui-ai/Qwen2.5-72B-Instruct-abliterated"
+MODEL_ID = "huihui-ai/Qwen2.5-72B-Instruct-abliterated"
 client = InferenceClient(token=HF_TOKEN)
 
-# المسار الذي ستوضع فيه الملفات
-frontend_path = os.path.join(os.path.dirname(__file__), "temp_unzip", "dist")
+# إعداد القوالب والملفات الثابتة بناءً على هيكل مجلد api الخاص بك
+base_path = os.path.dirname(os.path.abspath(__file__))
+# ربط مجلد القوالب (templates) الذي يحتوي على index.html
+templates = Jinja2Templates(directory=os.path.join(base_path, "api", "templates"))
+# ربط مجلد الملفات الثابتة (static)
+app.mount("/static", StaticFiles(directory=os.path.join(base_path, "api", "static")), name="static")
 
-@app.get("/")
-async def serve_index():
-    index_path = os.path.join(frontend_path, "index.html")
-    if os.path.exists(index_path):
-        return FileResponse(index_path)
-    return JSONResponse({"error": f"Frontend build files not found at {frontend_path}"}, status_code=404)
-
-# ربط الملفات الثابتة
-if os.path.exists(frontend_path):
-    app.mount("/assets", StaticFiles(directory=os.path.join(frontend_path, "assets")), name="assets")
+@app.get("/", response_class=HTMLResponse)
+async def serve_index(request: Request):
+    # استخدام Jinja2 لعرض ملف index.html من مجلد templates
+    return templates.TemplateResponse("index.html", {"request": request})
 
 @app.post("/chat")
 async def chat_handler(request: Request):
     try:
         body = await request.json()
         messages = body.get("messages", [])
-        response = client.chat_completion(model=MODEL_ID, messages=messages, max_tokens=1000)
-        return JSONResponse({"choices": [{"message": {"role": "assistant", "content": response.choices[0].message.content}}]})
+        response = client.chat_completion(model=MODEL_ID, messages=messages, max_tokens=1500)
+        return JSONResponse({
+            "choices": [{"message": {"role": "assistant", "content": response.choices[0].message.content}}]
+        })
     except Exception as e:
         return JSONResponse({"error": str(e)}, status_code=500)