Update app.py

app.py

@@ -1,6 +1,7 @@
 import gradio as gr
 from gradio_client import Client
 import random
+import os
 
 # API client for the external Space
 space_client = Client("prithivMLmods/Qwen-Image-Edit-2511-LoRAs-Fast")
@@ -28,6 +29,16 @@ LORA_STYLES = [
 ]
 MAX_SEED = 2**32-1
 
+def safe_image_path(path):
+    """Ensure the given path points to an existing, readable file."""
+    if not path or not isinstance(path, str):
+        return None
+    if os.path.isfile(path):
+        return path
+    # Try to handle Gradio's nested path structure if needed
+    # (Could be adjusted depending how Gradio saves upload images)
+    return None
+
 def infer(
     image,
     prompt,
@@ -43,14 +54,25 @@ def infer(
     if image is not None:
         if isinstance(image, list):  # Gradio Gallery
             for im in image:
-                images.append({"image": {"path": im}, "caption": None})
+                file_path = safe_image_path(im)
+                if file_path:
+                    images.append({"image": {"path": file_path}, "caption": None})
+                else:
+                    print(f"警告: 路径无效或找不到文件: {im}")
         else:
-            images.append({"image": {"path": image}, "caption": None})
+            file_path = safe_image_path(image)
+            if file_path:
+                images.append({"image": {"path": file_path}, "caption": None})
+            else:
+                print(f"警告: 路径无效或找不到文件: {image}")
+
+    if len(images) == 0:
+        print("未检测到有效图片路径，未能上传图片。")
+        return None, seed
 
     if randomize_seed:
         seed = random.randint(0, MAX_SEED)
 
-    # Print API call log
     print("[调用API] space_client.predict 输入参数：")
     print(f"  images: {images}")
     print(f"  prompt: {prompt}")
@@ -59,6 +81,7 @@ def infer(
     print(f"  randomize_seed: {randomize_seed}")
     print(f"  guidance_scale: {guidance_scale}")
     print(f"  steps: {steps}")
+
     try:
         result = space_client.predict(
             images=images,
@@ -80,6 +103,12 @@ def infer(
         import traceback
         traceback.print_exc()
         print(f"[调用API] 调用接口异常: {e}")
+        # More verbose error message for image-processing errors
+        if hasattr(e, 'message') and "Could not process uploaded images" in str(e):
+            print(
+                "\n[错误] 上传图片处理失败。请确保图片文件有效且未损坏。\n"
+                "可以尝试重新选择图片或修改图片格式。"
+            )
         return None, seed
 
 # For examples, use None for image input to avoid Gradio directory/File errors on startup.
@@ -173,4 +202,7 @@ with gr.Blocks() as demo:
     )
 
 if __name__ == "__main__":
-    demo.launch(css=css)
+    # See: https://prithivmlmods-qwen-image-edit-2511-loras-fast.hf.space
+    # To use SSR, remove 'ssr_mode=False' (can help with speed in some settings)
+    # To create a public link, set share=True
+    demo.launch(css=css, ssr_mode=True, share=True)