Update app.py

app.py

@@ -105,6 +105,20 @@ def reset_user_input():
 def reset_state(task_history):
     task_history.clear()
     return []
+    
+def regenerate(_chatbot, task_history):
+    if not task_history:
+        return _chatbot
+    item = task_history[-1]
+    if item[1] is None:
+        return _chatbot
+    task_history[-1] = (item[0], None)
+    chatbot_item = _chatbot.pop(-1)
+    if chatbot_item[0] is None:
+        _chatbot[-1] = (_chatbot[-1][0], None)
+    else:
+        _chatbot.append((chatbot_item[0], None))
+    return predict(_chatbot, task_history)
 
 with gr.Blocks() as demo:
     gr.Markdown("<center><font size=8>Qwen-VL-Chat Bot</center>")
@@ -125,5 +139,7 @@ with gr.Blocks() as demo:
     )
     submit_btn.click(reset_user_input, [], [query])
     empty_bin.click(reset_state, [task_history], [chatbot], show_progress=True)
+    regen_btn.click(regenerate, [chatbot, task_history], [chatbot], show_progress=True)
+    addfile_btn.upload(add_file, [chatbot, task_history, addfile_btn], [chatbot, task_history], show_progress=True)
 
 demo.launch()