Update app.py

app.py

@@ -75,13 +75,10 @@ def upload_to_s3(file_name, bucket, object_name=None):
 
     try:
         s3.upload_file(file_name, bucket, object_name)
-        # print(f"Uploaded {file_name} to {bucket}/{object_name}.")
         return True
     except FileNotFoundError:
-        # print(f"The file {file_name} was not found.")
         return False
     except NoCredentialsError:
-        # print("Credentials not available.")
         return False
 
 @spaces.GPU
@@ -129,7 +126,6 @@ def run_showui(image, query, session_id):
 def save_and_upload_data(image_path, query, session_id, is_example_image, votes=None):
     """Save the data to a JSON file and upload to S3."""
     if is_example_image:
-        # print("Example image used. Skipping upload.")
         return
 
     votes = votes or {"upvotes": 0, "downvotes": 0}
@@ -153,7 +149,6 @@ def save_and_upload_data(image_path, query, session_id, is_example_image, votes=
 def update_vote(vote_type, session_id, is_example_image):
     """Update the vote count and re-upload the JSON file."""
     if is_example_image:
-        # print("Example image used. Skipping vote update.")
         return "Example image used. No vote recorded."
 
     local_file_name = f"{session_id}.json"
@@ -227,9 +222,10 @@ def build_demo(embed_mode, concurrency_count=1):
                     ["./examples/safari_google.png", "Click on search bar."],
                 ]
 
-                def set_example_flag(image_path, query):
+                def set_example_flag(image, query):
+                    image_path = image if isinstance(image, str) else None
                     is_example = any(image_path == example[0] for example in examples)
-                    return image_path, query, is_example
+                    return is_example
 
                 gr.Examples(
                     examples=examples,