Update app.py

app.py

@@ -80,6 +80,11 @@ def delete_text(state, image_process_mode):
         prev_human_msg[1] = (*prev_human_msg[1][:2], image_process_mode)
     yield (state, state.to_gradio_chatbot(), "", None) + (disable_btn, disable_btn, disable_btn, enable_btn, enable_btn)
 
+def reset_state_on_image_change():
+    state = default_conversation.copy()
+    return (state, [], "", None, gr.update(interactive=False), gr.update(interactive=False))
+
+
 def regenerate(state, image_process_mode):
     state.messages[-1][-1] = None
     prev_human_msg = state.messages[-2]
@@ -249,11 +254,12 @@ with gr.Blocks(title="CuMo", theme=gr.themes.Default(), css=block_css) as demo:
                 regenerate_btn = gr.Button(value="🔄  Regenerate", interactive=False)
                 clear_btn = gr.Button(value="🗑️  Clear", interactive=False)
     imagebox.change(
-    fn=clear_history,
+    fn=reset_state_on_image_change,
     inputs=None,
     outputs=[state, chatbot, textbox, imagebox, regenerate_btn, clear_btn],
     queue=False
     )
+
     
     gr.Markdown(tos_markdown)
     gr.Markdown(learn_more_markdown)