Update app.py

app.py

@@ -140,6 +140,9 @@ 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,
@@ -159,12 +162,10 @@ 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."""
-    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)
+    message = record_vote(vote_type, image_path, query, action_generated, session_id)
+    return message
 
 # Load logo and encode to Base64
 with open("./assets/showui.png", "rb") as image_file:
@@ -297,31 +298,31 @@ def build_demo(embed_mode, concurrency_count=1):
                 [output_img, output_coords],
             )
 
-            # Record vote actions without feedback messages
+            # Define button actions
             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=[],
+                outputs=gr.Textbox(),  # Output the message to a Textbox or another suitable component
                 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=[],
+                outputs=gr.Textbox(),  # Output the message to a Textbox or another suitable component
                 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=[],
+                outputs=gr.Textbox(),  # Output the message to a Textbox or another suitable component
                 queue=False
             )