Upload app.py

app.py

@@ -229,79 +229,6 @@ def create_gradio_interface():
     
     return demo
 
-# カスタムFastAPIルートを追加
-def setup_mcp_routes(app: FastAPI):
-    """MCPエンドポイントをFastAPIアプリに追加"""
-    
-    @app.post("/api/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("/api/mcp/call_tool")
-    async def mcp_call_tool(request: Request):
-        """MCPツールを実行する"""
-        try:
-            data = await request.json()
-            tool_name = data.get("name")
-            arguments = data.get("arguments", {})
-            
-            if tool_name == "generate_image":
-                prompt = arguments.get("prompt", "")
-                
-                # 画像生成を実行
-                image, message = generate_image(prompt)
-                
-                if image:
-                    # 画像をbase64エンコード
-                    buffered = io.BytesIO()
-                    image.save(buffered, format="PNG")
-                    img_str = base64.b64encode(buffered.getvalue()).decode()
-                    
-                    return JSONResponse({
-                        "success": True,
-                        "message": message,
-                        "image_base64": img_str
-                    })
-                else:
-                    return JSONResponse({
-                        "success": False,
-                        "message": message
-                    })
-            
-            return JSONResponse({
-                "success": False,
-                "message": f"Unknown tool: {tool_name}"
-            }, status_code=400)
-            
-        except Exception as e:
-            logger.error(f"MCPエラー: {str(e)}")
-            return JSONResponse({
-                "success": False,
-                "message": f"Error: {str(e)}"
-            }, status_code=500)
-    
-    @app.get("/api/health")
-    async def health_check():
-        """ヘルスチェックエンドポイント"""
-        return {"status": "healthy", "service": "image-gen-mcp"}
 
 # メイン実行部分
 if __name__ == "__main__":
@@ -326,9 +253,79 @@ if __name__ == "__main__":
         logger.info("画像生成MCPサーバーを起動中...")
         demo = create_gradio_interface()
         
-        # Gradioアプリからunderlying FastAPIアプリを取得してMCPルートを追加
+        # Gradio内部のFastAPIアプリを取得
         app = demo.app
-        setup_mcp_routes(app)
+        
+        # MCPエンドポイントを直接追加
+        @app.post("/api/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("/api/mcp/call_tool")
+        async def mcp_call_tool(request: Request):
+            """MCPツールを実行する"""
+            try:
+                data = await request.json()
+                tool_name = data.get("name")
+                arguments = data.get("arguments", {})
+                
+                if tool_name == "generate_image":
+                    prompt = arguments.get("prompt", "")
+                    
+                    # 画像生成を実行
+                    image, message = generate_image(prompt)
+                    
+                    if image:
+                        # 画像をbase64エンコード
+                        buffered = io.BytesIO()
+                        image.save(buffered, format="PNG")
+                        img_str = base64.b64encode(buffered.getvalue()).decode()
+                        
+                        return JSONResponse({
+                            "success": True,
+                            "message": message,
+                            "image_base64": img_str
+                        })
+                    else:
+                        return JSONResponse({
+                            "success": False,
+                            "message": message
+                        })
+                
+                return JSONResponse({
+                    "success": False,
+                    "message": f"Unknown tool: {tool_name}"
+                }, status_code=400)
+                
+            except Exception as e:
+                logger.error(f"MCPエラー: {str(e)}")
+                return JSONResponse({
+                    "success": False,
+                    "message": f"Error: {str(e)}"
+                }, status_code=500)
+        
+        @app.get("/api/health")
+        async def health_check():
+            """ヘルスチェックエンドポイント"""
+            return {"status": "healthy", "service": "image-gen-mcp"}
         
         # Hugging Face Spaces用の設定
         demo.queue()