Update app.py

app.py

@@ -38,7 +38,9 @@ def _get_args():
     return args
 
 def handle_image_submission(_chatbot, task_history, file):
+    print("handle_image_submission called")
     if file is None:
+        print("No file uploaded")
         return _chatbot, task_history  
     file_path = save_image(file, uploaded_file_dir)  # Renamed for clarity
     history_item = ((file_path,), None)
@@ -146,8 +148,8 @@ 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_generated_image(image, uploaded_file_dir)  # Renamed for clarity
+                if image is not None:
+                image_path = save_image(image, uploaded_file_dir)
                 _chatbot[-1] = (chat_query, (image_path,))
             else:
                 _chatbot[-1] = (chat_query, "No image to display.")
@@ -156,7 +158,7 @@ def _launch_demo(args, model, tokenizer):
         return _chatbot
 
 
-    def save_image(image, uploaded_file_dir):
+    def save_uploaded_image(image_file, upload_dir):
         if image is None:
             return None
         temp_dir = secrets.token_hex(20)