h-siyuan commited on
Commit
b7895e4
·
verified ·
1 Parent(s): 782714c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -5
app.py CHANGED
@@ -75,13 +75,13 @@ def upload_to_s3(file_name, bucket, object_name=None):
75
 
76
  try:
77
  s3.upload_file(file_name, bucket, object_name)
78
- print(f"Uploaded {file_name} to {bucket}/{object_name}.")
79
  return True
80
  except FileNotFoundError:
81
- print(f"The file {file_name} was not found.")
82
  return False
83
  except NoCredentialsError:
84
- print("Credentials not available.")
85
  return False
86
 
87
  @spaces.GPU
@@ -129,7 +129,7 @@ def run_showui(image, query, session_id):
129
  def save_and_upload_data(image_path, query, session_id, is_example_image, votes=None):
130
  """Save the data to a JSON file and upload to S3."""
131
  if is_example_image:
132
- print("Example image used. Skipping upload.")
133
  return
134
 
135
  votes = votes or {"upvotes": 0, "downvotes": 0}
@@ -153,7 +153,7 @@ def save_and_upload_data(image_path, query, session_id, is_example_image, votes=
153
  def update_vote(vote_type, session_id, is_example_image):
154
  """Update the vote count and re-upload the JSON file."""
155
  if is_example_image:
156
- print("Example image used. Skipping vote update.")
157
  return "Example image used. No vote recorded."
158
 
159
  local_file_name = f"{session_id}.json"
@@ -250,6 +250,7 @@ def build_demo(embed_mode, concurrency_count=1):
250
  raise ValueError("No image provided. Please upload an image before submitting.")
251
 
252
  session_id = datetime.now().strftime("%Y%m%d_%H%M%S")
 
253
  is_example_image = isinstance(image, str) and image.startswith("./examples/")
254
 
255
  result_image, click_coords, image_path = run_showui(image, query, session_id)
 
75
 
76
  try:
77
  s3.upload_file(file_name, bucket, object_name)
78
+ # print(f"Uploaded {file_name} to {bucket}/{object_name}.")
79
  return True
80
  except FileNotFoundError:
81
+ # print(f"The file {file_name} was not found.")
82
  return False
83
  except NoCredentialsError:
84
+ # print("Credentials not available.")
85
  return False
86
 
87
  @spaces.GPU
 
129
  def save_and_upload_data(image_path, query, session_id, is_example_image, votes=None):
130
  """Save the data to a JSON file and upload to S3."""
131
  if is_example_image:
132
+ # print("Example image used. Skipping upload.")
133
  return
134
 
135
  votes = votes or {"upvotes": 0, "downvotes": 0}
 
153
  def update_vote(vote_type, session_id, is_example_image):
154
  """Update the vote count and re-upload the JSON file."""
155
  if is_example_image:
156
+ # print("Example image used. Skipping vote update.")
157
  return "Example image used. No vote recorded."
158
 
159
  local_file_name = f"{session_id}.json"
 
250
  raise ValueError("No image provided. Please upload an image before submitting.")
251
 
252
  session_id = datetime.now().strftime("%Y%m%d_%H%M%S")
253
+ print("image_path:", image)
254
  is_example_image = isinstance(image, str) and image.startswith("./examples/")
255
 
256
  result_image, click_coords, image_path = run_showui(image, query, session_id)