Update app.py

app.py

@@ -380,11 +380,13 @@ def visualize_all_bbox_together(image, generation):
     return pil_image, generation_colored
 
 
-def gradio_reset(chat_state, img_list):
+def gradio_reset(chat_state, img_list, path_list):
     if chat_state is not None:
         chat_state.messages = []
     if img_list is not None:
         img_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
 
@@ -481,7 +483,7 @@ def gradio_stream_answer(chatbot, chat_state, img_list, temperature):
     return chatbot, chat_state
 
 
-def gradio_visualize(chatbot, gr_img):
+def gradio_visualize(chatbot, gr_img, path_list):
     if isinstance(gr_img, dict):
         gr_img, mask = gr_img['image'], gr_img['mask']
 
@@ -491,6 +493,7 @@ def gradio_visualize(chatbot, gr_img):
         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
@@ -579,6 +582,7 @@ with gr.Blocks() as demo:
 
     upload_flag = gr.State(value=0)
     replace_flag = gr.State(value=0)
+    path_list = gr.State(value=0)
     image.upload(image_upload_trigger, [upload_flag, replace_flag, img_list], [upload_flag, replace_flag])
 
     with gr.Row():
@@ -623,8 +627,8 @@ with gr.Blocks() as demo:
         [chatbot, chat_state]
     ).success(
         gradio_visualize,
-        [chatbot, image],
-        [chatbot],
+        [chatbot, image, path_list],
+        [chatbot, path_list],
         queue=False,
     )
 
@@ -638,10 +642,10 @@ with gr.Blocks() as demo:
         [chatbot, chat_state]
     ).success(
         gradio_visualize,
-        [chatbot, image],
-        [chatbot],
+        [chatbot, image, path_list],
+        [chatbot, path_list],
     )
 
-    clear.click(gradio_reset, [chat_state, img_list], [chatbot, image, text_input, chat_state, img_list], queue=False)
+    clear.click(gradio_reset, [chat_state, img_list, path_list], [chatbot, image, text_input, chat_state, img_list], queue=False)
 
 demo.launch(enable_queue=True)