Update app.py

app.py

@@ -140,9 +140,6 @@ def run_showui(image, query):
 # Modify the record_vote function
 def record_vote(vote_type, image_path, query, action_generated, session_id):
     """Record a vote in a JSON file and upload to S3."""
-    if image_path is None:
-        return "No image uploaded. Please upload an image before voting."
-
     vote_data = {
         "vote_type": vote_type,
         "image_path": image_path,
@@ -162,10 +159,12 @@ def record_vote(vote_type, image_path, query, action_generated, session_id):
 
     return f"Your {vote_type} has been recorded. Thank you!"
 
+# Use session_id in the handle_vote function
 def handle_vote(vote_type, image_path, query, action_generated, session_id):
     """Handle vote recording by using the consistent image path."""
-    message = record_vote(vote_type, image_path, query, action_generated, session_id)
-    return message
+    if image_path is None:
+        return "No image uploaded. Please upload an image before voting."
+    return record_vote(vote_type, image_path, query, action_generated, session_id)
 
 # Load logo and encode to Base64
 with open("./assets/showui.png", "rb") as image_file:
@@ -298,31 +297,31 @@ def build_demo(embed_mode, concurrency_count=1):
                 [output_img, output_coords],
             )
 
-            # Define button actions
+            # Record vote actions without feedback messages
             vote_btn.click(
                 lambda image_path, query, action_generated, session_id: handle_vote(
                     "upvote", image_path, query, action_generated, session_id
                 ),
                 inputs=[state_image_path, textbox, output_coords, state_session_id],
-                outputs=gr.Textbox(),  # Output the message to a Textbox or another suitable component
+                outputs=[],
                 queue=False
             )
-            
+
             downvote_btn.click(
                 lambda image_path, query, action_generated, session_id: handle_vote(
                     "downvote", image_path, query, action_generated, session_id
                 ),
                 inputs=[state_image_path, textbox, output_coords, state_session_id],
-                outputs=gr.Textbox(),  # Output the message to a Textbox or another suitable component
+                outputs=[],
                 queue=False
             )
-            
+
             flag_btn.click(
                 lambda image_path, query, action_generated, session_id: handle_vote(
                     "flag", image_path, query, action_generated, session_id
                 ),
                 inputs=[state_image_path, textbox, output_coords, state_session_id],
-                outputs=gr.Textbox(),  # Output the message to a Textbox or another suitable component
+                outputs=[],
                 queue=False
             )