Update app.py

app.py

@@ -13,10 +13,9 @@ chroma_client = chromadb.PersistentClient(path=CHROMA_DATA_PATH)
 ef = embedding_functions.DefaultEmbeddingFunction()
 collection = chroma_client.get_or_create_collection(name="yousef_vault", embedding_function=ef)
 
-# النموذج الأساسي الافتراضي (يمكنك تغييره من هنا)
 DEFAULT_MODEL = "huihui-ai/Qwen2.5-72B-Instruct-abliterated"
 
-# --- 2. دالة استخراج النصوص (يدعم PDF, DOCX, TXT) ---
+# --- 2. دالة استخراج النصوص ---
 def process_document(file):
     if file is None: return "⚠️ لم يتم اختيار ملف."
     text = ""
@@ -34,95 +33,67 @@ def process_document(file):
                 text = f.read()
         
         if not text.strip():
-            return "❌ تعذر استخراج نص. تأكد أن الملف ليس صوراً فقط."
+            return "❌ تعذر استخراج نص."
 
-        # تقسيم النص لقطع وحقنها في الذاكرة
         chunks = [text[i:i+2000] for i in range(0, len(text), 2000)]
         ids = [str(uuid.uuid4()) for _ in chunks]
         collection.add(documents=chunks, ids=ids)
         
-        return f"✅ تمت الأرشفة: {len(chunks)} شظية معرفية حُقنت في الذاكرة."
+        return f"✅ تمت الأرشفة: {len(chunks)} شظية معرفية."
     except Exception as e:
-        return f"❌ خطأ في المعالجة: {str(e)}"
+        return f"❌ خطأ: {str(e)}"
 
-# --- 3. دالة الحوار (RAG) مع معالجة أخطاء الـ Stream ---
-def predict(message, history, system_prompt, temperature, custom_model, oauth_token: gr.OAuthToken | None):
-    # استخدام توكن المستخدم المسجل (OAuth) لخصم الرصيد الشخصي
-    api_key = oauth_token.token if oauth_token else os.getenv("HF_TOKEN")
+# --- 3. دالة الحوار المصححة ---
+def predict(message, history, system_prompt, temperature, custom_model):
+    # ملاحظة: تم إزالة oauth_token من هنا لتبسيط الاتصال بالـ API 
+    # تأكد من وضع HF_TOKEN في Secrets إذا كنت ترفعه على Space
+    api_key = os.getenv("HF_TOKEN")
     
     if not api_key:
-        yield "⚠️ يرجى تسجيل الدخول عبر Hugging Face أولاً للوصول إلى رصيدك الشخصي."
+        yield "⚠️ يرجى ضبط HF_TOKEN في الإعدادات."
         return
 
-    client = OpenAI(
-        base_url="https://router.huggingface.co/v1",
-        api_key=api_key
-    )
-    
-    # تحديد النموذج المستخدم
+    client = OpenAI(base_url="https://router.huggingface.co/v1", api_key=api_key)
     active_model = custom_model.strip() if custom_model and custom_model.strip() else DEFAULT_MODEL
     
-    # استرجاع سياق من الذاكرة
-    results = collection.query(query_texts=[message], n_results=10)
+    # RAG: استرجاع السياق
+    results = collection.query(query_texts=[message], n_results=5)
     context = "\n".join(results['documents'][0]) if results['documents'] else ""
     
-    # بناء قائمة الرسائل
     messages = [{"role": "system", "content": f"{system_prompt}\n\n[CONTEXT]:\n{context}"}]
-    
+    # تحويل التاريخ إلى صيغة OpenAI المتوافقة
     for msg in history:
-        messages.append({"role": msg['role'], "content": msg['content']})
-        
+        messages.append({"role": msg["role"], "content": msg["content"]})
     messages.append({"role": "user", "content": message})
 
     try:
         response = client.chat.completions.create(
             model=active_model,
             messages=messages,
-            temperature=temperature,
+            temperature=float(temperature),
             stream=True
         )
 
         partial_message = ""
         for chunk in response:
