Update app.py

app.py

@@ -12,45 +12,36 @@ scenarios = {
     "shopping": "You are in a store. Help the user ask for prices and sizes.",
 }
 
-# دالة لاختيار السيناريو المناسب
+# دالة لاختيار الرسالة المناسبة للسيناريو
 def scenario_prompt(choice):
     return scenarios.get(choice, "You are a language tutor AI. Help users practice real-life conversations.")
 
-# دالة للمحادثة مع الذكاء الاصطناعي
-def respond(*args):
-    """
-    نتوقع أن يتم تمرير 7 معاملات بالترتيب التالي:
-    0: message
-    1: history
-    2: scenario (من قائمة الاختيارات)
-    3: system_message (قيمة مُدخلة، لكن سنستخدم scenario لتحديدها)
-    4: max_tokens
-    5: temperature
-    6: top_p
-    """
-    # التأكد من وجود العدد الكافي من المعاملات:
-    if len(args) < 7:
-        return "Insufficient arguments provided."
+# دالة المحادثة مع الذكاء الاصطناعي
+def respond(message, history, scenario, system_message, max_tokens, temperature, top_p):
+    # التأكد من أن history عبارة عن قائمة، وإذا لم توجد بيانات سابقة، اجعلها قائمة فارغة.
+    if history is None:
+        history = []
     
-    message = args[0]
-    history = args[1]
-    scenario = args[2]
-    # تجاهل system_message المُمرر واستبداله بالرسالة المناسبة للسيناريو:
-    system_message = scenario_prompt(scenario)
-    max_tokens = args[4]
-    temperature = args[5]
-    top_p = args[6]
-
-    messages = [{"role": "system", "content": system_message}]
-    # إذا كان history يحتوي على محادثات سابقة:
-    for user_msg, assistant_msg in history:
-        if user_msg:
-            messages.append({"role": "user", "content": user_msg})
-        if assistant_msg:
-            messages.append({"role": "assistant", "content": assistant_msg})
+    # استبدال system_message بالرسالة الخاصة بالسيناريو
+    sys_msg = scenario_prompt(scenario)
+    
+    # إعداد قائمة الرسائل بما يتوافق مع تنسيق Chat Completion
+    messages = [{"role": "system", "content": sys_msg}]
+    
+    # التأكد من أن history عبارة عن قائمة من الـ tuples
+    for pair in history:
+        if isinstance(pair, (list, tuple)) and len(pair) == 2:
+            user_msg, assistant_msg = pair
+            if user_msg:
+                messages.append({"role": "user", "content": user_msg})
+            if assistant_msg:
+                messages.append({"role": "assistant", "content": assistant_msg})
+    
+    # إضافة الرسالة الحالية للمستخدم
     messages.append({"role": "user", "content": message})
-
+    
     response = ""
+    # الحصول على الرد مع البث (streaming)
     for m in client.chat_completion(
         messages,
         max_tokens=max_tokens,
@@ -58,17 +49,15 @@ def respond(*args):
         temperature=temperature,
         top_p=top_p,
     ):
-        token = m.choices[0].delta.content
+        # استخراج النص من كل توكن متدفق
+        token = m.choices[0].delta.content if m.choices[0].delta.content is not None else ""
         response += token
         yield response
 
-
-
-
-# واجهة `Gradio` للتفاعل مع المستخدم
+# إنشاء واجهة Gradio للتفاعل مع المستخدم
 demo = gr.ChatInterface(
-    respond,
-    chatbot=gr.Chatbot(type="messages"),  # إصلاح التحذير باستخدام `type="messages"`
+    fn=respond,
+    chatbot=gr.Chatbot(type="messages"),
     additional_inputs=[
         gr.Dropdown(choices=list(scenarios.keys()), label="Choose a scenario", value="restaurant"),
         gr.Textbox(value=scenario_prompt("restaurant"), label="System message"),