Update main.py

main.py

@@ -1,19 +1,20 @@
-from gradio_client import Client  # [تعديل] استيراد مكتبة gradio_client للاتصال بالسبيس
-import uvicorn
+import os
 import asyncio
 import json
+import uvicorn
 from fastapi import FastAPI, Request
 from fastapi.responses import StreamingResponse, HTMLResponse
+from gradio_client import Client
 
 app = FastAPI(title="ONYX Engine")
 
-# [تعديل] تهيئة عميل Gradio للاتصال بـ Hugging Face Space بدل OpenRouter
-client = Client("ONYX-APP/qwen-onnx-chat", hf_token = os.getenv("HF_TOKEN")
+# [تعديل] تهيئة عميل Gradio للاتصال بـ Hugging Face Space مع إصلاح الأقواس وos
+client = Client("ONYX-APP/qwen-onnx-chat", hf_token=os.getenv("HF_TOKEN"))
 
 # --- إضافة Route للرئيسية لحل مشكلة الـ 404 ---
 @app.get("/")
 async def root():
-    return {"status": "ONYX Engine is running", "model": "google/gemma-4-E2B-it"}  # [تعديل] تحديث اسم الموديل ليتطابق مع السبيس
+    return {"status": "ONYX Engine is running", "model": "google/gemma-4-E2B-it"}
 
 @app.post("/predict")
 async def generate(request: Request):
@@ -25,13 +26,13 @@ async def generate(request: Request):
     for msg in messages:
         if msg.get("role") == "user":
             last_message = msg.get("content", "")
-
+            
     async def generate_chunks():
         try:
             # [تعديل] تشغيل طلب السبيس بشكل غير متزامن وتدفق الجواب الناتج
             def run_client():
                 return client.predict(last_message, [], fn_index=0)
-            
+                
             response_text = await asyncio.to_thread(run_client)
             
             # إرسال النص الناتج على دفعات متتالية لتوافق الـ StreamingResponse