Update app.py

app.py

@@ -250,6 +250,7 @@ We currently only support one image at the start of the context! Please start a
           time.sleep(0.02)
           yield generated_text_without_prompt
     else:
+        global_think_mode = False
         ####################################################### thinking #######################################################
         generation_config = dict(max_new_tokens= 2000, do_sample=False, num_beams = 2, repetition_penalty=2.5)
     
@@ -410,6 +411,9 @@ def toggle_think_mode(current_state):
     print("global_think_mode: ",global_think_mode,"="*20)
     button_label = "🧠DeepThink💡1minute⏳" if global_think_mode else "🧠Think"
     return new_state, button_label
+
+def reset_think_mode():
+    return False, "🧠Think"  # Trả về trạng thái mặc định
     
 demo = gr.Blocks(css=CSS,js=js, theme='NoCrypt/miku')
 # demo = gr.Blocks( theme='NoCrypt/miku')
@@ -438,10 +442,7 @@ with demo:
     
     # Khi nhấn nút, trạng thái think_mode thay đổi + đổi nhãn nút
     think_button.click(toggle_think_mode, inputs=[think_mode], outputs=[think_mode, think_button])
-    # chat_demo_interface.submit(
-    #     after_response,
-    #     inputs=[think_mode],
-    #     outputs=[think_mode, think_button]
-    # )
+    # Reset nút Think sau khi chat hoàn tất
+    chat_demo_interface.submit(reset_think_mode, inputs=[], outputs=[think_mode, think_button])
 
 demo.queue().launch()