Update app.py

app.py

@@ -1,14 +1,16 @@
 import gradio as gr
-from gradio_client import Client, handle_file
+from gradio_client import Client
 import random
 import os
+import requests
+import tempfile
 
 HF_TOKEN = os.environ.get("girlToken")
 
 space_client = Client(
     "prithivMLmods/Qwen-Image-Edit-2511-LoRAs-Fast",
     token=HF_TOKEN
-    )
+)
 
 LORA_STYLES = [
     'Multiple-Angles', 'Photo-to-Anime', 'Anime-V2', 'Light-Migration',
@@ -21,6 +23,23 @@ LORA_STYLES = [
 MAX_SEED = 2**31 - 1
 
 
+def upload_to_imgbb(image_path):
+    """上传图片到免费图床，返回公开 URL"""
+    IMGBB_API_KEY = os.environ.get("IMGBB_API_KEY")
+    if not IMGBB_API_KEY:
+        return None
+    with open(image_path, "rb") as f:
+        import base64
+        b64 = base64.b64encode(f.read()).decode()
+    resp = requests.post(
+        "https://api.imgbb.com/1/upload",
+        data={"key": IMGBB_API_KEY, "image": b64}
+    )
+    if resp.status_code == 200:
+        return resp.json()["data"]["url"]
+    return None
+
+
 def infer(
     image,
     prompt,
@@ -31,31 +50,25 @@ def infer(
     steps,
     progress=gr.Progress(track_tqdm=True),
 ):
-    if image is None:
+    if image is None or not os.path.exists(image):
         print("未上传图片")
         return None, seed
 
-    if not os.path.exists(image):
-        print(f"图片路径不存在: {image}")
-        return None, seed
-
     if randomize_seed:
         seed = random.randint(0, MAX_SEED)
 
-    # 关键：用 handle_file 上传到目标 Space，得到远端可访问的文件对象
-    uploaded = handle_file(image)
+    # 方案：上传到公开图床，拿到 URL 再传给远端
+    public_url = upload_to_imgbb(image)
+    if not public_url:
+        print("图床上传失败，请检查 IMGBB_API_KEY")
+        return None, seed
+
+    print(f"图片公开 URL: {public_url}")
 
-    # Gallery 元素格式：{"image": <上传后的文件对象>, "caption": None}
-    images_input = [{"image": uploaded, "caption": None}]
+    # 直接传 URL 字符串，让远端自己下载
+    images_input = [{"image": {"path": None, "url": public_url, "size": None, "orig_name": os.path.basename(image), "mime_type": None, "is_stream": False, "meta": {"_type": "gradio.FileData"}}, "caption": None}]
 
-    print("[调用API] 输入参数：")
-    print(f"  image path: {image}")
-    print(f"  uploaded: {uploaded}")
-    print(f"  prompt: {prompt}")
-    print(f"  lora_adapter: {lora_adapter}")
-    print(f"  seed: {seed}")
-    print(f"  guidance_scale: {guidance_scale}")
-    print(f"  steps: {steps}")
+    print(f"prompt: {prompt}, lora: {lora_adapter}, seed: {seed}")
 
     try:
         result = space_client.predict(
@@ -69,21 +82,18 @@ def infer(
             api_name="/infer",
         )
 
-        print(f"[调用API] 返回值: {result}")
-
+        print(f"返回值: {result}")
         image_info, seed_used = result
-
         if isinstance(image_info, dict):
             img_out = image_info.get("path") or image_info.get("url")
         else:
             img_out = image_info
-
         return img_out, int(seed_used)
 
     except Exception as e:
         import traceback
         traceback.print_exc()
-        print(f"[调用API] 异常: {e}")
+        print(f"异常: {e}")
         return None, seed
 
 
@@ -98,50 +108,17 @@ with gr.Blocks(css=css) as demo:
     with gr.Column(elem_id="col-container"):
         gr.Markdown("# 图像编辑 Demo\n基于 prithivMLmods/Qwen-Image-Edit-2511-LoRAs-Fast")
 
-        image = gr.Image(
-            label="上传图片",
-            sources=["upload"],
-            type="filepath",
-        )
-
-        prompt = gr.Text(
-            label="编辑描述（Prompt）",
-            placeholder="请输入图片编辑描述...",
-        )
-
-        lora_adapter = gr.Dropdown(
-            label="编辑风格（Style）",
-            choices=LORA_STYLES,
-            value="Photo-to-Anime"
-        )
-
+        image = gr.Image(label="上传图片", sources=["upload"], type="filepath")
+        prompt = gr.Text(label="编辑描述（Prompt）", placeholder="请输入图片编辑描述...")
+        lora_adapter = gr.Dropdown(label="编辑风格（Style）", choices=LORA_STYLES, value="Photo-to-Anime")
         run_button = gr.Button("执行编辑", variant="primary")
-
-        result = gr.Image(label="结果图片", show_label=True)
+        result = gr.Image(label="结果图片")
 
         with gr.Accordion("高级设置", open=False):
-            seed = gr.Slider(
-                label="随机种子",
-                minimum=0,
-                maximum=MAX_SEED,
-                step=1,
-                value=0,
-            )
+            seed = gr.Slider(label="随机种子", minimum=0, maximum=MAX_SEED, step=1, value=0)
             randomize_seed = gr.Checkbox(label="随机化种子", value=True)
-            guidance_scale = gr.Slider(
-                label="引导强度 (Guidance Scale)",
-                minimum=0.1,
-                maximum=10.0,
-                step=0.1,
-                value=1.0,
-            )
-            steps = gr.Slider(
-                label="推理步数 (Steps)",
-                minimum=1,
-                maximum=50,
-                step=1,
-                value=4,
-            )
+            guidance_scale = gr.Slider(label="引导强度", minimum=0.1, maximum=10.0, step=0.1, value=1.0)
+            steps = gr.Slider(label="推理步数", minimum=1, maximum=50, step=1, value=4)
 
     gr.on(
         triggers=[run_button.click, prompt.submit],
@@ -151,4 +128,11 @@ with gr.Blocks(css=css) as demo:
     )
 
 if __name__ == "__main__":
-    demo.launch(ssr_mode=False, share=True)
+    demo.launch(ssr_mode=False, share=True)
+# ```
+
+# 然后去 [imgbb.com](https://imgbb.com/api) 免费申请一个 API key，存到 HF Space 的 Secrets 里，变量名 `IMGBB_API_KEY`。
+
+# 这样的流程是：
+# ```
+