Update app.py

app.py

@@ -2,7 +2,6 @@ import gradio as gr
 from gradio_client import Client, handle_file
 import random
 import os
-from PIL import Image
 
 # API client for the external Space
 space_client = Client("prithivMLmods/Qwen-Image-Edit-2511-LoRAs-Fast")
@@ -31,20 +30,11 @@ LORA_STYLES = [
 MAX_SEED = 2**31 - 1
 
 def encode_image_to_gallery_dict(image_path):
-    """
-    Returns the dict structure required for GalleryData, containing "image" (binary data) and "orig_name".
-    This directly loads the image and provides the binary data instead of a path, fixing the GalleryData validation error.
-    """
-    if not image_path or not isinstance(image_path, str) or not os.path.isfile(image_path):
+    if not image_path or not isinstance(image_path, str):
         return None
     try:
-        # Read image into bytes
-        with open(image_path, "rb") as f:
-            image_bytes = f.read()
-        return {
-            "image": image_bytes,
-            "orig_name": os.path.basename(image_path)
-        }
+        file_data = handle_file(image_path)
+        return {"image": file_data, "caption": ""}
     except Exception as e:
         print(f"无法读取图片: {image_path}: {e}")
         return None
@@ -59,10 +49,10 @@ def infer(
     steps,
     progress=gr.Progress(track_tqdm=True),
 ):
-    # Prepare images_input as a list of dicts containing "image" (binary data)
+    # Process input image(s) as a list of dict, each of form {"image": {"data": ...}}
     images_input = []
     if image is not None:
-        if isinstance(image, list):
+        if isinstance(image, list):  # Gradio Gallery
             for im in image:
                 img_obj = encode_image_to_gallery_dict(im)
                 if img_obj:
@@ -85,7 +75,6 @@ def infer(
 
     print("[调用API] space_client.predict 输入参数：")
     print(f"  images count: {len(images_input)}")
-    print(f"  images structure: {images_input}")
     print(f"  prompt: {prompt}")
     print(f"  lora_adapter: {lora_adapter}")
     print(f"  seed: {seed}")
@@ -94,7 +83,7 @@ def infer(
     print(f"  steps: {steps}")
 
     try:
-        # Use plain list of dicts with "image" key as images parameter
+        # Gradio Space expects list of {"image": {"data": <b64 str>}} elements (see AppError in prompt)
         result = space_client.predict(
             images=images_input,
             prompt=prompt,
@@ -108,6 +97,7 @@ def infer(
         print(f"[调用API] space_client.predict 返回值: {result}")
         # result可能不是元组形式，需增加健壮性
         if isinstance(result, dict):
+            # 新API只返回dict
             image_info = result
             seed_used = result.get("seed", seed)
         elif isinstance(result, (tuple, list)) and len(result) == 2:
@@ -116,9 +106,12 @@ def infer(
             print(f"[错误] space_client.predict 返回类型未知: {type(result)}，内容: {result}")
             return None, seed
 
+        # 检查image_info是否包含url/path
         if isinstance(image_info, dict):
             img_url = image_info.get("url") or image_info.get("path") or None
+            # 获取base64图片（如果有）
             if img_url is None and "data" in image_info:
+                # 返回base64图片（data格式）
                 return f"data:image/png;base64,{image_info['data']}", seed_used
             return img_url, seed_used
         else:
@@ -133,11 +126,6 @@ def infer(
                 "\n[错误] 上传图片处理失败。请确保图片文件有效且未损坏。\n"
                 "可以尝试重新选择图片或修改图片格式。"
             )
-        elif "validation errors for GalleryData" in str(e):
-            print(
-                "\n[错误] 图片数据上传格式错误。请检查图片输入的数据结构，"
-                "确保为文件路径字符串或有效图片文件。"
-            )
         else:
             print("\n[错误] 调用推理服务失败，请稍后重试或联系开发者。")
         return None, seed