Support dragdrop for URLs

modules/images.py

@@ -772,18 +772,31 @@ Steps: {json_info["steps"]}, Sampler: {sampler}, CFG scale: {json_info["scale"]}
 def image_data(data):
     import gradio as gr
 
-    try:
-        image = read(io.BytesIO(data))
-        textinfo, _ = read_info_from_image(image)
-        return textinfo, None
-    except Exception:
-        pass
+    if not data:
+        return gr.update(), None
 
-    try:
-        text = data.decode('utf8')
-        assert len(text) < 10000
-        return text, None
+    if isinstance(data, bytes):
+        try:
+            image = Image.open(io.BytesIO(data))
+            textinfo, _ = read_info_from_image(image)
+            return textinfo, None
+        except Exception:
+            pass
 
+        try:
+            text = data.decode('utf8')
+            assert len(text) < 10000
+            return text, None
+        except Exception:
+            pass
+
+    import requests
+    try:
+        r = requests.get(data, timeout=5)
+        if r.status_code == 200:
+            image = Image.open(io.BytesIO(r.content))
+            textinfo, _ = read_info_from_image(image)
+            return textinfo, None
     except Exception:
         pass
 

modules/ui_toprow.py

@@ -82,6 +82,7 @@ class Toprow:
             with gr.Row(elem_id=f"{self.id_part}_prompt_row", elem_classes=["prompt-row"]):
                 self.prompt = gr.Textbox(label="Prompt", elem_id=f"{self.id_part}_prompt", show_label=False, lines=3, placeholder="Prompt\n(Press Ctrl+Enter to generate, Alt+Enter to skip, Esc to interrupt)", elem_classes=["prompt"])
                 self.prompt_img = gr.File(label="", elem_id=f"{self.id_part}_prompt_image", file_count="single", type="binary", visible=False)
+                self.prompt_url = gr.Textbox(label="", elem_id=f"{self.id_part}_prompt_url", visible=False)
 
             with gr.Row(elem_id=f"{self.id_part}_neg_prompt_row", elem_classes=["prompt-row"]):
                 self.negative_prompt = gr.Textbox(label="Negative prompt", elem_id=f"{self.id_part}_neg_prompt", show_label=False, lines=3, placeholder="Negative prompt\n(Press Ctrl+Enter to generate, Alt+Enter to skip, Esc to interrupt)", elem_classes=["prompt"])
@@ -93,6 +94,13 @@ class Toprow:
             show_progress=False,
         )
 
+        self.prompt_url.input(
+            fn=modules.images.image_data,
+            inputs=[self.prompt_url],
+            outputs=[self.prompt, self.prompt_url],
+            show_progress=False,
+        )
+
     def create_submit_box(self):
         with gr.Row(elem_id=f"{self.id_part}_generate_box", elem_classes=["generate-box"] + (["generate-box-compact"] if self.is_compact else []), render=not self.is_compact) as submit_box:
             self.submit_box = submit_box