Upload 3 files

Dockerfile

@@ -0,0 +1,21 @@
+FROM python:3.10-slim
+
+WORKDIR /app
+
+# 复制依赖并安装
+COPY requirements.txt .
+RUN pip install --no-cache-dir -r requirements.txt
+
+# 复制代码
+COPY . .
+
+# 关键：赋予权限，解决 OpenPermissionError
+RUN chmod -R 777 /app
+
+# 环境变量
+ENV GRADIO_SERVER_NAME="0.0.0.0" \
+    GRADIO_SERVER_PORT="7860"
+
+EXPOSE 7860
+
+CMD ["python", "app.py"]

app.py

@@ -0,0 +1,326 @@
+import gradio as gr
+import httpx  # 替换 requests 实现异步
+import re
+import base64
+import os
+import asyncio
+from PIL import Image
+from openai import AsyncOpenAI # 使用异步客户端
+from fastapi import FastAPI
+import uvicorn
+
+# ==========================================
+# 全局配置
+# ==========================================
+
+# 1. 基础分辨率
+AR_BASES = {
+    "1:1 (SDXL标准 1024)": (1024, 1024),
+    "16:9 (SDXL标准 768p)": (1344, 768),
+    "16:9 (1080p Full HD)": (1920, 1080),
+    "9:16 (手机竖屏)": (768, 1344),
+    "4:3 (标准)":  (1152, 896),
+    "21:9 (宽屏)": (1536, 640)
+}
+
+# 2. 倍率 (Scale)
+SCALE_MULTIPLIERS = {
+    "1k": 1.0,
+    "2k": 1.5,
+    "4k": 2.0
+}
+
+# ==========================================
+# 工具函数
+# ==========================================
+
+def process_image_format(image_path, target_format):
+    """
+    CPU 密集型任务，保持同步即可，影响不大
+    """
+    try:
+        print(f"⚙️ [System] 正在转换格式: {target_format.upper()}...")
+        img = Image.open(image_path)
+        base_name = os.path.splitext(image_path)[0]
+        output_filename = f"{base_name}_final.{target_format}"
+        
+        if target_format == "jpg":
+            img = img.convert("RGB")
+            img.save(output_filename, format="JPEG", quality=95)
+        else:
+            img.save(output_filename, format="PNG")
+            
+        print(f"✅ [System] 转换完成: {output_filename}")
+        return output_filename
+    except Exception as e:
+        print(f"⚠️ [System] 格式转换失败: {e}，将返回原图")
+        return image_path
+
+async def download_image_async(url, mode_prefix):
+    """
+    异步下载图片，非阻塞
+    """
+    filename = f"temp_{mode_prefix.lower().strip('[]')}.png"
+    print(f"{mode_prefix} 检测到 URL，开始异步下载: {url}")
+    
+    try:
+        async with httpx.AsyncClient(timeout=30) as client:
+            response = await client.get(url)
+            
+        if response.status_code == 200:
+            # 写入文件操作通常很快，直接同步写入即可，或者用 aiofiles (这里保持简单)
+            with open(filename, "wb") as f:
+                f.write(response.content)
+            print(f"{mode_prefix} ✅ 图片下载并保存成功")
+            return filename
+        else:
+            raise Exception(f"HTTP {response.status_code}")
+    except Exception as e:
+        raise Exception(f"下载失败: {e}")
+
+# ==========================================
+# 核心逻辑 (异步)
+# ==========================================
+
+# 分支 1: Standard Logic (Nano Banana)
+async def generate_and_save_image_logic(client: AsyncOpenAI, model, prompt, size_str):
+    PREFIX = "🔵 [Standard]"
+    print(f"{PREFIX} 1. 开始请求 | 模型: {model} | 分辨率: {size_str}")
+    
+    try:
+        # 异步调用 API
+        response = await client.images.generate(
+            model=model,
+            prompt=prompt,
+            n=1,
+            size=size_str, 
+            response_format="b64_json"
+        )
+        print(f"{PREFIX} 2. 收到响应 | 状态: Success")
+
+        first_data = response.data[0]
+        image_b64 = getattr(first_data, 'b64_json', None)
+        
+        # URL 兼容
+        if not image_b64 and hasattr(first_data, 'url'):
+            print(f"{PREFIX} 3. 转为 URL 下载")
+            return await download_image_async(first_data.url, PREFIX)
+
+        if image_b64:
+            print(f"{PREFIX} 3. 解码 Base64...")
+            image_data = base64.b64decode(image_b64)
+            filename = "temp_standard.png"
+            with open(filename, "wb") as f:
+                f.write(image_data)
+            print(f"{PREFIX} ✅ 4. 保存成功: {filename}")
+            return filename
+        
+        raise Exception("未找到图片数据")
+
+    except Exception as e:
+        print(f"{PREFIX} ❌ 错误: {e}")
+        raise e
+
+# 分支 2: Chat Logic (Gemini)
+async def handle_gemini_output_logic(client: AsyncOpenAI, model, prompt):
+    PREFIX = "🟣 [Chat]"
+    print(f"{PREFIX} 1. 开始请求 | 模型: {model}")
+    print(f"{PREFIX}    Prompt片段: {prompt[:60]}...")
+
+    try:
+        # 异步调用 API
+        response = await client.chat.completions.create(
+            model=model,
+            messages=[{"role": "user", "content": prompt}],
+            extra_body={"modalities": ["image", "text"]}
+        )
+        print(f"{PREFIX} 2. 收到响应 | 状态: Success")
+
+        content = response.choices[0].message.content
+        filename = "temp_chat.png"
+
+        # 正则提取
+        url_pattern = r"!\[.*?\]\((https?://[^\)]+)\)"
+        url_match = re.search(url_pattern, content)
+        if url_match:
+            return await download_image_async(url_match.group(1), PREFIX)
+
+        raw_url_pattern = r"(https?://\S+\.(?:png|jpg|jpeg|webp))"
+        raw_url_match = re.search(raw_url_pattern, content)
+        if raw_url_match:
+            return await download_image_async(raw_url_match.group(1), PREFIX)
+
+        if "base64," in content:
+            base64_data = content.split("base64,")[1].split(")")[0].strip()
+            with open(filename, "wb") as f:
+                f.write(base64.b64decode(base64_data))
+            print(f"{PREFIX} ✅ 4. Base64 保存成功")
+            return filename
+
+        print(f"{PREFIX} ❌ 未能识别图片内容: {content[:100]}")
+        raise Exception("未能从对话中提取出图片")
+
+    except Exception as e:
+        print(f"{PREFIX} ❌ 错误: {e}")
+        raise e
+
+# 主调度入口 (Async)
+async def main_generate(api_key, model_name, prompt, ratio_selection, scale_selection, custom_size_input, format_selection):
+    if not api_key:
+        raise gr.Error("请输入 API Key")
+
+    # 分辨率计算
+    final_size = "1024x1024"
+    ratio_str = "1:1"
+
+    if ratio_selection == "自定义 (Custom)":
+        final_size = custom_size_input.strip()
+    else:
+        base_w, base_h = AR_BASES.get(ratio_selection, (1024, 1024))
+        multiplier = SCALE_MULTIPLIERS.get(scale_selection, 1.0)
+        final_w = int(base_w * multiplier)
+        final_h = int(base_h * multiplier)
+        final_size = f"{final_w}x{final_h}"
+        ratio_str = ratio_selection.split(" ")[0]
+
+    # Prompt 处理
+    final_prompt = prompt
+    if "gemini" in model_name.lower():
+        if ratio_selection == "自定义 (Custom)":
+             final_prompt = f"{prompt} --resolution {final_size}"
+        else:
+             final_prompt = f"{prompt} --ar {ratio_str} --resolution {final_size}"
+
+    # 初始化异步客户端
+    client = AsyncOpenAI(
+        api_key=api_key, 
+        base_url="https://api.cloudapp.ink/v1"
+    )
+    
+    try:
+        if "gemini" in model_name.lower():
+            temp_path = await handle_gemini_output_logic(client, model_name, final_prompt)
+        else:
+            temp_path = await generate_and_save_image_logic(client, model_name, final_prompt, final_size)
+            
+        final_path = process_image_format(temp_path, format_selection)
+        return final_path
+
+    except Exception as e:
+        error_str = str(e)
+        if "500" in error_str:
+            gr.Warning("服务端返回 500。4k 分辨率可能过高，建议尝试 1k 或 2k。")
+            raise gr.Error(f"生成失败 (500): {error_str}")
+        else:
+            raise gr.Error(f"生成失败: {error_str}")
+
+# ==========================================
+# UI 构建
+# ==========================================
+
+# CSS: 彻底隐藏 Footer
+css = """
+#col-container {max-width: 900px; margin-left: auto; margin-right: auto;}
+footer { display: none !important; }
+.gradio-container footer { display: none !important; }
+"""
+
+# JS: 页面加载时强制移除 Footer 元素
+js_remove_footer = """
+function() {
+    const footer = document.querySelector("footer");
+    if (footer) { footer.remove(); }
+    const footerDiv = document.querySelector(".footer");
+    if (footerDiv) { footerDiv.remove(); }
+}
+"""
+
+with gr.Blocks(css=css, js=js_remove_footer, title="Nano Banana Generator") as demo:
+    with gr.Column(elem_id="col-container"):
+        gr.Markdown(f"# Nano Banana & Gemini 图片生成")
+        gr.Markdown("支持 Nano Banana Pro 4k 及 Gemini 3 Pro 系列模型。")
+        
+        with gr.Row():
+            with gr.Column(scale=1):
+                api_key_input = gr.Textbox(
+                    label="API Key", 
+                    type="password", 
+                    placeholder="sk-..."
+                )
+                
+                # 模型列表
+                model_input = gr.Dropdown(
+                    label="模型名称", 
+                    choices=[
+                        "nano-banana",
+                        "nano-banana-pro",
+                        "nano-banana-pro-4k",
+                        "gemini-2.5-flash-image",
+                        "gemini-3-pro-image",
+                        "gemini-3-pro-image-preview"
+                    ],
+                    value="nano-banana",
+                    allow_custom_value=True 
+                )
+                
+                with gr.Row():
+                    ratio_input = gr.Dropdown(
+                        label="1. 基础比例",
+                        choices=list(AR_BASES.keys()) + ["自定义 (Custom)"],
+                        value="16:9 (1080p Full HD)" 
+                    )
+                    scale_input = gr.Dropdown(
+                        label="2. 清晰度",
+                        choices=list(SCALE_MULTIPLIERS.keys()), 
+                        value="1k"
+                    )
+                
+                custom_size_input = gr.Textbox(
+                    label="手动分辨率 (WxH)",
+                    value="1920x1080",
+                    visible=False 
+                )
+                
+                format_input = gr.Radio(
+                    label="输出格式",
+                    choices=["png", "jpg"],
+                    value="png"
+                )
+                
+                run_btn = gr.Button("🚀 生成图片", variant="primary")
+
+            with gr.Column(scale=2):
+                prompt_input = gr.Textbox(
+                    label="提示词", 
+                    placeholder="请输入提示词...", 
+                    lines=4
+                )
+                
+                result_output = gr.Image(
+                    label="生成结果", 
+                    type="filepath", 
+                    interactive=False
+                )
+
+        def toggle_custom_box(ratio_val):
+            return gr.update(visible=(ratio_val == "自定义 (Custom)"))
+
+        ratio_input.change(fn=toggle_custom_box, inputs=ratio_input, outputs=custom_size_input)
+
+        run_btn.click(
+            fn=main_generate,
+            inputs=[api_key_input, model_input, prompt_input, ratio_input, scale_input, custom_size_input, format_input],
+            outputs=[result_output]
+        )
+
+# ==========================================
+# FastAPI & Health Check
+# ==========================================
+app = FastAPI()
+
+@app.get("/health")
+def health_check():
+    return {"status": "ok", "message": "Service is healthy"}
+
+# 挂载 Gradio 到根路径
+app = gr.mount_gradio_app(app, demo, path="/")

requirements.txt

@@ -0,0 +1,6 @@
+gradio
+httpx
+pillow
+openai>=1.0.0
+fastapi
+uvicorn