Update app.py

app.py

@@ -274,24 +274,34 @@ def upload_file_to_s3(file_path, bucket_name, object_name):
 
 @app.post("/upload/")
 async def upload_image(file: UploadFile = File(...)):
-    try:
-        # Convert uploaded file to PIL Image
-        image = Image.open(io.BytesIO(await file.read()))
-
-        # Generate OBJ file in the temporary directory
-        obj_file_path = step_1_generate_obj(image)
-
-        # Generate a unique name for the S3 object
-        timestamp = int(time.time()) 
-        obj_name = f"object{timestamp}.obj"
-
-        # Upload the OBJ file to S3
-        s3_url = upload_file_to_s3(obj_file_path, 'framebucket3d', obj_name)
+    image_bytes = await file.read()
+    img_input = Image.open(BytesIO(image_bytes))
+    model_output = LitModel3D(
+                clear_color=[0.1, 0.1, 0.1, 0],  # can adjust background color for better contrast
+                label="3D Model Visualization",
+                scale=1.0,
+                tonemapping="aces",  # can use aces tonemapping for more realistic lighting
+                exposure=1.0,        # can adjust exposure to control brightness
+                contrast=1.1,        # can slightly increase contrast for better depth
+                camera_position=(0, 0, 2),  # will set initial camera position to center the model
+                zoom_speed=0.5,      # will adjust zoom speed for better control
+                pan_speed=0.5,       # will adjust pan speed for better control
+                interactive=True     # this allow users to interact with the model
+            )
+
+    obj_file_output, model_output = step_1_generate_obj(img_input)
+
+    timestamp = datetime.datetime.now().astimezone().strftime('%Y%m%d%H%M%S%f%z')
+    object_name = f'object_{timestamp}.obj'
+
+    if upload_file_to_s3(obj_file_output, 'framebucket3d',object_name):
+
+            return {
+                "obj_path": f"https://framebucket3d.s3.amazonaws.com/{object_name}"
+            }
 
-        return JSONResponse(content={"Output": s3_url})
-
-    except Exception as e:
-        raise HTTPException(status_code=500, detail=f"Error processing the image: {str(e)}")
+    
+    
 
 # Run FastAPI
 if __name__ == "__main__":