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 |
|