Update app.py

app.py

@@ -1,96 +1,98 @@
 import os
-import traceback
-import tempfile
-from typing import Optional
-
 import gradio as gr
 from huggingface_hub import InferenceClient
 from PIL import Image
+import traceback
 
+# --- 核心函数：调用HuggingFace推理API ---
+def run_image_to_image(img: Image.Image, prompt: str, model: str, strength: float, guidance_scale: float) -> Image.Image:
+    """
+    使用HuggingFace InferenceClient运行image-to-image任务。
 
-def run_image_to_image(img: Image.Image, prompt: str, model: str) -> Image.Image:
+    参数:
+    - img: 输入的PIL Image对象。
+    - prompt: 描述图像变化的文本提示词。
+    - model: 要使用的HuggingFace模型库ID。
+    - strength: 控制对原始图像的修改程度 (0.0到1.0)。
+    - guidance_scale: 控制提示词与输出图像的匹配程度。
+    """
     token = os.getenv("HF_TOKEN")
     if not token:
-        raise RuntimeError("HF_TOKEN is not set. Please set your Hugging Face token in environment variables.")
+        raise gr.Error("错误：环境变量 HF_TOKEN 未设置。请在HuggingFace Space的Secrets中设置您的Hugging Face Token。")
 
-    # 直接使用Hugging Face的推理服务
-    client = InferenceClient(token=token)
-
-    # Convert PIL image to bytes
-    with tempfile.NamedTemporaryFile(suffix=".png", delete=True) as tmp:
-        img.save(tmp.name, format="PNG")
-        with open(tmp.name, "rb") as f:
-            input_bytes = f.read()
+    # 在每次调用时，根据用户选择的模型ID创建客户端
+    client = InferenceClient(model=model, token=token)
 
     try:
