Update app.py

app.py

@@ -44,10 +44,26 @@ def respond(
         yield f"Error: {str(e)}"
 
 def format_response(response: str) -> str:
-    """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
+    """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
 
 # Custom CSS for styling
 css = """
@@ -81,8 +97,8 @@ details[open] summary:after {
 
 # Create Gradio interface
 with gr.Blocks(css=css) as demo:
-    gr.Markdown("# Chat with Superthoughts lite! (1.7B)")
-    gr.Markdown("**Warning:** The first output from the AI may take a few moments. After the first message, it should work at a decent speed, keep in mind that this chat is only meant for testing and experimenting.")
+    gr.Markdown("## Chat with Superthoughts 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...")
@@ -139,9 +155,18 @@ with gr.Blocks(css=css) as demo:
         chatbot
     )
 
-    # Add a clear button
-    clear = gr.Button("Clear Conversation")
-    clear.click(lambda: None, None, chatbot, queue=False)
+    with gr.Row():
+        clear = gr.Button("Clear Conversation")
+        stop = gr.Button("Stop Generation")
+    
+    # Add disclaimer
+    gr.Markdown(
+        """
+        ---
+        ⚠️ **Disclaimer:** Superthoughts may make mistakes. Always verify important information. 
+        This chat interface is intended for testing and experimentation purposes only.
+        """
+    )
 
 # Launch the interface
 if __name__ == "__main__":