Add image link input box

app.py

@@ -25,6 +25,9 @@ from preprocess.openpose.run_openpose import OpenPose
 from detectron2.data.detection_utils import convert_PIL_to_numpy,_apply_exif_orientation
 from torchvision.transforms.functional import to_pil_image
 import pillow_heif  # HEIC 이미지 처리용 (아이폰 촬영 사진 포맷)
+import requests
+from urllib.parse import urlparse
+
 
 def pil_to_binary_mask(pil_image, threshold=0):
     np_image = np.array(pil_image)
@@ -156,6 +159,30 @@ def preprocess_image(image):
     return resized_img
 
 
+# URL에서 이미지 가져오기 함수
+def load_image_from_url(url):
+    try:
+        response = requests.get(url, stream=True)
+        response.raise_for_status()  # HTTP 오류 확인
+        img = Image.open(response.raw).convert("RGB")
+        return img
+    except Exception as e:
+        print(f"Error loading image from URL: {e}")
+        return None
+    
+
+#URL 입력 처리 함수
+def process_url_image(url, is_garment=False):
+    if not url:
+        return None
+    img = load_image_from_url(url)
+    if img and is_garment:  # 의상 이미지의 경우 전처리 생략 가능
+        return img
+    elif img:
+        return preprocess_image(img)  # 대상 이미지는 전처리 적용
+    return None
+
+
 @spaces.GPU
 def start_tryon(dict,garm_img,garment_des,is_checked,denoise_steps,seed, is_checked_crop):
     device = "cuda"
@@ -301,6 +328,7 @@ with image_blocks as demo:
     with gr.Row():
         with gr.Column():
             imgs = gr.ImageEditor(sources='upload', type="pil", label='대상 이미지', interactive=True)
+            img_url_input = gr.Textbox(label="대상 이미지 URL", placeholder="예) https://example.com/human_image.jpg")
             with gr.Row():
                 with gr.Row():
                     is_checked = gr.Checkbox(label="Yes", info="자동 마스킹",value=True)
@@ -312,6 +340,7 @@ with image_blocks as demo:
 
         with gr.Column():
             garm_img = gr.Image(label="의상 이미지", sources='upload', type="pil")
+            garm_url_input = gr.Textbox(label="의상 이미지 URL", placeholder="예) https://example.com/garment.jpg")
             with gr.Row(elem_id="prompt-container"):
                 with gr.Row():
                     prompt = gr.Textbox(placeholder="Description of garment ex) Short Sleeve Round Neck T-shirts", show_label=False, elem_id="prompt")
@@ -344,6 +373,21 @@ with image_blocks as demo:
         outputs=imgs,  # 전처리된 이미지를 ImageEditor에 다시 표시
     )
 
+
+    # 대상 이미지: URL 입력 처리
+    img_url_input.change(
+        fn=lambda url: process_url_image(url, is_garment=False),
+        inputs=img_url_input,
+        outputs=imgs,
+    )
+
+    # 의상 이미지: URL 입력 처리
+    garm_url_input.change(
+        fn=lambda url: process_url_image(url, is_garment=True),
+        inputs=garm_url_input,
+        outputs=garm_img,
+    )
+
     is_checked_crop = True
     denoise_steps = 30
     seed = 42

requirements.txt

@@ -23,3 +23,4 @@ omegaconf
 pycocotools
 huggingface_hub==0.25.0
 pillow-heif
+requests