Enable thought demo mode

ui.py

@@ -140,24 +140,24 @@ def create_demo():
                             label="Disable agentic reasoning",
                             info="Use MedSwin model alone without agentic reasoning, RAG, or web search"
                         )
-                        # show_agentic_thought = gr.Button(
-                        #     "Show agentic thought",
-                        #     size="sm"
-                        # )
+                        show_agentic_thought = gr.Button(
+                            "Show agentic thought",
+                            size="sm"
+                        )
                     enable_clinical_intake = gr.Checkbox(
                         value=True,
                         label="Enable clinical intake (max 5 Q&A)",
                         info="Ask focused follow-up questions before breaking down the case"
                     )
-                    # agentic_thoughts_box = gr.Textbox(
-                    #     label="Agentic Thoughts",
-                    #     placeholder="Internal thoughts from MedSwin and supervisor will appear here...",
-                    #     lines=8,
-                    #     max_lines=15,
-                    #     interactive=False,
-                    #     visible=False,
-                    #     elem_classes="agentic-thoughts"
-                    # )
+                    agentic_thoughts_box = gr.Textbox(
+                        label="Agentic Thoughts",
+                        placeholder="Internal thoughts from MedSwin and supervisor will appear here...",
+                        lines=8,
+                        max_lines=15,
+                        interactive=False,
+                        visible=False,
+                        elem_classes="agentic-thoughts"
+                    )
                     with gr.Row():
                         use_rag = gr.Checkbox(
                             value=False,
@@ -244,11 +244,11 @@ def create_demo():
                     new_state = not current_state
                     return gr.update(visible=new_state), new_state
                 
-                # show_agentic_thought.click(
-                #     fn=toggle_thoughts_box,
-                #     inputs=[show_thoughts_state],
-                #     outputs=[agentic_thoughts_box, show_thoughts_state]
-                # )
+                show_agentic_thought.click(
+                    fn=toggle_thoughts_box,
+                    inputs=[show_thoughts_state],
+                    outputs=[agentic_thoughts_box, show_thoughts_state]
+                )
                 
                 submit_button.click(
                     fn=stream_chat,
@@ -270,7 +270,7 @@ def create_demo():
                         disable_agentic_reasoning,
                         show_thoughts_state
                     ],
-                    outputs=[chatbot]#, agentic_thoughts_box]
+                    outputs=[chatbot, agentic_thoughts_box]
                 )
                 
                 message_input.submit(
@@ -293,7 +293,7 @@ def create_demo():
                         disable_agentic_reasoning,
                         show_thoughts_state
                     ],
-                    outputs=[chatbot]#, agentic_thoughts_box]
+                    outputs=[chatbot, agentic_thoughts_box]
                 )
     
     return demo