Update app.py
Browse files
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 |
-
|
271 |
-
|
272 |
-
|
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().
|
296 |
-
object_name = f'
|
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 |
|