Update app.py

app.py

@@ -135,7 +135,7 @@ def run_showui(image, query):
 
     # Draw the point on the image
     result_image = draw_point(image_path, click_xy, radius=10)
-    return result_image, str(click_xy), session_id
+    return result_image, str(click_xy), image_path, session_id
 
 # Modify the record_vote function
 def record_vote(vote_type, image_path, query, action_generated, session_id):
@@ -259,8 +259,8 @@ def build_demo(embed_mode, concurrency_count=1):
                     raise ValueError("No image provided. Please upload an image before submitting.")
                 
                 # Generate consistent image path and store it in the state
-                result_image, click_coords, session_id = run_showui(image, query)
-                return result_image, click_coords, session_id
+                result_image, click_coords, image_path, session_id = run_showui(image, query)
+                return result_image, click_coords, image_path, session_id
 
             def on_image_upload(image):
                 """Generate a new session ID when a new image is uploaded."""
@@ -281,20 +281,20 @@ def build_demo(embed_mode, concurrency_count=1):
             submit_btn.click(
                 on_submit,
                 [imagebox, textbox],
-                [output_img, output_coords, state_session_id],
+                [output_img, output_coords, state_image_path, state_session_id],
             )
 
             clear_btn.click(
                 lambda: (None, None, None, None, None),
                 inputs=None,
-                outputs=[imagebox, textbox, output_img, output_coords, state_session_id],  # Clear all outputs
+                outputs=[imagebox, textbox, output_img, output_coords, state_image_path, state_session_id],  # Clear all outputs
                 queue=False
             )
 
             regenerate_btn.click(
                 lambda image, query, state_image_path: run_showui(image, query),
                 [imagebox, textbox, state_image_path],
-                [output_img, output_coords],
+                [output_img, output_coords, state_image_path],
             )
 
             # Record vote actions without feedback messages