Update app.py

app.py

@@ -434,26 +434,23 @@ css = '''
 button{min-height: 60px}
 '''
 def create_chatbot_demo():
-    with gr.Blocks(css=css, theme=gr.themes.Soft()) as demo:
+    with gr.Blocks(css=css) as demo:
         gr.Markdown("# Dream 7B - Diffusion Language Model Demo")
         gr.Markdown(
             "[[Model Card](https://huggingface.co/Dream-org/Dream-v0-Instruct-7B)] "
             "[[Blog](https://hkunlp.github.io/blog/2025/dream/)]"
         )
 
-        # Use Chatbot directly for state, matching the expected format
-        chatbot_ui = gr.Chatbot(
-            label="Conversation",
-            height=500,
-            show_copy_button=True,
-            bubble_full_width=False,
-            value=[], # Initialize empty
-            type="messages" # Crucial: Use the messages format
-         )
-
         with gr.Row():
             with gr.Column(scale=3):
-                # Chatbot moved above this row
+                chatbot_ui = gr.Chatbot(
+                    label="Conversation",
+                    height=500,
+                    show_copy_button=True,
+                    bubble_full_width=False,
+                    value=[], # Initialize empty
+                    type="messages" # Crucial: Use the messages format
+                )
                 with gr.Group():
                     with gr.Row():
                         user_input = gr.Textbox(
@@ -468,11 +465,11 @@ def create_chatbot_demo():
                 )
             with gr.Column(scale=2):
                 output_vis = gr.HighlightedText(
-                    label="Denoising Process Visualization", combine_adjacent=True,
-                    show_legend=False, interactive=False
+                    label="Denoising Process Visualization", combine_adjacent=False,
+                    show_legend=True, interactive=False
                 )
                 response_text_display = gr.Textbox(
-                    label="Current/Final Response", interactive=False, lines=5
+                    label="Current/Final Response", interactive=False, lines=5, visible=False
                 )
 
         # [Keep Accordion with Generation Settings the same]
@@ -489,7 +486,7 @@ def create_chatbot_demo():
              with gr.Row():
                  remasking_strategy = gr.Radio(choices=['origin', 'maskgit_plus', 'topk_margin', 'entropy'], value='entropy', label="Remasking Strategy")
              with gr.Row():
-                visualization_delay = gr.Slider(minimum=0.0, maximum=0.5, value=0.03, step=0.01, label="Visualization Delay (s)")
+                visualization_delay = gr.Slider(minimum=0.0, maximum=0.5, value=0.0, step=0.01, label="Visualization Delay (s)")
 
         clear_btn = gr.Button("Clear Conversation")