Update app.py

app.py

@@ -37,10 +37,10 @@ def _get_args():
     args = parser.parse_args()
     return args
 
-def handle_image_submission(_chatbot, task_history, file, uploaded_file_dir):
+def handle_image_submission(_chatbot, task_history, file):
     if file is None:
         return _chatbot, task_history  
-    file_path = save_image(file, uploaded_file_dir)
+    file_path = save_image(file, uploaded_file_dir)  # Renamed for clarity
     history_item = ((file_path,), None)
     _chatbot.append(history_item)
     task_history.append(history_item)
@@ -143,15 +143,15 @@ def _launch_demo(args, model, tokenizer):
         if 'image' in query[0]:
             image = tokenizer.draw_bbox_on_latest_picture(response)
             if image is not None:
-                image_path = save_image(image)
+                image_path = save_generated_image(image, uploaded_file_dir)  # Renamed for clarity
                 _chatbot[-1] = (chat_query, (image_path,))
             else:
                 _chatbot[-1] = (chat_query, "No image to display.")
         else:
             _chatbot[-1] = (chat_query, response)
-    
         return _chatbot
 
+
     def save_image(image, uploaded_file_dir):
         if image is None:
             return None