Update app.py

app.py

@@ -12,10 +12,11 @@ UPLOAD_DIR = "static/uploads"
 os.makedirs(UPLOAD_DIR, exist_ok=True)
 app.mount("/static", StaticFiles(directory="static"), name="static")
 
+# التأكد من جلب التوكن
 HF_TOKEN = os.environ.get("HF_TOKEN")
+# استخدام العميل مع التوكن مباشرة
 client = InferenceClient(api_key=HF_TOKEN)
 
-# دالة استخراج المحتوى من مختلف أنواع الملفات
 def extract_file_content(file_id):
     filepath = os.path.join(UPLOAD_DIR, file_id)
     if not os.path.exists(filepath): return ""
@@ -31,16 +32,16 @@ def extract_file_content(file_id):
         elif ext == ".zip":
             with zipfile.ZipFile(filepath, 'r') as z:
                 names = z.namelist()
-                content = f"ملف مضغوط يحتوي على: {', '.join(names)}\n"
-                for name in names[:3]: # قراءة أول 3 ملفات نصية فقط لتجنب الضغط
+                content = f"ملف ZIP يحتوي على: {', '.join(names)}\n"
+                for name in names[:3]:
                     if name.endswith(('.txt', '.py', '.html', '.json')):
                         with z.open(name) as f:
-                            content += f"\n--- محتوى {name} ---\n{f.read().decode('utf-8')[:1000]}"
+                            content += f"\n-- {name} --\n{f.read().decode('utf-8')[:1000]}"
         elif ext in [".txt", ".py", ".js", ".html", ".json"]:
             with open(filepath, "r", encoding="utf-8") as f:
                 content = f.read()
     except Exception as e:
-        return f"خطأ في قراءة الملف: {str(e)}"
+        return f"خطأ قراءة: {str(e)}"
     return content[:15000]
 
 @app.get("/")
@@ -50,7 +51,6 @@ async def read_index():
 
 @app.post("/api/upload")
 async def upload_file(file: UploadFile = File(...)):
-    # حفظ الملف بامتداده الأصلي
     ext = os.path.splitext(file.filename)[1]
     file_id = f"{uuid.uuid4()}{ext}"
     filepath = os.path.join(UPLOAD_DIR, file_id)
@@ -64,26 +64,35 @@ async def chat_stream(request: Request):
     message = data.get("message", "")
     file_id = data.get("file_id")
     
+    if not HF_TOKEN:
+        return JSONResponse({"error": "HF_TOKEN missing in Settings!"}, status_code=500)
+
     context = ""
     if file_id:
         file_text = extract_file_content(file_id)
-        context += f"\n[محتوى الملف المرفوع]:\n{file_text}\n"
+        context += f"\n[محتوى الملف]:\n{file_text}\n"
 
     messages = [
-        {"role": "system", "content": "أنت مساعد خبير. حلل الملفات المرفقة بدقة وأجب بالعربية."},
-        {"role": "user", "content": f"{context}\nسؤال المستخدم: {message}"}
+        {"role": "system", "content": "أنت مساعد ذكي. حلل الملفات وأجب بالعربية."},
+        {"role": "user", "content": f"{context}\n{message}"}
     ]
 
     async def gen():
         try:
+            # تم تحديث الاستدعاء ليكون أكثر استقراراً
             stream = client.chat.completions.create(
                 model="huihui-ai/Qwen2.5-72B-Instruct-abliterated",
-                messages=messages, stream=True, max_tokens=2048
+                messages=messages,
+                stream=True,
+                max_tokens=2048,
+                temperature=0.7
             )
             for chunk in stream:
                 if chunk.choices[0].delta.content:
                     yield f"data: {json.dumps({'token': chunk.choices[0].delta.content})}\n\n"
         except Exception as e:
-            yield f"data: {json.dumps({'error': str(e)})}\n\n"
+            # إرسال الخطأ الحقيقي للواجهة لنعرف السبب (مثل توكن خاطئ أو ضغط على السيرفر)
+            error_msg = str(e)
+            yield f"data: {json.dumps({'error': error_msg})}\n\n"
 
     return StreamingResponse(gen(), media_type="text/event-stream")