Update app.py

app.py

@@ -2,49 +2,185 @@ import gradio as gr
 from openai import OpenAI
 import os
 
-# التوكن من الـ Secrets
+# ---------------------------------------------------------
+# 1. إعدادات الاتصال والنموذج
+# ---------------------------------------------------------
 hf_token = os.getenv("HF_TOKEN")
 
-# العنوان الرسمي والمستقر للـ Router
 client = OpenAI(
     base_url="https://router.huggingface.co/hf-inference/v1",
     api_key=hf_token
 )
 
-MODEL_ID = "zetasepic/Qwen2.5-72B-Instruct-abliterated-v2"
-def chat_function(message, history):
-    # تحويل التاريخ إلى الصيغة التي يفهمها OpenAI
-    messages = []
-    for h, a in history:
-        messages.append({"role": "user", "content": h})
-        messages.append({"role": "assistant", "content": a})
+MODEL_ID = "huihui-ai/Qwen2.5-72B-Instruct-abliterated"
+
+# ---------------------------------------------------------
+# 2. منطق المعالجة (Back-end Logic)
+# ---------------------------------------------------------
+def predict(message, history, system_prompt, temperature, max_tokens):
+    """
+    دالة التفاعل مع النموذج.
+    تستقبل الرسالة، التاريخ، وإعدادات التخصيص.
+    """
+    messages_payload = []
+
+    # إضافة طبقة التخصيص (System Prompt) إذا وجدت
+    if system_prompt and system_prompt.strip():
+        messages_payload.append({"role": "system", "content": system_prompt})
+
+    # تحويل سجل المحادثة من صيغة Gradio إلى صيغة OpenAI
+    # history يأتي كقائمة: [[user_msg, bot_msg], ...]
+    for user_msg, bot_msg in history:
+        if user_msg:
+            messages_payload.append({"role": "user", "content": user_msg})
+        if bot_msg:
+            messages_payload.append({"role": "assistant", "content": bot_msg})
     
-    messages.append({"role": "user", "content": message})
+    # إضافة الرسالة الحالية
+    messages_payload.append({"role": "user", "content": message})
 
     try:
         response = client.chat.completions.create(
             model=MODEL_ID,
-            messages=messages,
-            max_tokens=2048,
-            temperature=0.85,
+            messages=messages_payload,
+            max_tokens=int(max_tokens),
+            temperature=float(temperature),
             stream=True
         )
 
-        full_response = ""
+        partial_message = ""
         for chunk in response:
             if chunk.choices[0].delta.content:
-                full_response += chunk.choices[0].delta.content
-                yield full_response
-                
+                partial_message += chunk.choices[0].delta.content
+                yield partial_message
+
     except Exception as e:
-        yield f"⚠️ انكسار في الواقع التقني: {str(e)}"
+        yield f"⚠️ System Error: {str(e)}"
 
