update code

app.py

@@ -231,7 +231,7 @@ def add_text(state, imagebox, textbox, image_process_mode, gen_image):
     state.append_message(state.roles[0], textbox)
     state.append_message(state.roles[1], None)
 
-    yield (state, state.to_gradio_chatbot(), "", None) + (disable_btn, disable_btn, disable_btn, enable_btn, enable_btn)
+    yield (state, state.to_gradio_chatbot(), "", None, gen_image) + (disable_btn, disable_btn, disable_btn, enable_btn, enable_btn)
 
 
 def delete_text(state, image_process_mode):
@@ -328,7 +328,7 @@ with gr.Blocks(title='Mini-Gemini') as demo:
     textbox.submit(
         add_text,
         [state, imagebox, textbox, image_process_mode, gen_image],
-        [state, chatbot, textbox, imagebox] + btn_list,
+        [state, chatbot, textbox, imagebox, gen_image] + btn_list,
     ).then(
         generate,
         [state, imagebox, textbox, image_process_mode, gen_image, temperature, top_p, max_output_tokens],
@@ -337,7 +337,7 @@ with gr.Blocks(title='Mini-Gemini') as demo:
     submit_btn.click(
         add_text,
         [state, imagebox, textbox, image_process_mode, gen_image],
-        [state, chatbot, textbox, imagebox] + btn_list,
+        [state, chatbot, textbox, imagebox, gen_image] + btn_list,
     ).then(
         generate,
         [state, imagebox, textbox, image_process_mode, gen_image, temperature, top_p, max_output_tokens],