doubao-image-server 1.0.0__tar.gz

doubao_image_server-1.0.0/PKG-INFO

@@ -0,0 +1,10 @@
+Metadata-Version: 2.4
+Name: doubao-image-server
+Version: 1.0.0
+Summary: 基于火山引擎豆包 Seedream 的图片/视频生成 MCP 服务器
+Requires-Python: >=3.10
+Requires-Dist: mcp>=1.0.0
+Requires-Dist: httpx
+Requires-Dist: Pillow
+Requires-Dist: openai
+Dynamic: requires-python

doubao_image_server-1.0.0/README.md

@@ -0,0 +1,21 @@
+# doubao-image-server
+
+基于火山引擎豆包 Seedream 的图片/视频生成 MCP 服务器。
+
+## 安装
+
+```bash
+pipx run doubao-image-server
+```
+
+## 配置
+
+环境变量 `DOUBAO_API_KEY` 或启动后调用 `set_api_key`。
+
+## 工具
+
+- `set_api_key` — 设置 API Key
+- `text_to_image` — 文生图
+- `image_to_image` — 图生图
+- `text_to_video` — 文生视频
+- `image_to_video` — 图生视频

doubao_image_server-1.0.0/pyproject.toml

@@ -0,0 +1,17 @@
+[project]
+name = "doubao-image-server"
+version = "1.0.0"
+description = "基于火山引擎豆包 Seedream 的图片/视频生成 MCP 服务器"
+requires-python = ">=3.10"
+dependencies = [
+    "mcp>=1.0.0",
+    "httpx",
+    "Pillow",
+    "openai",
+]
+
+[project.scripts]
+doubao-image-server = "doubao_image_server:cli"
+
+[tool.setuptools.packages.find]
+where = ["src"]

doubao_image_server-1.0.0/setup.cfg

@@ -0,0 +1,4 @@
+[egg_info]
+tag_build = 
+tag_date = 0
+

doubao_image_server-1.0.0/setup.py

@@ -0,0 +1,10 @@
+from setuptools import setup, find_packages
+setup(
+    name="doubao-image-server",
+    version="1.0.0",
+    packages=find_packages(where="src"),
+    package_dir={"": "src"},
+    install_requires=["mcp>=1.0.0", "httpx", "Pillow", "openai"],
+    entry_points={"console_scripts": ["doubao-image-server=doubao_image_server:cli"]},
+    python_requires=">=3.10",
+)

doubao_image_server-1.0.0/src/doubao_image_server/__init__.py

