Upload 8 files

app.py

@@ -8,12 +8,12 @@ from google.genai import types
 import json
 import asyncio
 from typing import Dict, Any, List, Optional
-from mcp import Server, NotificationOptions
-from mcp.server.models import InitializationOptions
-import mcp.server.stdio
-import mcp.types as types_mcp
 import logging
 import traceback
+from fastapi import FastAPI
+from fastapi.responses import JSONResponse
+import uvicorn
+from threading import Thread
 
 # ログ設定
 logging.basicConfig(
@@ -22,6 +22,9 @@ logging.basicConfig(
 )
 logger = logging.getLogger(__name__)
 
+# FastAPIアプリケーションの初期化
+app = FastAPI()
+
 # Gemini APIクライアントの初期化
 def get_gemini_client():
     api_key = os.environ.get("GEMINI_API_KEY")
@@ -119,90 +122,67 @@ def generate_image(prompt: str, previous_image: Optional[Image.Image] = None) ->
         logger.error(traceback.format_exc())
         return None, f"エラーが発生しました: {str(e)}"
 
-# MCPサーバーの設定
-class ImageGenMCPServer:
-    def __init__(self):
-        self.server = Server("image-gen-mcp")
-        self.setup_handlers()
+# MCPエンドポイント
+@app.post("/mcp/list_tools")
+async def mcp_list_tools():
+    """MCPツールのリストを返す"""
+    return {
+        "tools": [
+            {
+                "name": "generate_image",
+                "description": "Gemini 2.0 Flash Previewを使用して画像を生成します",
+                "inputSchema": {
+                    "type": "object",
+                    "properties": {
+                        "prompt": {
+                            "type": "string",
+                            "description": "生成したい画像の説明"
+                        }
+                    },
+                    "required": ["prompt"]
+                }
+            }
+        ]
+    }
+
+@app.post("/mcp/call_tool")
+async def mcp_call_tool(request: Dict[str, Any]):
+    """MCPツールを実行する"""
+    tool_name = request.get("name")
+    arguments = request.get("arguments", {})
     
-    def setup_handlers(self):
-        @self.server.list_tools()
-        async def handle_list_tools() -> list[types_mcp.Tool]:
-            return [
-                types_mcp.Tool(
-                    name="generate_image",
-                    description="Gemini 2.0 Flashを使用して画像を生成します",
-                    inputSchema={
-                        "type": "object",
-                        "properties": {
-                            "prompt": {
-                                "type": "string",
-                                "description": "生成したい画像の説明"
-                            }
-                        },
-                        "required": ["prompt"]
-                    }
-                )
-            ]
+    if tool_name == "generate_image":
+        prompt = arguments.get("prompt", "")
         
-        @self.server.call_tool()
-        async def handle_call_tool(
-            name: str, 
-            arguments: Optional[Dict[str, Any]] = None
-        ) -> List[types_mcp.TextContent]:
-            if name == "generate_image":
-                prompt = arguments.get("prompt", "")
-                
-                logger.info(f"MCPツール呼び出し: generate_image, プロンプト='{prompt[:50]}...'")
-                
-                # 同期関数を非同期で実行
-                loop = asyncio.get_event_loop()
-                image, message = await loop.run_in_executor(
-                    None, generate_image, prompt, None
-                )
-                
-                if image:
-                    # 画像をbase64エンコード
-                    buffered = io.BytesIO()
-                    image.save(buffered, format="PNG")
-                    img_str = base64.b64encode(buffered.getvalue()).decode()
-                    logger.info("MCPレスポンス: 成功")
-                    
-                    return [
-                        types_mcp.TextContent(
-                            type="text",
-                            text=json.dumps({
-                                "success": True,
-                                "message": message,
-                                "image_base64": img_str
-                            })
-                        )
-                    ]
-                else:
-                    logger.warning(f"MCPレスポンス: 失敗 - {message}")
-                    return [
-                        types_mcp.TextContent(
-                            type="text",
-                            text=json.dumps({
-                                "success": False,
-                                "message": message
-                            })
-                        )
-                    ]
+        # 画像生成を実行
+        image, message = generate_image(prompt)
+        
+        if image:
+            # 画像をbase64エンコード
+            buffered = io.BytesIO()
+            image.save(buffered, format="PNG")
+            img_str = base64.b64encode(buffered.getvalue()).decode()
             
-            raise ValueError(f"Unknown tool: {name}")
+            return JSONResponse({
+                "success": True,
+                "message": message,
+                "image_base64": img_str
+            })
+        else:
+            return JSONResponse({
+                "success": False,
+                "message": message
+            })
     
-    async def run(self):
-        logger.info("MCPサーバー起動中...")
-        async with mcp.server.stdio.stdio_server() as (read_stream, write_stream):
-            await self.server.run(
-                read_stream,
-                write_stream,
-                InitializationOptions(
-                    server_name="image-gen-mcp",
-                    server_version="0.1.0"
-                )
-            )
+    return JSONResponse({
+        "success": False,
+        "message": f"Unknown tool: {tool_name}"
+    }, status_code=400)
+
+@app.get("/health")
+async def health_check():
+    """ヘルスチェックエンドポイント"""
+    return {"status": "healthy"}
 
 # Gradioインターフェースの作成
 def create_gradio_interface():
@@ -210,57 +190,73 @@ def create_gradio_interface():
         gr.Markdown("""
         # 画像生成MCPサーバー
         
-        Gemini 2.0 Flashを使用した画像生成アプリケーションです。
-        ClaudeCodeから画像生成MCPツールとして利用できます。
-        """)
+        このアプリケーションは主にClaudeCodeから利用するためのMCPサーバーです。
+        Gemini 2.0 Flash Previewを使用して画像を生成します。
         
-        with gr.Row():
-            with gr.Column():
-                prompt_input = gr.Textbox(
-                    label="プロンプト",
-                    placeholder="生成したい画像の説明を入力してください...",
-                    lines=3
-                )
-                
-                reference_image = gr.Image(
-                    label="参考画像（オプション）",
-                    type="pil"
-                )
-                
-                generate_btn = gr.Button("画像を生成", variant="primary")
-                
-            with gr.Column():
-                output_image = gr.Image(
-                    label="生成された画像",
-                    type="pil"
-                )
-                
-                status_output = gr.Textbox(
-                    label="ステータス",
-                    interactive=False
-                )
+        ## ClaudeCodeでの使用方法
         
-        # MCPサーバー情報の表示
-        gr.Markdown("""
-        ### MCPサーバー情報
+        以下のエンドポイントをMCP設定に追加してください：
+        ```
+        https://[your-space-name].hf.space/mcp/
+        ```
+        """)
         
-        このアプリケーションはMCPサーバーとして動作します。
-        ClaudeCodeで使用する場合は、以下の設定を使用してください：
+        with gr.Tab("画像生成テスト"):
+            with gr.Row():
+                with gr.Column():
+                    prompt_input = gr.Textbox(
+                        label="プロンプト",
+                        placeholder="生成したい画像の説明を入力してください...",
+                        lines=3
+                    )
+                    
+                    reference_image = gr.Image(
+                        label="参考画像（オプション）",
+                        type="pil"
+                    )
+                    
+                    generate_btn = gr.Button("画像を生成", variant="primary")
+                    
+                with gr.Column():
+                    output_image = gr.Image(
+                        label="生成された画像",
+                        type="pil"
+                    )
+                    
+                    status_output = gr.Textbox(
+                        label="ステータス",
+                        interactive=False
+                    )
         
-        ```json
-        {
-          "mcpServers": {
-            "image-gen": {
-              "command": "python",
-              "args": ["app.py", "--mcp"],
-              "env": {
-                "GEMINI_API_KEY": "your-api-key-here"
-              }
+        with gr.Tab("MCPエンドポイント情報"):
+            gr.Markdown("""
+            ### エンドポイント一覧
+            
+            - `POST /mcp/list_tools` - 利用可能なツールのリストを取得
+            - `POST /mcp/call_tool` - ツールを実行
+            - `GET /health` - ヘルスチェック
+            
+            ### ツール: generate_image
+            
+            **パラメータ:**
+            ```json
+            {
+                "name": "generate_image",
+                "arguments": {
+                    "prompt": "生成したい画像の説明"
+                }
             }
-          }
-        }
-        ```
-        """)
+            ```
+            
+            **レスポンス:**
+            ```json
+            {
+                "success": true,
+                "message": "画像生成に成功しました！",
+                "image_base64": "base64エンコードされた画像データ"
+            }
+            ```
+            """)
         
         # イベントハンドラ
         generate_btn.click(
@@ -271,17 +267,33 @@ def create_gradio_interface():
     
     return demo
 
+# FastAPIをバックグラウンドで起動
+def run_fastapi():
+    uvicorn.run(app, host="0.0.0.0", port=7860)
+
 # メイン実行部分
 if __name__ == "__main__":
-    import sys
-    
-    if "--mcp" in sys.argv:
-        # MCPサーバーモードで実行
-        logger.info("MCPサーバーモードで起動")
-        mcp_server = ImageGenMCPServer()
-        asyncio.run(mcp_server.run())
+    # APIキーチェック
+    if not os.environ.get("GEMINI_API_KEY"):
+        logger.error("GEMINI_API_KEY環境変数が設定されていません")
+        # Gradioで警告を表示
+        with gr.Blocks(title="エラー - 画像生成MCP") as demo:
+            gr.Markdown("""
+            # ⚠️ 設定エラー
+            
+            GEMINI_API_KEY環境変数が設定されていません。
+            
+            Hugging Face SpacesのSettings > Repository secretsで以下を設定してください：
+            - Name: `GEMINI_API_KEY`
+            - Value: あなたのGemini APIキー
+            """)
+        demo.launch(server_name="0.0.0.0", server_port=7860)
     else:
-        # Gradio UIモードで実行
-        logger.info("Gradio UIモードで起動")
+        # GradioとFastAPIを統合
+        logger.info("MCPサーバーとGradio UIを起動中...")
         demo = create_gradio_interface()
-        demo.launch(server_name="0.0.0.0", server_port=7860)
+        # GradioにFastAPIアプリケーションをマウント
+        gr.mount_gradio_app(app, demo, path="/")
+        
+        # 統合サーバーを起動
+        uvicorn.run(app, host="0.0.0.0", port=7860)

mcp_client.py

@@ -0,0 +1,75 @@
+"""
+MCPクライアントのサンプルコード
+ClaudeCodeからこのMCPサーバーを利用する際の参考実装
+"""
+
+import requests
+import base64
+from PIL import Image
+import io
+
+class ImageGenMCPClient:
+    def __init__(self, base_url: str):
+        """
+        Args:
+            base_url: MCPサーバーのベースURL (例: "https://your-space.hf.space")
+        """
+        self.base_url = base_url.rstrip('/')
+    
+    def list_tools(self):
+        """利用可能なツールのリストを取得"""
+        response = requests.post(f"{self.base_url}/mcp/list_tools")
+        return response.json()
+    
+    def generate_image(self, prompt: str) -> tuple[Image.Image, str]:
+        """
+        画像を生成する
+        
+        Args:
+            prompt: 生成したい画像の説明
+            
+        Returns:
+            生成された画像とメッセージ
+        """
+        payload = {
+            "name": "generate_image",
+            "arguments": {
+                "prompt": prompt
+            }
+        }
+        
+        response = requests.post(f"{self.base_url}/mcp/call_tool", json=payload)
+        result = response.json()
+        
+        if result.get("success"):
+            # base64デコードして画像を取得
+            img_data = base64.b64decode(result["image_base64"])
+            image = Image.open(io.BytesIO(img_data))
+            return image, result["message"]
+        else:
+            return None, result["message"]
+    
+    def health_check(self):
+        """ヘルスチェック"""
+        response = requests.get(f"{self.base_url}/health")
+        return response.json()
+
+# 使用例
+if __name__ == "__main__":
+    # MCPサーバーのURL
+    client = ImageGenMCPClient("https://your-space.hf.space")
+    
+    # ヘルスチェック
+    print("Health check:", client.health_check())
+    
+    # ツールリストの取得
+    tools = client.list_tools()
+    print("Available tools:", tools)
+    
+    # 画像生成
+    image, message = client.generate_image("美しい夕日の風景")
+    if image:
+        print(f"Success: {message}")
+        image.save("generated_image.png")
+    else:
+        print(f"Error: {message}")

requirements.txt

@@ -1,4 +1,6 @@
 gradio==5.31.0
 google-genai
-mcp
-pillow
+pillow
+fastapi
+uvicorn
+python-multipart