Fix claude mistake v7

app.py

@@ -33,10 +33,9 @@ tts_model = TTS(model_name=TTS_MODEL_NAME, progress_bar=False)
 def chat_with_bot(message, history, chat_history_ids=None):
     """
     Chat with the conversational AI model using DialoGPT.
-    CRITICAL FIX: Ensures history is a list of tuples [("user", "bot"), ...]
-    Returns: (updated_history, updated_chat_ids, response_text)
+    CRITICAL FIX: Uses Dictionary format {"role": "...", "content": "..."}
     """
-    # Ensure history is correctly initialized as a list of tuples
+    # Ensure history is a list
     if history is None:
         history = []
 
@@ -73,14 +72,15 @@ def chat_with_bot(message, history, chat_history_ids=None):
             chat_history_ids[:, bot_input_ids.shape[-1] :][0], skip_special_tokens=True
         )
 
-        # FIX: Append to history in the required Gradio Chatbot (list of tuples) format
-        history.append((message, response))
+        # CRITICAL FIX: Append dictionaries, not tuples
+        history.append({"role": "user", "content": message})
+        history.append({"role": "assistant", "content": response})
 
         return history, chat_history_ids, response
 
     except Exception as e:
-        # FIX: Append error to history in the required Gradio Chatbot (list of tuples) format
-        history.append((message, f"Error: {e}"))
+        history.append({"role": "user", "content": message})
+        history.append({"role": "assistant", "content": f"Error: {e}"})
         return history, chat_history_ids, f"Error: {e}"
 
 
@@ -176,12 +176,16 @@ with gr.Blocks() as demo:
 
     with gr.Tabs():
 
-        # --- FULL VOICE CHAT TAB (STT -> CHAT -> TTS) ---
+        # --- FULL VOICE CHAT TAB ---
         with gr.TabItem("🗣️ Voice Assistant"):
             gr.Markdown("## Talk to the AI Assistant")
 
+            # CRITICAL FIX: type="messages"
             voice_chat_history = gr.Chatbot(
-                label="Conversation Log", elem_classes=["chatbot"], value=[]
+                label="Conversation Log",
+                elem_classes=["chatbot"],
+                value=[],
+                type="messages",
             )
             voice_chat_state = gr.State(value=None)
 
@@ -202,7 +206,6 @@ with gr.Blocks() as demo:
 
             voice_status = gr.Textbox(elem_id="status", label="Status")
 
-            # Chain the functions together
             run_btn.click(
                 fn=speech_to_text_and_chat,
                 inputs=[audio_in, voice_chat_history, voice_chat_state],
@@ -233,8 +236,12 @@ with gr.Blocks() as demo:
         with gr.TabItem("💬 Chat → Voice Output"):
             gr.Markdown("## 💬 Chat with Voice Output")
 
+            # CRITICAL FIX: type="messages"
             tts_chatbot = gr.Chatbot(
-                label="Conversation", elem_classes=["chatbot"], value=[]
+                label="Conversation",
+                elem_classes=["chatbot"],
+                value=[],
+                type="messages",
             )
             tts_msg = gr.Textbox(
                 label="Your Message",
@@ -255,9 +262,7 @@ with gr.Blocks() as demo:
                     tts_status = gr.Textbox(elem_id="status", label="Status")
 
             def chat_and_speak(message, history, chat_ids):
-                """Send message to chat and convert response to speech."""
                 # 1. Chatbot
-                # The chat_with_bot returns history in the correct tuple format
                 updated_history, updated_ids, last_response = chat_with_bot(
                     message, history, chat_ids
                 )
@@ -267,8 +272,7 @@ with gr.Blocks() as demo:
 
                 return updated_history, updated_ids, last_response, audio_path, status
 
-            # --- Submit listener for the Enter key ---
-            fn_call = tts_msg.submit(
+            tts_msg.submit(
                 fn=chat_and_speak,
                 inputs=[tts_msg, tts_chatbot, tts_chat_state],
                 outputs=[
@@ -308,8 +312,12 @@ with gr.Blocks() as demo:
         with gr.TabItem("💬 Text Chat Only"):
             gr.Markdown("## Chat with AI Assistant")
 
+            # CRITICAL FIX: type="messages"
             chatbot = gr.Chatbot(
-                label="Conversation", elem_classes=["chatbot"], value=[]
+                label="Conversation",
+                elem_classes=["chatbot"],
+                value=[],
+                type="messages",
             )
             msg = gr.Textbox(
                 label="Your Message",
@@ -321,21 +329,19 @@ with gr.Blocks() as demo:
                 submit_btn = gr.Button("Send", variant="primary")
                 clear_btn = gr.Button("Clear Chat")
 
-            # Chat functionality (uses global state, already wired for Enter key)
-            fn_call = msg.submit(
-                # Use slice [:2] to discard the third output (response text)
-                lambda message, history, chat_state: chat_with_bot(
-                    message, history, chat_state
-                )[:2],
+            # Helper to just return history/state
+            def chat_text_only(message, history, chat_state):
+                h, s, _ = chat_with_bot(message, history, chat_state)
+                return h, s
+
+            msg.submit(
+                fn=chat_text_only,
                 inputs=[msg, chatbot, global_chat_state],
                 outputs=[chatbot, global_chat_state],
             ).then(lambda: "", None, msg)
 
             submit_btn.click(
-                # Use slice [:2] to discard the third output (response text)
-                lambda message, history, chat_state: chat_with_bot(
-                    message, history, chat_state
-                )[:2],
+                fn=chat_text_only,
                 inputs=[msg, chatbot, global_chat_state],
                 outputs=[chatbot, global_chat_state],
             ).then(lambda: "", None, msg)