Update app.py

app.py

@@ -21,6 +21,8 @@ from functools import partial
 import io
 from io import BytesIO
 from botocore.exceptions import NoCredentialsError, PartialCredentialsError
+import datetime
+
 
 
 app = FastAPI()
@@ -241,10 +243,19 @@ async def process_image(
 
     preprocessed = preprocess(image_pil, do_remove_background, foreground_ratio)
     mesh_name_obj, mesh_name_glb = generate(preprocessed, mc_resolution)
+    timestamp = datetime.datetime.now().strftime('%Y%m%d%H%M%S%f')
+    object_name = f'object_{timestamp}_1.obj'
+    object_name_2 = f'object_{timestamp}_2.glb'
+
+    if upload_file_to_s3(mesh_name_obj, 'framebucket3d',object_name) and upload_file_to_s3(mesh_name_glb, 'framebucket3d',object_name_2):
 
-    return {
-        "obj_path": upload_file_to_s3(mesh_name_obj, 'framebucket3d')
-    }
+        return {
+            "obj_path": f"https://framebucket3d.s3.amazonaws.com//tmp/{object_name}",
+            "glb_path": f"https://framebucket3d.s3.amazonaws.com//tmp/{object_name_2}"
+        }
+
+    else:
+        return {"Internal Server Error": False}
 
 if __name__ == "__main__":
     import uvicorn