Update app.py

app.py

@@ -20,6 +20,7 @@ from PIL import Image
 from functools import partial
 import io
 from io import BytesIO
+from botocore.exceptions import NoCredentialsError, PartialCredentialsError
 
 app = FastAPI()
 
@@ -187,11 +188,37 @@ def generate(image, mc_resolution, formats=["obj", "glb"]):
     
     return mesh_path_obj.name, mesh_path_glb.name
 
+def upload_file_to_s3(file_path, bucket_name, object_name=None):
+    s3_client = boto3.client('s3',aws_access_key_id = ACCESS, aws_secret_access_key = SECRET, region_name='us-east-1')
+    
+    if object_name is None:
+        object_name = file_path
+    
+    try:
+        s3_client.upload_file(file_path, bucket_name, object_name)
+    except FileNotFoundError:
+        print(f"The file {file_path} was not found.")
+        return False
+    except NoCredentialsError:
+        print("Credentials not available.")
+        return False
+    except PartialCredentialsError:
+        print("Incomplete credentials provided.")
+        return False
+    except Exception as e:
+        print(f"An error occurred: {e}")
+        return False
+    
+    print(f"File {file_path} uploaded successfully to {bucket_name}/{object_name}.")
+    return True
+
+    
+
 @app.post("/process_image/")
 async def process_image(
     file: UploadFile = File(...),
     seed: int = Form(...),
-    use_image: bool = Form(...),
+    enhance_image: bool = Form(...),
     do_remove_background: bool = Form(...),
     foreground_ratio: float = Form(...),
     mc_resolution: int = Form(...),
@@ -200,7 +227,7 @@ async def process_image(
     image_bytes = await file.read()
     input_image = Image.open(BytesIO(image_bytes))
 
-    if use_image:
+    if enhance_image:
         image_pil = generate_image_from_text(encoded_image=input_image, seed=seed, pos_prompt=text_prompt)
     else:
         image_pil = input_image
@@ -209,8 +236,8 @@ async def process_image(
     mesh_name_obj, mesh_name_glb = generate(preprocessed, mc_resolution)
 
     return {
-        "obj_path": mesh_name_obj,
-        "glb_path": mesh_name_glb
+        "obj_path": upload_file_to_s3(mesh_name_obj, 'framebucket3d')
+        "glb_path": upload_file_to_s3(mesh_name_glb, 'framebucket3d')
     }
 
 if __name__ == "__main__":