vibs08 commited on
Commit
7e7ca67
·
verified ·
1 Parent(s): 3d64050

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -6
app.py CHANGED
@@ -267,10 +267,9 @@ def step_3_display_3d_model(mesh_file):
267
  return mesh_file
268
 
269
  def upload_file_to_s3(file_path, bucket_name, object_name):
270
- try:
271
- s3_client.upload_file(file_path, bucket_name, object_name)
272
- except Exception as e:
273
- raise HTTPException(status_code=500, detail=f"Failed to upload file to S3: {str(e)}")
274
 
275
 
276
  @app.post("/upload/")
@@ -292,8 +291,8 @@ async def upload_image(file: UploadFile = File(...)):
292
 
293
  obj_file_output, model_output = step_1_generate_obj(img_input)
294
 
295
- timestamp = datetime.datetime.now().astimezone().strftime('%Y%m%d%H%M%S%f%z')
296
- object_name = f'object_{timestamp}.obj'
297
 
298
  if upload_file_to_s3(obj_file_output, 'framebucket3d',object_name):
299
 
 
267
  return mesh_file
268
 
269
  def upload_file_to_s3(file_path, bucket_name, object_name):
270
+ if s3_client.upload_file(file_path, bucket_name, object_name):
271
+ return True
272
+
 
273
 
274
 
275
  @app.post("/upload/")
 
291
 
292
  obj_file_output, model_output = step_1_generate_obj(img_input)
293
 
294
+ timestamp = datetime.datetime.now().strftime('%Y%m%d%H%M%S%f')
295
+ object_name = f'frameobject_{timestamp}.obj'
296
 
297
  if upload_file_to_s3(obj_file_output, 'framebucket3d',object_name):
298