Update app.py

app.py

@@ -276,18 +276,16 @@ async def upload_image(file: UploadFile = File(...)):
         # Convert uploaded file to PIL Image
         image = Image.open(io.BytesIO(await file.read()))
 
-        # 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 OBJ file in the temporary directory
+        obj_file_path = step_1_generate_obj(image)
 
-            # 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"object{uuid.uuid4()}.obj"
 
-            # Upload the OBJ file to S3
-            s3_url = save_file_to_s3(obj_file_path, 'framebucket3d', obj_name)
+        # Upload the OBJ file to S3
+        s3_url = upload_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)}")