Update app.py

app.py

@@ -64,18 +64,17 @@ SYSTEM_PROMPT = f"""You are Foton, Swahili AI assistant. Tasks:
 Maintain friendly, patient demeanor with cultural context.
 """
 
-WELCOME_MESSAGE = "**Karibu Lugha Tausi!** Mimi ni Foton, msaidizi wako wa Kiswahili. Niko hapa kukusaidia kujifunza na kuzungumza Kiswahili. **Ninaweza kukusaidiaje leo?** 😊"
-
+# Update the WELCOME_MESSAGE format
+WELCOME_MESSAGE = {
+    "role": "assistant",
+    "content": "**Karibu Lugha Tausi!** Mimi ni Foton, msaidizi wako wa Kiswahili. Niko hapa kukusaidia kujifunza na kuzungumza Kiswahili. **Ninaweza kukusaidiaje leo?** 😊"
+}
 def format_messages(history):
     """Format chat history with system prompt"""
     messages = [{"role": "system", "content": SYSTEM_PROMPT}]
-    for entry in history:
-        if isinstance(entry, tuple):
-            user, bot = entry
-            messages.extend([
-                {"role": "user", "content": user},
-                {"role": "assistant", "content": bot}
-            ])
+    for msg in history:
+        if isinstance(msg, dict) and "role" in msg and "content" in msg:
+            messages.append(msg)
     return messages
 
 def generate_response(message, history):
@@ -103,23 +102,31 @@ def generate_response(message, history):
 with gr.Blocks(theme=gr.themes.Soft()) as demo:
     gr.Markdown("# Lugha Tausi - Swahili Assistant")
     
+    # Update the Chatbot initialization
     chatbot = gr.Chatbot(
-        value=[[None, WELCOME_MESSAGE]],
-        height=600,
-        show_label=False,
-        avatar_images=(None, "user.png"),
-        bubble_full_width=False,
-        show_share_button=False,
-        type="messages"
+    value=[WELCOME_MESSAGE],  # Now using proper message format
+    height=600,
+    show_label=False,
+    avatar_images=(None, "user.png"),
+    bubble_full_width=False,
+    show_share_button=False,
+    type="messages"
     )
     
     msg = gr.Textbox(placeholder="Andika ujumbe wako hapa...", show_label=False)
     clear = gr.Button("Futa Mazungumzo")
 
     def respond(message, chat_history):
-        bot_response = generate_response(message, chat_history)
-        chat_history.append((message, bot_response))
-        return "", chat_history
+            # Add user message
+            chat_history.append({"role": "user", "content": message})
+            
+            # Generate response
+            bot_response = generate_response(message, chat_history)
+            
+            # Add assistant response
+            chat_history.append({"role": "assistant", "content": bot_response})
+            
+            return "", chat_history
 
     msg.submit(respond, [msg, chatbot], [msg, chatbot])
     clear.click(lambda: None, None, chatbot, queue=False)