Update app.py

app.py

@@ -11,8 +11,6 @@ import ast
 import os
 import numpy as np
 from huggingface_hub import hf_hub_download, list_repo_files
-import boto3
-from botocore.exceptions import NoCredentialsError
 
 # Define constants
 DESCRIPTION = "[ShowUI Demo](https://huggingface.co/showlab/ShowUI-2B)"
@@ -66,22 +64,6 @@ def array_to_image_path(image_array, session_id):
     img.save(filename)
     return os.path.abspath(filename)
 
-def upload_to_s3(file_name, bucket, object_name=None):
-    """Upload a file to an S3 bucket."""
-    if object_name is None:
-        object_name = file_name
-
-    s3 = boto3.client('s3')
-
-    try:
-        s3.upload_file(file_name, bucket, object_name)
-        # print(f"successfully upload {file_name}")
-        return True
-    except FileNotFoundError:
-        return False
-    except NoCredentialsError:
-        return False
-
 def crop_image(image_path, click_xy, crop_factor=0.5):
     """Crop the image around the click point."""
     image = Image.open(image_path)
@@ -175,10 +157,6 @@ def save_and_upload_data(image, query, session_id, is_example_image, votes=None)
     
     with open(local_file_name, "w") as f:
         json.dump(data, f)
-    
-    # Upload JSON and image to S3
-    upload_to_s3(local_file_name, 'altair.storage', object_name=f"ootb/{local_file_name}")
-    upload_to_s3(image_file_name, 'altair.storage', object_name=f"ootb/{image_file_name}")
 
     return data
 
@@ -199,8 +177,6 @@ def update_vote(vote_type, session_id, is_example_image):
     
     with open(local_file_name, "w") as f:
         json.dump(data, f)
-    
-    upload_to_s3(local_file_name, 'altair.storage', object_name=f"ootb/{local_file_name}")
 
     return f"Thank you for your {vote_type}!"