Update app.py

app.py

@@ -233,7 +233,8 @@ 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:
-        generation_config = dict(max_new_tokens= 2000, do_sample=False, num_beams = 3, repetition_penalty=2.0)
+        ####################################################### thinking #######################################################
+        generation_config = dict(max_new_tokens= 2000, do_sample=False, num_beams = 2, repetition_penalty=2.0)
     
         if len(history) == 0:
             if pixel_values is not None:
@@ -273,7 +274,7 @@ We currently only support one image at the start of the context! Please start a
         for char in thinking:
             temp_text += char
             yield accumulated_text + temp_text + "\n</code></pre>\n\n---\n"
-            time.sleep(0.02)
+            time.sleep(0.002)
 
         accumulated_text += temp_text + "\n</code></pre>\n\n---\n"
 
@@ -284,7 +285,7 @@ We currently only support one image at the start of the context! Please start a
         for char in conclusion_part:
             temp_text += char
             yield accumulated_text + temp_text + "\n\n---\n"
-            time.sleep(0.02)
+            time.sleep(0.01)
         
         accumulated_text += temp_text + "\n\n---\n"
 
@@ -313,9 +314,9 @@ CSS ="""
     border-style: solid;
     overflow: hidden;
     flex-grow: 7;
-    min-width: min(160px, 100%);
+    min-width: min(160px, 100%); 
     border-width: var(--block-border-width);
-    height: 100px !important;
+    height: 10dvh !important;
 }
 
 #think-button {
@@ -383,13 +384,21 @@ function forceLightTheme() {
 }
 """
 
-# Hàm bật/tắt Think Mode
+# def toggle_think_mode(current_state):
+#     global global_think_mode
+#     new_state = not current_state
+#     global_think_mode = not global_think_mode
+#     button_label = "🧠Think💡" if new_state else "🧠Think"
+#     return new_state, button_label
+
 def toggle_think_mode(current_state):
     global global_think_mode
     new_state = not current_state
-    global_think_mode = not global_think_mode
+    global_think_mode = new_state
     button_label = "🧠Think💡" if new_state else "🧠Think"
-    return new_state, button_label
+    # Khi Think Mode bật, đổi nền chat thành đỏ, khi tắt quay về mặc định
+    chat_bg_color = "#ffcccc" if new_state else "#ffffff"
+    return new_state, button_label, gr.update(css=f"body {{ background-color: {chat_bg_color}; }}")
     
 demo = gr.Blocks(css=CSS,js=js, theme='NoCrypt/miku')