-            # حماية ضد خطأ list index out of range
-            if hasattr(chunk, 'choices') and chunk.choices and len(chunk.choices) > 0:
-                content = chunk.choices[0].delta.content
-                if content:
-                    partial_message += content
-                    yield partial_message
-            else:
-                # تجاهل الإطارات الفارغة التي تظهر في نهاية البث
-                continue
-                
+            if chunk.choices and chunk.choices[0].delta.content:
+                partial_message += chunk.choices[0].delta.content
+                yield partial_message
     except Exception as e:
-        if partial_message:
-            # إذا كان هناك نص تم توليده بالفعل، اعرضه ولا تظهر رسالة الخطأ
-            yield partial_message
-        else:
-            yield f"❌ فشل الاتصال بالنموذج ({active_model}): {str(e)}"
+        yield f"❌ فشل: {str(e)}"
 
-# --- 4. بناء الواجهة (بسيطة، داكنة، ومتجاوبة) ---
-with gr.Blocks(fill_height=True, theme=gr.themes.Soft(primary_hue="indigo")) as demo:
+# --- 4. بناء الواجهة (التحديث الجديد) ---
+with gr.Blocks(fill_height=True) as demo:
     with gr.Sidebar():
         gr.Markdown("# 🧬 دهليز يوسف")
-        gr.LoginButton("Sign in")
         
-        with gr.Accordion("⚙️ الضبط والوعي", open=True):
-            model_input = gr.Textbox(
-                label="المحرك المستهدف (Model ID)",
-                placeholder="اتركه فارغاً للنموذج الأساسي",
-                value="",
-                lines=1
-            )
-            system_input = gr.Textbox(
-                value="أنت رفيق حكيم يغوص في التيه مع المستخدم. أجب بصدق مستعيناً بالذاكرة.", 
-                label="البرومبت النظامي"
-            )
-            temp_slider = gr.Slider(0.1, 1.5, 0.8, label="درجة التشظي (Temperature)")
+        with gr.Accordion("⚙️ الضبط", open=True):
+            model_input = gr.Textbox(label="المحرك المستهدف", placeholder="Default Qwen", value="")
+            system_input = gr.Textbox(value="أنت رفيق حكيم.", label="البرومبت النظامي")
+            temp_slider = gr.Slider(0.1, 1.5, 0.8, label="Temperature")
         
         gr.Markdown("---")
-        gr.Markdown("### 📂 ملقم الذاكرة (PDF, DOCX, TXT)")
-        file_box = gr.File(label="ارفع مسوداتك هنا", file_types=[".pdf", ".docx", ".txt"])
+        file_box = gr.File(label="ارفع مسوداتك", file_types=[".pdf", ".docx", ".txt"])
         status_msg = gr.Markdown("*الذاكرة بانتظار الملفات...*")
         
         file_box.change(process_document, inputs=file_box, outputs=status_msg)
@@ -130,17 +101,17 @@ with gr.Blocks(fill_height=True, theme=gr.themes.Soft(primary_hue="indigo")) as
         clear_btn = gr.Button("🗑 مسح الذاكرة")
         def clear_memory():
             all_data = collection.get()
-            if all_data['ids']: 
-                collection.delete(ids=all_data['ids'])
-            return "🗑 تم تصفير الذاكرة بالكامل."
+            if all_data['ids']: collection.delete(ids=all_data['ids'])
+            return "🗑 تم تصفير الذاكرة."
         clear_btn.click(clear_memory, outputs=status_msg)
 
+    # التعديل هنا: حذف type="messages" لأنه أصبح افتراضياً أو غير مطلوب بهذا الشكل
     gr.ChatInterface(
         predict,
         additional_inputs=[system_input, temp_slider, model_input],
-        fill_height=True,
-        type="messages"
+        fill_height=True
     )
 
 if __name__ == "__main__":
-    demo.launch()
+    # تم نقل theme إلى هنا حسب تحذير Gradio 6
+    demo.launch(theme=gr.themes.Soft(primary_hue="indigo"))