Changed to iclight

app.py

@@ -464,20 +464,49 @@ def generate_ai_bg(input_img, prompt):
     # input_img = resize_image(input_img, 0.01)
     hf_input_img = encode_image(input_img)
 
-    handler = fal_client.submit(
-        "fal-ai/iclight-v2",
-        arguments={
-            "prompt": prompt,
-            "image_url": hf_input_img
-        },
-        webhook_url="https://optional.webhook.url/for/results",
+    # handler = fal_client.submit(
+    #     "fal-ai/iclight-v2",
+    #     arguments={
+    #         "prompt": prompt,
+    #         "image_url": hf_input_img
+    #     },
+    #     webhook_url="https://optional.webhook.url/for/results",
+    # )
+
+    # request_id = handler.request_id
+
+    # status = fal_client.status("fal-ai/iclight-v2", request_id, with_logs=True)
+
+    # result = fal_client.result("fal-ai/iclight-v2", request_id)
+
+    # relight_img_path = result['images'][0]['url']
+
+    # response = requests.get(relight_img_path, stream=True)
+
+    # relight_img = Image.open(BytesIO(response.content)).convert("RGBA")
+
+    from gradio_client import Client, handle_file
+
+    client = Client("lllyasviel/iclight-v2-vary")
+    result = client.predict(
+            input_fg=handle_file(hf_input_img),
+            bg_source="None",
+            prompt=prompt,
+            image_width=896,
+            image_height=1152,
+            num_samples=1,
+            seed=12345,
+            steps=25,
+            n_prompt="lowres, bad anatomy, bad hands, cropped, worst quality",
+            cfg=2,
+            gs=5,
+            enable_hr_fix=True,
+            hr_downscale=0.5,
+            lowres_denoise=0.8,
+            highres_denoise=0.99,
+            api_name="/process"
     )
-
-    request_id = handler.request_id
-
-    status = fal_client.status("fal-ai/iclight-v2", request_id, with_logs=True)
-
-    result = fal_client.result("fal-ai/iclight-v2", request_id)
+    print(result)
 
     relight_img_path = result['images'][0]['url']