Update app.py

app.py

@@ -44,26 +44,10 @@ def respond(
         yield f"Error: {str(e)}"
 
 def format_response(response: str) -> str:
-    """Format the response with collapsible thinking sections that maintain state"""
-    import re
-    import hashlib
-    
-    def get_section_id(content):
-        # Create a unique ID for each thinking section based on its content
-        return hashlib.md5(content.encode()).hexdigest()[:8]
-    
-    # Find all thinking sections and replace them with uniquely identified sections
-    pattern = r"<think>(.*?)</think>"
-    sections = re.findall(pattern, response, re.DOTALL)
-    
-    formatted = response
-    for section in sections:
-        section_id = get_section_id(section)
-        old = f"<think>{section}</think>"
-        new = f'<details id="think_{section_id}" open><summary>Show thinking 🧠</summary><div class="thoughts">{section}</div></details>'
-        formatted = formatted.replace(old, new)
-    
-    return formatted
+    """Format the response with collapsible thinking sections"""
+    response = response.replace("<think>", '<details open><summary>Show thinking 🧠</summary><div class="thoughts">')
+    response = response.replace("</think>", "</div></details>")
+    return response
 
 # Custom CSS for styling
 css = """
@@ -97,11 +81,14 @@ details[open] summary:after {
 
 # Create Gradio interface
 with gr.Blocks(css=css) as demo:
-    gr.Markdown("## Chat with Superthoughts lite! (1.7B)")
+    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)
-    msg = gr.Textbox(label="Your message", placeholder="Type your message here...")
+    with gr.Row():
+        msg = gr.Textbox(label="Your message", placeholder="Type your message here...", scale=9)
+        stop = gr.Button("Stop", scale=1)
+    
     
     with gr.Accordion("Advanced Settings", open=False):
         system_message = gr.Textbox(
@@ -143,7 +130,8 @@ with gr.Blocks(css=css) as demo:
             history[-1][1] = partial_response
             yield history
 
-    # Set up chat message handling
+    # Set up chat message handling and stop button
+    stop.click(fn=None, inputs=None, outputs=None, cancels=msg.submit)
     msg.submit(
         user,
         [msg, chatbot],
@@ -155,9 +143,9 @@ with gr.Blocks(css=css) as demo:
         chatbot
     )
 
-    with gr.Row():
-        clear = gr.Button("Clear Conversation")
-        stop = gr.Button("Stop Generation")
+    # Add a clear button
+    clear = gr.Button("Clear Conversation")
+    clear.click(lambda: None, None, chatbot, queue=False)
     
     # Add disclaimer
     gr.Markdown(