-# حذفنا الوسيط 'type' لتجنب الخطأ
-demo = gr.ChatInterface(
-    fn=chat_function,
-    title="دهليز يوسف: Qwen-72B",
-    description="بوابة الحوار مع العقل المُحرر من القيود.",
+# ---------------------------------------------------------
+# 3. واجهة المستخدم الاحترافية (Front-end / UI)
+# ---------------------------------------------------------
+# استخدام ثيم ناعم واحترافي
+theme = gr.themes.Soft(
+    primary_hue="blue",
+    secondary_hue="slate",
+    neutral_hue="slate",
 )
 
+css = """
+footer {visibility: hidden}
+.container {max-width: 1200px; margin: auto; padding-top: 20px}
+"""
+
+with gr.Blocks(theme=theme, css=css, title="Qwen Pro Interface") as demo:
+    
+    gr.Markdown(
+        """
+        # 🤖 Qwen 2.5-72B Professional Interface
+        <div style='opacity: 0.7; font-size: 0.9em;'>
+        Advanced interface for 'huihui-ai/Qwen2.5-72B-Instruct-abliterated' via HF Inference Router.
+        </div>
+        """
+    )
+
+    with gr.Tabs():
+        # ================= TAB 1: ساحة المحادثة =================
+        with gr.TabItem("💬 Chat Workspace", id="chat_tab"):
+            with gr.Row():
+                # عمود المحادثة
+                with gr.Column(scale=4):
+                    chatbot = gr.Chatbot(
+                        height=600,
+                        placeholder="Start a new conversation...",
+                        show_copy_button=True,
+                        avatar_images=(None, "🤖"), # يمكن وضع مسار صورة للمستخدم وللبوت
+                        layout="bubble"
+                    )
+                    
+                    with gr.Row():
+                        msg_input = gr.Textbox(
+                            placeholder="Type your message here...",
+                            show_label=False,
+                            scale=8,
+                            container=False,
+                            autofocus=True
+                        )
+                        send_btn = gr.Button("Send ➤", variant="primary", scale=1)
+                        clear_btn = gr.Button("🗑️ Clear", variant="stop", scale=1)
+
+        # ================= TAB 2: إعدادات النموذج =================
+        with gr.TabItem("⚙️ Model Settings & Customization", id="settings_tab"):
+            gr.Markdown("### 🧠 Model Behavior (Custom Layer)")
+            
+            with gr.Row():
+                with gr.Column():
+                    system_prompt_input = gr.TextArea(
+                        label="System Prompt (Custom Instruction Layer)",
+                        value="You are a helpful, professional, and intelligent AI assistant.",
+                        placeholder="Define the AI's persona, rules, or role here. (e.g., 'You are a senior python developer...')",
+                        lines=5,
+                        info="This acts as a base layer for the model's behavior throughout the chat."
+                    )
+            
+            gr.Markdown("### 🎛️ Inference Parameters")
+            with gr.Row():
+                with gr.Column():
+                    temp_slider = gr.Slider(
+                        minimum=0.0, maximum=1.5, value=0.7, step=0.05,
+                        label="Temperature (Creativity)",
+                        info="Lower values for precise facts, higher for creative writing."
+                    )
+                with gr.Column():
+                    tokens_slider = gr.Slider(
+                        minimum=256, maximum=8192, value=2048, step=256,
+                        label="Max Tokens (Response Length)",
+                        info="Maximum number of tokens the model can generate."
+                    )
+
+    # ---------------------------------------------------------
+    # 4. ربط المكونات (Wiring)
+    # ---------------------------------------------------------
+    
+    # دالة مساعدة لتحديث واجهة المستخدم (مسح صندوق النص وإضافة رسالة المستخدم للشات)
+    def user_turn(user_message, history):
+        return "", history + [[user_message, None]]
+
+    # دالة مساعدة لتوليد الرد وتحديث الشات
+    def bot_turn(history, system_p, temp, tokens):
+        user_message = history[-1][0]
+        # نقوم باستدعاء المولد (generator)
+        bot_response_generator = predict(user_message, history[:-1], system_p, temp, tokens)
+        
+        history[-1][1] = ""
+        for chunk in bot_response_generator:
+            history[-1][1] = chunk
+            yield history
+
+    # عند الضغط على زر الإرسال أو Enter
+    send_event = msg_input.submit(
+        user_turn, 
+        [msg_input, chatbot], 
+        [msg_input, chatbot], 
+        queue=False
+    ).then(
+        bot_turn, 
+        [chatbot, system_prompt_input, temp_slider, tokens_slider], 
+        chatbot
+    )
+
+    click_event = send_btn.click(
+        user_turn, 
+        [msg_input, chatbot], 
+        [msg_input, chatbot], 
+        queue=False
+    ).then(
+        bot_turn, 
+        [chatbot, system_prompt_input, temp_slider, tokens_slider], 
+        chatbot
+    )
+
+    # زر المسح
+    clear_btn.click(lambda: None, None, chatbot, queue=False)
+
 if __name__ == "__main__":
-    demo.launch()
+    demo.queue().launch()