Update app.py

app.py

@@ -28,7 +28,6 @@ from minigpt4.runners import *
 from minigpt4.tasks import *
 
 import warnings
-
 warnings.filterwarnings("ignore")
 
 def parse_args():
@@ -383,14 +382,11 @@ def visualize_all_bbox_together(image, generation):
     return pil_image, generation_colored
 
 
-def gradio_reset(chat_state, img_list, path_list):
+def gradio_reset(chat_state, img_list):
     if chat_state is not None:
         chat_state.messages = []
     if img_list is not None:
         img_list = []
-    if isinstance(path_list, list):
-        for path in path_list:
-            os.remove(path)
     return None, gr.update(value=None, interactive=True), gr.update(placeholder='Upload your image and chat',
                                                                     interactive=True), chat_state, img_list
 
@@ -414,11 +410,12 @@ def example_trigger(text_input, image, upload_flag, replace_flag, img_list):
     return upload_flag, replace_flag
 
 
-def gradio_ask(user_message, chatbot, chat_state, gr_img, img_list, upload_flag, replace_flag, path_list):
+def gradio_ask(user_message, chatbot, chat_state, gr_img, img_list, upload_flag, replace_flag):
     if len(user_message) == 0:
         text_box_show = 'Input should not be empty!'
     else:
         text_box_show = ''
+
     if isinstance(gr_img, dict):
         gr_img, mask = gr_img['image'], gr_img['mask']
     else:
@@ -451,10 +448,9 @@ def gradio_ask(user_message, chatbot, chat_state, gr_img, img_list, upload_flag,
         visual_img, _ = visualize_all_bbox_together(gr_img, user_message)
         if visual_img is not None:
             file_path = save_tmp_img(visual_img)
-            path_list.append(file_path)
             chatbot = chatbot + [[(file_path,), None]]
 
-    return text_box_show, chatbot, chat_state, img_list, upload_flag, replace_flag, path_list
+    return text_box_show, chatbot, chat_state, img_list, upload_flag, replace_flag
 
 
 def gradio_answer(chatbot, chat_state, img_list, temperature):
@@ -487,16 +483,16 @@ def gradio_stream_answer(chatbot, chat_state, img_list, temperature):
     return chatbot, chat_state
 
 
-def gradio_visualize(chatbot, gr_img, path_list):
+def gradio_visualize(chatbot, gr_img):
     if isinstance(gr_img, dict):
         gr_img, mask = gr_img['image'], gr_img['mask']
+
     unescaped = reverse_escape(chatbot[-1][1])
     visual_img, generation_color = visualize_all_bbox_together(gr_img, unescaped)
     if visual_img is not None:
         if len(generation_color):
             chatbot[-1][1] = generation_color
         file_path = save_tmp_img(visual_img)
-        path_list.append(file_path)
         chatbot = chatbot + [[None, (file_path,)]]
 
     return chatbot
@@ -585,7 +581,6 @@ with gr.Blocks() as demo:
 
     upload_flag = gr.State(value=0)
     replace_flag = gr.State(value=0)
-    path_list = gr.State(value=[])
     image.upload(image_upload_trigger, [upload_flag, replace_flag, img_list], [upload_flag, replace_flag])
 
     with gr.Row():
@@ -622,33 +617,33 @@ with gr.Blocks() as demo:
 
     text_input.submit(
         gradio_ask,
-        [text_input, chatbot, chat_state, image, img_list, upload_flag, replace_flag, path_list],
-        [text_input, chatbot, chat_state, img_list, upload_flag, replace_flag, path_list], queue=False
+        [text_input, chatbot, chat_state, image, img_list, upload_flag, replace_flag],
+        [text_input, chatbot, chat_state, img_list, upload_flag, replace_flag], queue=False
     ).success(
         gradio_stream_answer,
         [chatbot, chat_state, img_list, temperature],
         [chatbot, chat_state]
     ).success(
         gradio_visualize,
-        [chatbot, image, path_list],
-        [chatbot, path_list],
+        [chatbot, image],
+        [chatbot],
         queue=False,
     )
 
     send.click(
         gradio_ask,
-        [text_input, chatbot, chat_state, image, img_list, upload_flag, replace_flag, path_list],
-        [text_input, chatbot, chat_state, img_list, upload_flag, replace_flag, path_list]
+        [text_input, chatbot, chat_state, image, img_list, upload_flag, replace_flag],
+        [text_input, chatbot, chat_state, img_list, upload_flag, replace_flag]
     ).success(
         gradio_stream_answer,
         [chatbot, chat_state, img_list, temperature],
         [chatbot, chat_state]
     ).success(
         gradio_visualize,
-        [chatbot, image, path_list],
-        [chatbot, path_list],
+        [chatbot, image],
+        [chatbot],
     )
 
-    clear.click(gradio_reset, [chat_state, img_list, path_list], [chatbot, image, text_input, chat_state, img_list], queue=False)
+    clear.click(gradio_reset, [chat_state, img_list], [chatbot, image, text_input, chat_state, img_list], queue=False)
 
 demo.launch(enable_queue=True)