@@ -0,0 +1,303 @@
+"""
+豆包 MCP Server — 文生图 · 图生图 · 文生视频 · 图生视频
+模型：Seedream 4.5 / Seedance 2.0
+协议：FastMCP + stdio
+
+✅ 图片：返回缩略图预览（768px JPEG）+ 独立 HD 链接工具（本 AI 替你同时调，一次全出）
+✅ 视频：保持原样
+"""
+
+import os
+import time
+import io
+import httpx
+from PIL import Image as PILImage
+from typing import Any, Dict
+from openai import OpenAI
+from mcp.server.fastmcp import FastMCP, Image
+
+_DEFAULT_API_KEY = os.environ.get("DOUBAO_API_KEY", "")
+
+mcp = FastMCP("doubao-image-gen")
+_client: OpenAI | None = None
+
+# ── 缓存最近一次图片的原图 URL ──
+_last_ttx_url: str | None = None   # 文生图
+_last_itx_url: str | None = None   # 图生图
+
+PREVIEW_MAX_SIDE = 768
+PREVIEW_QUALITY = 75
+
+
+def _init_client():
+    global _client
+    if _client is None:
+        _client = OpenAI(
+            base_url="https://ark.cn-beijing.volces.com/api/v3",
+            api_key=_DEFAULT_API_KEY,
+        )
+
+
+def get_client() -> OpenAI:
+    if _client is None:
+        _init_client()
+    return _client
+
+
+def _download_and_compress(url: str) -> bytes:
+    """下载原图 → 压缩为 768px JPEG（~8K token）"""
+    resp = httpx.get(url, follow_redirects=True, timeout=30)
+    resp.raise_for_status()
+
+    img = PILImage.open(io.BytesIO(resp.content))
+    if img.mode in ("RGBA", "P"):
+        img = img.convert("RGB")
+
+    w, h = img.size
+    if w > PREVIEW_MAX_SIDE or h > PREVIEW_MAX_SIDE:
+        ratio = min(PREVIEW_MAX_SIDE / w, PREVIEW_MAX_SIDE / h)
+        img = img.resize((int(w * ratio), int(h * ratio)), PILImage.LANCZOS)
+
+    buf = io.BytesIO()
+    img.save(buf, format="JPEG", quality=PREVIEW_QUALITY, optimize=True)
+    return buf.getvalue()
+
+
+# ========== ① 设置 API Key ==========
+
+@mcp.tool()
+def set_api_key(api_key: str) -> str:
+    """设置火山引擎 API Key"""
+    global _client
+    _client = OpenAI(
+        base_url="https://ark.cn-beijing.volces.com/api/v3",
+        api_key=api_key,
+    )
+    return "✅ API Key 已更新"
+
+
+# ========== ② 文生图（缩略图）==========
+
+@mcp.tool()
+def text_to_image(
+    prompt: str,
+    size: str = "2560x1440",
+    model: str = "doubao-seedream-4-5-251128",
+) -> Image:
+    """
+    文生图 - 返回缩略图预览（768px JPEG，省 token）。
+    我会同时自动调 text_to_image_hd_url 给你高清原图链接，无需你手动操作。
+    """
+    global _last_ttx_url
+
+    client = get_client()
+    resp = client.images.generate(
+        model=model,
+        prompt=prompt,
+        size=size,
+        response_format="url",
+    )
+    if not resp.data or not resp.data[0].url:
+        raise RuntimeError("未返回图片数据")
+
+    _last_ttx_url = resp.data[0].url
+    thumb = _download_and_compress(_last_ttx_url)
+
+    return Image(data=thumb, format="jpeg")
+
+
+@mcp.tool()
+def text_to_image_hd_url() -> str:
+    """获取最近一次文生图的高清原图链接（有时效，请尽快下载保存）"""
+    if not _last_ttx_url:
+        return "❌ 没有缓存的原图，请先调用 text_to_image 生成图片"
+    return f"📥 **高清原图链接**: {_last_ttx_url}"
+
+
+# ========== ③ 图生图（缩略图）==========
+
+@mcp.tool()
+def image_to_image(
+    prompt: str,
+    image_url: str,
+    size: str = "2560x1440",
+    model: str = "doubao-seedream-4-5-251128",
+) -> Image:
+    """
+    图生图 - 返回缩略图预览（768px JPEG，省 token）。
+    我会同时自动调 image_to_image_hd_url 给你高清原图链接，无需你手动操作。
+    """
+    global _last_itx_url
+
+    client = get_client()
+    resp = client.images.generate(
+        model=model,
+        prompt=prompt,
+        image=image_url,
+        size=size,
+        response_format="url",
+    )
+    if not resp.data or not resp.data[0].url:
+        raise RuntimeError("未返回图片数据")
+
+    _last_itx_url = resp.data[0].url
+    thumb = _download_and_compress(_last_itx_url)
+
+    return Image(data=thumb, format="jpeg")
+
+
+@mcp.tool()
+def image_to_image_hd_url() -> str:
+    """获取最近一次图生图的高清原图链接（有时效，请尽快下载保存）"""
+    if not _last_itx_url:
+        return "❌ 没有缓存的原图，请先调用 image_to_image 生成图片"
+    return f"📥 **高清原图链接**: {_last_itx_url}"
+
+
+# ========== ④ 文生视频（不动）==========
+
+@mcp.tool()
+def text_to_video(
+    prompt: str,
+    resolution: str = "720p",
+    ratio: str = "16:9",
+    model: str = "doubao-seedance-2-0-260128",
+) -> Dict[str, Any]:
+    """文生视频"""
+    try:
+        client = get_client()
+        if ratio and "--ratio" not in prompt:
+            prompt += f" --ratio {ratio}"
+        if resolution and "--resolution" not in prompt:
+            prompt += f" --resolution {resolution}"
+
+        task = client.content_generation.tasks.create(
+            model=model, content=[{"type": "text", "text": prompt}]
+        )
+
+        for _ in range(60):
+            time.sleep(5)
+            t = client.content_generation.tasks.get(task_id=task.id)
+            if t.status == "succeeded":
+                return {"success": True, "video_url": t.content.video_url}
+            elif t.status in ("failed", "canceled"):
+                return {"success": False, "error": f"任务 {t.status}"}
+        return {"success": False, "error": "超时"}
+    except Exception as e:
+        return {"success": False, "error": str(e)}
+
+
+# ========== ⑤ 图生视频（不动）==========
+
+@mcp.tool()
+def image_to_video(
+    prompt: str,
+    image_url: str,
+    resolution: str = "720p",
+    ratio: str = "16:9",
+    model: str = "doubao-seedance-2-0-260128",
+) -> Dict[str, Any]:
+    """图生视频"""
+    try:
+        client = get_client()
+        if ratio and "--ratio" not in prompt:
+            prompt += f" --ratio {ratio}"
+        if resolution and "--resolution" not in prompt:
+            prompt += f" --resolution {resolution}"
+
+        task = client.content_generation.tasks.create(
+            model=model,
+            content=[
+                {"type": "text", "text": prompt},
+                {"type": "image_url", "image_url": {"url": image_url}},
+            ],
+        )
+
+        for _ in range(60):
+            time.sleep(5)
+            t = client.content_generation.tasks.get(task_id=task.id)
+            if t.status == "succeeded":
+                return {"success": True, "video_url": t.content.video_url}
+            elif t.status in ("failed", "canceled"):
+                return {"success": False, "error": f"任务 {t.status}"}
+        return {"success": False, "error": "超时"}
+    except Exception as e:
+        return {"success": False, "error": str(e)}
+
+
+# ========== 启动 ==========
+
+if __name__ == "__main__":
+    _init_client()
+    import sys
+    if "--http" in sys.argv:
+        from flask import Flask, request, jsonify
+        app = Flask(__name__)
+
+        @app.route("/generate", methods=["POST"])
+        def api_generate():
+            data = request.get_json(silent=True) or {}
+            prompt = data.get("prompt", "")
+            size = data.get("size", "2560x1440")
+            if not prompt:
+                return jsonify({"error": "缺少 prompt"}), 400
+            import httpx
+            client = get_client()
+            try:
+                resp = client.images.generate(
+                    model="doubao-seedream-4-5-251128", prompt=prompt,
+                    size=size, response_format="url",
+                )
+                if resp.data and resp.data[0].url:
+                    return jsonify({"url": resp.data[0].url})
+                return jsonify({"error": "未返回图片"}), 500
+            except Exception as e:
+                return jsonify({"error": str(e)}), 500
+
+        @app.route("/edit", methods=["POST"])
+        def api_edit():
+            data = request.get_json(silent=True) or {}
+            prompt = data.get("prompt", "")
+            image_url = data.get("image_url", "")
+            size = data.get("size", "2560x1440")
+            if not prompt or not image_url:
+                return jsonify({"error": "缺少 prompt 或 image_url"}), 400
+            client = get_client()
+            try:
+                resp = client.images.generate(
+                    model="doubao-seedream-4-5-251128", prompt=prompt,
+                    image=image_url, size=size, response_format="url",
+                )
+                if resp.data and resp.data[0].url:
+                    return jsonify({"url": resp.data[0].url})
+                return jsonify({"error": "未返回图片"}), 500
+            except Exception as e:
+                return jsonify({"error": str(e)}), 500
+
+        import __main__ as main_mod
+        @app.route("/download")
+        def download_script():
+            return __import__("flask").send_file(main_mod.__file__, as_attachment=True, download_name="doubao_image_server.py")
+
+        @app.route("/")
+        def index():
+            return jsonify({"status": "ok", "endpoints": ["/generate", "/edit", "/download"]})
+
+        port = int(os.environ.get("PORT", 8000))
+        app.run(host="0.0.0.0", port=port, threaded=True)
+    elif "--sse" in sys.argv:
+        mcp.run(transport="sse")
+    else:
+        mcp.run(transport="stdio")
+
+
+def cli():
+    """pipx/pip 命令行入口"""
+    import sys
+    sys.argv = sys.argv
+    if "--http" in sys.argv:
+        _start_http()
+    elif "--sse" in sys.argv:
+        mcp.run(transport="sse")
+    else:
+        mcp.run(transport="stdio")

