Update app.py

app.py

@@ -39,11 +39,27 @@ model.renderer.set_chunk_size(131072)
 model.to(device)
 
 rembg_session = rembg.new_session()
+# my_aws_access_key_id = os.getenv("ACCESS")
+# my_aws_secret_access_key = os.getenv("SECRET")
+# bedrock_runtime = boto3.client(service_name='bedrock-runtime', aws_access_key_id = my_aws_access_key_id, aws_secret_access_key = my_aws_secret_access_key, region_name='us-east-1')
+
+# def generate_image_from_text(pos_prompt):
+#     # bedrock_runtime = boto3.client(region_name = 'us-east-1', service_name='bedrock-runtime')
+#     parameters = {'text_prompts': [{'text': pos_prompt , 'weight':1},
+#                                     {'text': """Blurry, out of frame, out of focus, Detailed, dull, duplicate, bad quality, low resolution, cropped""", 'weight': -1}],
+#                   'cfg_scale': 7, 'seed': 0, 'samples': 1}
+#     request_body = json.dumps(parameters)
+#     response = bedrock_runtime.invoke_model(body=request_body,modelId = 'stability.stable-diffusion-xl-v1')
+#     response_body = json.loads(response.get('body').read())
+#     base64_image_data = base64.b64decode(response_body['artifacts'][0]['base64'])
+    
+#     return Image.open(io.BytesIO(base64_image_data))
 
 def check_input_image(input_image):
     if input_image is None:
         raise gr.Error("No image uploaded!")
 
+
 def preprocess(input_image, do_remove_background, foreground_ratio):
     def fill_background(image):
         image = np.array(image).astype(np.float32) / 255.0
@@ -62,6 +78,7 @@ def preprocess(input_image, do_remove_background, foreground_ratio):
             image = fill_background(image)
     return image
 
+
 @spaces.GPU
 def generate(image, mc_resolution, formats=["obj", "glb"]):
     scene_codes = model(image, device=device)
@@ -77,19 +94,11 @@ def generate(image, mc_resolution, formats=["obj", "glb"]):
     
     return mesh_path_obj.name, mesh_path_glb.name
 
-def run_example(text_prompt, do_remove_background, foreground_ratio, mc_resolution):
-    # Step 1: Generate the image from text prompt
-    image_pil = generate_image_from_text(text_prompt)
-    
-    # Step 2: Preprocess the image
-    preprocessed = preprocess(image_pil, do_remove_background, foreground_ratio)
-    
-    # Step 3: Generate the 3D model
-    mesh_name_obj, mesh_name_glb = generate(preprocessed, mc_resolution, ["obj", "glb"])
-    
+def run_example(image_pil):
+    preprocessed = preprocess(image_pil, False, 0.9)
+    mesh_name_obj, mesh_name_glb = generate(preprocessed, 256, ["obj", "glb"])
     return preprocessed, mesh_name_obj, mesh_name_glb
 
-
 with gr.Blocks() as demo:
     gr.Markdown(HEADER)
     with gr.Row(variant="panel"):