Update app.py

app.py

@@ -29,6 +29,8 @@ import uuid
 from fastapi import FastAPI, File, UploadFile, HTTPException
 from fastapi.responses import JSONResponse
 import datetime
+import tempfile
+
 
 app = FastAPI()
 
@@ -274,16 +276,18 @@ async def upload_image(file: UploadFile = File(...)):
         # Convert uploaded file to PIL Image
         image = Image.open(io.BytesIO(await file.read()))
 
-        # Generate OBJ file
-        obj_file_path = step_1_generate_obj(image)
+        # Create a temporary directory to store the OBJ file
+        with tempfile.TemporaryDirectory() as temp_dir:
+            # Generate OBJ file in the temporary directory
+            obj_file_path = step_1_generate_obj(image, temp_dir)
 
-        # Generate a unique name for the S3 object
-        obj_name = f"{uuid.uuid4()}.obj"
+            # Generate a unique name for the S3 object
+            obj_name = f"{uuid.uuid4()}.obj"
 
-        # Upload the OBJ file to S3
-        s3_url = upload_file_to_s3(obj_file_path, "framebucket3d", obj_name)
+            # Upload the OBJ file to S3
+            s3_url = save_file_to_s3(obj_file_path, 'framebucket3d', obj_name)
 
-        return JSONResponse(content={"Output": s3_url})
+            return JSONResponse(content={"Output": s3_url})
 
     except Exception as e:
         raise HTTPException(status_code=500, detail=f"Error processing the image: {str(e)}")