-        result = client.image_to_image(
-            input_bytes,
+        # 直接将PIL图像和参数传递给客户端
+        # 新版库支持直接传入PIL Image对象
+        result_img = client.image_to_image(
+            image=img,
             prompt=prompt,
-            model=model,
+            strength=strength,
+            guidance_scale=guidance_scale,
         )
-        return result
+        return result_img
     except Exception as e:
         traceback.print_exc()
-        error_msg = str(e)
-        if "402" in error_msg:
-            raise RuntimeError("错误402：付费要求。可能是模型需要付费使用或超出免费额度。请检查您的HuggingFace账户状态。")
-        raise
+        # 将详细错误反馈给用户
+        raise gr.Error(f"模型推理失败，请检查模型是否支持Image-to-Image任务或更换模型尝试。错误详情: {e}")
 
 
-with gr.Blocks(title="图像编辑工具", css=".gradio-container {max-width: 980px}") as demo:
-    gr.Markdown("**图像编辑工具** · 上传图片，输入中文或英文提示词，一键完成图像变换")
+# --- Gradio Blocks 界面 ---
+with gr.Blocks(title="图像风格转换", css=".gradio-container {max-width: 1080px}") as demo:
+    gr.Markdown("# 图像风格转换工具 🎨")
+    gr.Markdown("上传一张图片，输入提示词，选择一个模型，然后点击“一键转换”来生成新的图片。")
 
-    with gr.Row():
-        with gr.Column():
-            in_img = gr.Image(type="pil", label="上传图片", height=420)
+    with gr.Row(variant="panel"):
+        with gr.Column(scale=1):
+            in_img = gr.Image(type="pil", label="🖼️ 上传图片", height=512)
             prompt = gr.Textbox(
-                value="将图片中的人物变成卡通风格", 
-                label="提示词（支持中文和英文）", 
-                lines=2,
-                placeholder="请输入您想要的图像变换效果，例如：'将图片变成水彩画风格'、'给图片添加雪景效果'等"
+                label="📝 提示词",
+                value="A beautiful painting of a castle in a forest, fantasy, detailed, epic",
+                lines=3,
+                placeholder="输入你想要的画面效果，例如：'A cute cat wearing a wizard hat'"
             )
+            # 为了保证成功率，我们选择一个已知可靠的免费模型
+            # 您的原始模型列表可以作为备选，但需确认它们在HF上部署了免费的推理API
             model = gr.Dropdown(
                 choices=[
-                    "svjack/Stable-Diffusion-FineTuned-zh-v2",
-                    "Tencent-Hunyuan/HunyuanDiT-v1.1-Diffusers-Distilled",
-                    "tencent/HunyuanImage-3.0"
+                    "stabilityai/stable-diffusion-xl-refiner-1.0",
+                    "runwayml/stable-diffusion-v1-5",
+                    "SG161222/Realistic_Vision_V5.1_noVAE"
                 ],
-                value="svjack/Stable-Diffusion-FineTuned-zh-v2",
-                label="模型（均支持中文输入）",
+                value="stabilityai/stable-diffusion-xl-refiner-1.0",
+                label="🤖 选择模型",
+                info="这些是经过验证的免费模型，主要支持英文提示词。"
             )
-            with gr.Row():
-                run_btn = gr.Button("一键转换", variant="primary")
-                clear_btn = gr.Button("清空")
-        with gr.Column():
-            out_img = gr.Image(type="pil", label="输出结果", height=420)
-            download = gr.File(label="下载输出", interactive=False)
-            
-    gr.Markdown("""
-    ### 使用说明
-    1. 上传您想要编辑的图片
-    2. 输入中文或英文提示词，描述您想要的变换效果
-    3. 选择支持中文的模型（所有列出的模型均为免费使用）
-    4. 点击"一键转换"按钮
-    
-    ### 提示词示例
-    - 将图片变成水彩画风格
-    - 给图片添加雪景效果
-    - 将图片中的人物变成卡通风格
-    - 将图片背景变成樱花盛开的场景
-    """)
+            with gr.Accordion("高级参数 (选填)", open=False):
+                strength = gr.Slider(minimum=0.0, maximum=1.0, value=0.5, step=0.1, label="修改强度", info="值越高，图片变化越大")
+                guidance_scale = gr.Slider(minimum=0.0, maximum=20.0, value=7.5, step=0.5, label="引导强度", info="值越高，图片越接近提示词")
 
-    def _run_and_pack(img: Image.Image, p: str, m: str):
-        result_img = run_image_to_image(img, p, m)
-        # 保存为临时文件以供下载
-        with tempfile.NamedTemporaryFile(suffix=".png", delete=False) as tmp:
-            result_img.save(tmp.name)
-            return result_img, tmp.name
+            with gr.Row():
+                run_btn = gr.Button("🚀 一键转换", variant="primary", scale=2)
 
-    run_btn.click(fn=_run_and_pack, inputs=[in_img, prompt, model], outputs=[out_img, download])
-    clear_btn.click(lambda: (None, "", None, None), outputs=[in_img, prompt, out_img, download])
+        with gr.Column(scale=1):
+            out_img = gr.Image(type="pil", label="✨ 输出结果", height=512, interactive=False)
+            gr.Markdown("### 💡 提示词示例 (Examples)")
+            gr.Examples(
+                examples=[
+                    ["A beautiful oil painting of a landscape."],
+                    ["photo of a cute white cat, cinematic, 8k"],
+                    ["Make it in anime style, detailed, beautiful."],
+                    ["cyberpunk city, neon lights, rainy night"]
+                ],
+                inputs=prompt,
+                label="点击示例以填充提示词"
+            )
 
+    # --- 按钮点击事件 ---
+    run_btn.click(
+        fn=run_image_to_image,
+        inputs=[in_img, prompt, model, strength, guidance_scale],
+        outputs=out_img
+    )
 
+# --- 启动应用 ---
 if __name__ == "__main__":
-    demo.launch(server_name="0.0.0.0", server_port=7860)
-
-
+    demo.launch(share=True)