add image from url

app.py

@@ -16,7 +16,10 @@ from gradio_imageslider import ImageSlider
 from diffusers import FlowMatchEulerDiscreteScheduler, AutoencoderKL
 from diffusers.models.transformers.transformer_flux import FluxTransformer2DModel
 from transformers import CLIPTextModel, CLIPTokenizer,T5EncoderModel, T5TokenizerFast
-
+import requests
+from io import BytesIO
+import PIL.Image
+import requests
 
 MARKDOWN = """
 # FLUX.1 Inpainting with lora
@@ -107,6 +110,8 @@ def resize_image_dimensions(
 @spaces.GPU(duration=100)
 def process(
     input_image_editor: dict,
+    image_url: str,
+    mask_url: str,
     lora_path: str,
     lora_weights: str,
     lora_scale: float,
@@ -122,9 +127,24 @@ def process(
         gr.Info("Please enter a text prompt.")
         return None, None
 
+    # default image edtiro
     image = input_image_editor['background']
     mask = input_image_editor['layers'][0]
 
+    if image_url:
+        print("start to fetch image from url", image_url)
+        response = requests.get(image_url)
+        response.raise_for_status()
+        image = PIL.Image.open(BytesIO(response.content))
+        print("fetch image success")
+    
+    if mask_url:
+        print("start to fetch mask from url", mask_url)
+        response = requests.get(mask_url)
+        response.raise_for_status()
+        mask = PIL.Image.open(BytesIO(response.content))
+        print("fetch mask success")
+
     if not image:
         gr.Info("Please upload an image.")
         return None, None
@@ -173,7 +193,19 @@ with gr.Blocks() as demo:
                 image_mode='RGB',
                 layers=False,
                 brush=gr.Brush(colors=["#FFFFFF"], color_mode="fixed"))
-            
+                
+            image_url =  gr.Textbox(
+                    label="image url",
+                    show_label=True,
+                    max_lines=1,
+                    placeholder="Enter your image url (Optional)",
+                )
+            mask_url =  gr.Textbox(
+                    label="Mask image url",
+                    show_label=True,
+                    max_lines=1,
+                    placeholder="Enter your mask image url (Optional)",
+                )
         
             with gr.Accordion("Prompt Settings", open=True):
 
@@ -266,6 +298,8 @@ with gr.Blocks() as demo:
         fn=process,
         inputs=[
             input_image_editor_component,
+            image_url,
+            mask_url,
             lora_path,
             lora_weights,
             lora_scale,

requirements.txt

@@ -6,4 +6,5 @@ sentencepiece
 git+https://github.com/Gothos/diffusers.git@flux-inpaint
 huggingface_hub
 peft
-gradio_imageslider
+gradio_imageslider
+requests