Update app.py

app.py

@@ -7,11 +7,18 @@ from dashscope import Generation
 from dashscope.api_entities.dashscope_response import Role
 from typing import List, Optional, Tuple, Dict
 from urllib.error import HTTPError
+
+# الإعدادات الافتراضية
 default_system = 'You are a helpful assistant.'
+default_temp = 0.7
+default_top_p = 0.9
+default_max_tokens = 1024
+default_max_history = 5
 
 YOUR_API_TOKEN = os.getenv('YOUR_API_TOKEN')
 dashscope.api_key = YOUR_API_TOKEN
 
+# تعريف أنواع البيانات
 History = List[Tuple[str, str]]
 Messages = List[Dict[str, str]]
 
@@ -19,18 +26,21 @@ def clear_session() -> History:
     return '', []
 
 def modify_system_session(system: str) -> str:
-    if system is None or len(system) == 0:
+    if not system:
         system = default_system
     return system, system, []
 
-def history_to_messages(history: History, system: str) -> Messages:
+def history_to_messages(history: History, system: str, max_history: int) -> Messages:
+    # تحديد عدد الرسائل التاريخية المسموح بها
+    if max_history > 0:
+        history = history[-max_history:]
+    
     messages = [{'role': Role.SYSTEM, 'content': system}]
     for h in history:
         messages.append({'role': Role.USER, 'content': h[0]})
         messages.append({'role': Role.ASSISTANT, 'content': h[1]})
     return messages
 
-
 def messages_to_history(messages: Messages) -> Tuple[str, History]:
     assert messages[0]['role'] == Role.SYSTEM
     system = messages[0]['content']
@@ -39,59 +49,101 @@ def messages_to_history(messages: Messages) -> Tuple[str, History]:
         history.append([q['content'], r['content']])
     return system, history
 
-
-def model_chat(query: Optional[str], history: Optional[History], system: str
+def model_chat(
+    query: Optional[str],
+    history: Optional[History],
+    system: str,
+    temperature: float,
+    top_p: float,
+    max_tokens: int,
+    max_history: int
 ) -> Tuple[str, str, History]:
-    if query is None:
+    if not query:
         query = ''
-    if history is None:
-        history = []
-    messages = history_to_messages(history, system)
+    history = history or []
+    
+    # تحويل التاريخ إلى رسائل مع تطبيق الحد الأقصى
+    messages = history_to_messages(history, system, max_history)
     messages.append({'role': Role.USER, 'content': query})
-    gen = Generation.call(
-        model = "qwen-72b-chat",
-        messages=messages,
-        result_format='message',
-        stream=True
-    )
-    for response in gen:
-        if response.status_code == HTTPStatus.OK:
-            role = response.output.choices[0].message.role
-            response = response.output.choices[0].message.content
-            system, history = messages_to_history(messages + [{'role': role, 'content': response}])
-            yield '', history, system
-        else:
-            raise HTTPError('Request id: %s, Status code: %s, error code: %s, error message: %s' % (
-                response.request_id, response.status_code,
-                response.code, response.message
-            ))
-
+    
+    try:
+        gen = Generation.call(
+            model="qwen-72b-chat",
+            messages=messages,
+            result_format='message',
+            stream=True,
+            temperature=temperature,
+            top_p=top_p,
+            max_tokens=max_tokens
+        )
+        
+        for response in gen:
+            if response.status_code == HTTPStatus.OK:
+                role = response.output.choices[0].message.role
+                content = response.output.choices[0].message.content
+                system, history = messages_to_history(messages + [{'role': role, 'content': content}])
+                yield '', history, system
+            else:
+                raise HTTPError(f'Error: {response.code} - {response.message}')
+    except Exception as e:
+        raise gr.Error(f'حدث خطأ في الاتصال: {str(e)}')
 
+# واجهة المستخدم
 with gr.Blocks() as demo:
     gr.Markdown("""<center><font size=8>Qwen-72B-Chat</center>""")
-    gr.Markdown("""<center><font size=4>Qwen-72B-Chat is the 72-billion parameter chat model of the Qwen series.</center>""")
-
+    gr.Markdown("""<center><font size=4>نموذج محادثة متقدم بدعم الذاكرة والإعدادات</center>""")
+    
+    # قسم إعدادات النظام
     with gr.Row():
         with gr.Column(scale=3):
-            system_input = gr.Textbox(value=default_system, lines=1, label='System')
+            system_input = gr.Textbox(value=default_system, lines=2, label='تعليمات النظام')
+            gr.Markdown("""
+            **ملاحظة:** يمكنك كتابة توجيهات خاصة للنموذج هنا. مثال:
+            - "أنت مساعد يتحدث العربية بطلاقة"
+            - "أجب دائمًا بصيغة الشعر"
+            """)
         with gr.Column(scale=1):
-            modify_system = gr.Button("🛠️ Set system prompt and clear history.", scale=2)
+            modify_system = gr.Button("🛠️ تحديث التعليمات ومسح الذاكرة")
         system_state = gr.Textbox(value=default_system, visible=False)
-    chatbot = gr.Chatbot(label='Qwen-72B-Chat')
-    textbox = gr.Textbox(lines=2, label='Input')
-
+    
+    # عناصر التحكم في النموذج
+    with gr.Accordion("الإعدادات المتقدمة", open=False):
+        with gr.Row():
+            temperature = gr.Slider(0.0, 2.0, value=default_temp, label="الابتكار (Temperature)")
+            top_p = gr.Slider(0.0, 1.0, value=default_top_p, label="الدقة (Top-P)")
+        with gr.Row():
+            max_tokens = gr.Number(default_max_tokens, label="الحد الأقصى للرموز")
+            max_history = gr.Number(default_max_history, label="الحد الأقصى للذاكرة (عدد الدورات)")
+    
+    # منطقة المحادثة
+    chatbot = gr.Chatbot(label='المحادثة')
+    textbox = gr.Textbox(lines=2, label='الرسالة')
+    
+    # أزرار التحكم
     with gr.Row():
-        clear_history = gr.Button("🧹 Clear history")
-        sumbit = gr.Button("🚀 Send")
-
-    sumbit.click(model_chat,
-                 inputs=[textbox, chatbot, system_state],
-                 outputs=[textbox, chatbot, system_input])
-    clear_history.click(fn=clear_session,
-                        inputs=[],
-                        outputs=[textbox, chatbot])
-    modify_system.click(fn=modify_system_session,
-                        inputs=[system_input],
-                        outputs=[system_state, system_input, chatbot])
+        clear_history = gr.Button("🧹 مسح المحادثة")
+        submit = gr.Button("🚀 إرسال")
+    
+    # معالجة الأحداث
+    submit.click(
+        model_chat,
+        inputs=[textbox, chatbot, system_state, temperature, top_p, max_tokens, max_history],
+        outputs=[textbox, chatbot, system_input]
+    )
+    clear_history.click(
+        fn=clear_session,
+        inputs=[],
+        outputs=[textbox, chatbot]
+    )
+    modify_system.click(
+        fn=modify_system_session,
+        inputs=[system_input],
+        outputs=[system_state, system_input, chatbot]
+    )
 
-demo.queue(api_open=False).launch(max_threads=10,height=800, share=False)
+# تشغيل التطبيق
+demo.queue(api_open=False).launch(
+    max_threads=10,
+    height=800,
+    share=False
+)