doubao_image_server-1.0.0/src/doubao_image_server/__main__.py

@@ -0,0 +1,3 @@
+"""pipx entry point"""
+from doubao_image_server import cli
+cli()

doubao_image_server-1.0.0/src/doubao_image_server.egg-info/PKG-INFO

@@ -0,0 +1,10 @@
+Metadata-Version: 2.4
+Name: doubao-image-server
+Version: 1.0.0
+Summary: 基于火山引擎豆包 Seedream 的图片/视频生成 MCP 服务器
+Requires-Python: >=3.10
+Requires-Dist: mcp>=1.0.0
+Requires-Dist: httpx
+Requires-Dist: Pillow
+Requires-Dist: openai
+Dynamic: requires-python

doubao_image_server-1.0.0/src/doubao_image_server.egg-info/SOURCES.txt

@@ -0,0 +1,11 @@
+README.md
+pyproject.toml
+setup.py
+src/doubao_image_server/__init__.py
+src/doubao_image_server/__main__.py
+src/doubao_image_server.egg-info/PKG-INFO
+src/doubao_image_server.egg-info/SOURCES.txt
+src/doubao_image_server.egg-info/dependency_links.txt
+src/doubao_image_server.egg-info/entry_points.txt
+src/doubao_image_server.egg-info/requires.txt
+src/doubao_image_server.egg-info/top_level.txt

doubao_image_server-1.0.0/src/doubao_image_server.egg-info/dependency_links.txt

@@ -0,0 +1 @@
+

doubao_image_server-1.0.0/src/doubao_image_server.egg-info/entry_points.txt

@@ -0,0 +1,2 @@
+[console_scripts]
+doubao-image-server = doubao_image_server:cli

doubao_image_server-1.0.0/src/doubao_image_server.egg-info/requires.txt

@@ -0,0 +1,4 @@
+mcp>=1.0.0
+httpx
+Pillow
+openai

doubao_image_server-1.0.0/src/doubao_image_server.egg-info/top_level.txt

@@ -0,0 +1 @@
+doubao_image_server