Update app.py

app.py

@@ -84,7 +84,7 @@ with gr.Blocks(css=css) as demo:
     gr.Markdown("## Chat with <span style='color: #d026ff'>Superthoughts</span> lite! (1.7B)")
     gr.Markdown("**Note:** First response may take a moment to initialize. Subsequent responses will be faster.")
     
-    chatbot = gr.Chatbot(height=600)
+    chatbot = gr.Chatbot(height=600, type='messages')
     with gr.Row():
         msg = gr.Textbox(label="Your message", placeholder="Type your message here...", scale=9)
         stop = gr.Button("Stop", scale=1)
@@ -119,20 +119,17 @@ with gr.Blocks(css=css) as demo:
 
     def user(user_message: str, history: list) -> tuple[str, list]:
         """Add user message to history"""
-        return "", history + [[user_message, None]]
+        return "", history + [{"role": "user", "content": user_message}, {"role": "assistant", "content": None}]
 
     def bot(history: list, system_message: str, max_tokens: int, temperature: float, top_p: float) -> Iterator[list]:
         """Generate and stream bot responses"""
-        user_message, _ = history[-1]
-        history[-1][1] = ""  # Initialize bot's response
-        
-        for partial_response in respond(user_message, history[:-1], system_message, max_tokens, temperature, top_p):
-            history[-1][1] = partial_response
+        user_message = history[-2]["content"]  # Get the last user message
+        for partial_response in respond(user_message, history[:-2], system_message, max_tokens, temperature, top_p):
+            history[-1]["content"] = partial_response
             yield history
 
-    # Set up chat message handling and stop button
-    stop.click(fn=None, inputs=None, outputs=None, cancels=msg.submit)
-    msg.submit(
+    # Create event handlers
+    submit_event = msg.submit(
         user,
         [msg, chatbot],
         [msg, chatbot],
@@ -142,6 +139,9 @@ with gr.Blocks(css=css) as demo:
         [chatbot, system_message, max_tokens, temperature, top_p],
         chatbot
     )
+    
+    # Set up stop button
+    stop.click(fn=None, inputs=None, outputs=None, cancels=[submit_event])
 
     # Add a clear button
     clear = gr.Button("Clear Conversation")