Update app.py

app.py

@@ -4,95 +4,80 @@ from fastapi.responses import StreamingResponse, JSONResponse, FileResponse
 from fastapi.staticfiles import StaticFiles
 from huggingface_hub import InferenceClient
 from duckduckgo_search import DDGS
+import fitz  # PyMuPDF لقراءة ملفات PDF
 
-# إعداد التطبيق والمسارات
 app = FastAPI()
 UPLOAD_DIR = "static/uploads"
 os.makedirs(UPLOAD_DIR, exist_ok=True)
-
-# ربط الملفات الثابتة
 app.mount("/static", StaticFiles(directory="static"), name="static")
 
-# إعداد العميل (يجب إضافة HF_TOKEN في Secrets)
 HF_TOKEN = os.environ.get("HF_TOKEN")
 client = InferenceClient(api_key=HF_TOKEN)
 
 MODELS = {
     "qwen_abliterated": "huihui-ai/Qwen2.5-72B-Instruct-abliterated",
-    "internvl_vision": "OpenGVLab/InternVL2-Llama3-76B",
-    "kimi_k2": "moonshotai/Kimi-K2-Instruct"
 }
 
-def web_search(query):
+# دالة لقراءة محتوى الملف المرفوع
+def extract_file_content(file_id):
+    filepath = os.path.join(UPLOAD_DIR, file_id)
+    if not os.path.exists(filepath):
+        return ""
+    
+    ext = os.path.splitext(file_id)[1].lower()
+    content = ""
+    
     try:
-        with DDGS() as ddgs:
-            results = [r for r in ddgs.text(query, max_results=3)]
-            return "\n".join([f"العنوان: {r['title']}\nالوصف: {r['body']}" for r in results])
-    except: return "فشل البحث في الويب."
+        if ext == ".pdf":
+            doc = fitz.open(filepath)
+            for page in doc:
+                content += page.get_text()
+        elif ext in [".txt", ".py", ".js", ".html", ".css"]:
+            with open(filepath, "r", encoding="utf-8") as f:
+                content = f.read()
+    except Exception as e:
+        return f"خطأ في قراءة الملف: {str(e)}"
+    
+    return content[:10000] # نأخذ أول 10 آلاف حرف لضمان عدم تجاوز حدود الذاكرة
 
 @app.get("/")
 async def read_index():
-    # البحث عن ملف index.html في مكانين مختلفين لتجنب خطأ 404
-    paths = ["static/index.html", "index.html"]
-    for path in paths:
-        if os.path.exists(path):
-            return FileResponse(path)
-    return JSONResponse({"error": "ملف index.html غير موجود!"}, status_code=404)
+    path = "index.html" if os.path.exists("index.html") else "static/index.html"
+    return FileResponse(path)
 
 @app.post("/api/upload")
 async def upload_file(file: UploadFile = File(...)):
-    try:
-        file_id = str(uuid.uuid4())
-        ext = os.path.splitext(file.filename)[1]
-        filename = f"{file_id}{ext}"
-        filepath = os.path.join(UPLOAD_DIR, filename)
-        
-        with open(filepath, "wb") as f:
-            f.write(await file.read())
-            
-        # إضافة file_id لأن جافاسكريبت في index.html تتوقعه
-        return {
-            "file_id": filename, 
-            "url": f"/static/uploads/{filename}", 
-            "filename": file.filename
-        }
-    except Exception as e:
-        return JSONResponse({"error": str(e)}, status_code=500)
+    file_id = f"{uuid.uuid4()}{os.path.splitext(file.filename)[1]}"
+    filepath = os.path.join(UPLOAD_DIR, file_id)
+    with open(filepath, "wb") as f:
+        f.write(await file.read())
+    return {"file_id": file_id, "url": f"/static/uploads/{file_id}"}
 
 @app.post("/api/chat/stream")
 async def chat_stream(request: Request):
     data = await request.json()
     message = data.get("message", "")
-    model_choice = data.get("model", "qwen_abliterated")
-    tools_enabled = data.get("tools", [])
-    file_id = data.get("file_id") # استقبال معرف الملف إذا وجد
+    file_id = data.get("file_id") # استقبال معرف الملف من الواجهة
     
-    current_model = MODELS.get(model_choice, MODELS["qwen_abliterated"])
-    system_prompt = "أنت مساعد ذكي ومحرر من القيود. أجب بوضوح وباللغة العربية."
-
+    system_prompt = "أنت مساعد ذكي ومحرر. إذا تم تزويدك بمحتوى ملف، قم بتحليله والإجابة بناءً عليه."
     context_extra = ""
-    
-    # إذا تم رفع ملف، يمكن إضافة ملاحظة للنموذج (تطوير مستقبلي لقراءة المحتوى)
-    if file_id:
-        context_extra += f"\n[ملاحظة: قام المستخدم برفع ملف برمز: {file_id}]\n"
 
-    if "web_search" in tools_enabled and len(message) > 5:
-        context_extra += f"\nنتائج البحث:\n{web_search(message)}"
+    # --- الجزء الهام: قراءة الملف ودمجه في السياق ---
+    if file_id:
+        file_text = extract_file_content(file_id)
+        if file_text:
+            context_extra = f"\nمحتوى الملف المرفوع:\n{file_text}\n"
 
-    full_message = f"{context_extra}\n\nسؤال المستخدم: {message}"
-    messages = [
-        {"role": "system", "content": system_prompt},
-        {"role": "user", "content": full_message}
-    ]
+    full_message = f"{context_extra}\nسؤال المستخدم: {message}"
+    messages = [{"role": "system", "content": system_prompt}, {"role": "user", "content": full_message}]
 
     async def gen():
         try:
             stream = client.chat.completions.create(
-                model=current_model,
+                model=MODELS["qwen_abliterated"],
                 messages=messages,
                 stream=True,
-                max_tokens=2048,
-                temperature=0.7
+                max_tokens=4096
             )
             for chunk in stream:
                 if chunk.choices[0].delta.content: