Upload 9 files

README.md

@@ -81,7 +81,8 @@ Add this configuration to your Claude Code settings:
 **Returns:**
 - `success` (boolean): Whether generation was successful
 - `message` (string): Status message
-- `image_base64` (string): Base64-encoded PNG image (if successful)
+- `image_url` (string): URL to access the generated image (if successful)
+- `image_path` (string): Local file path of the generated image (if successful)
 
 ## Requirements
 

app.py

@@ -10,9 +10,14 @@ import logging
 import traceback
 from typing import Optional
 from fastapi import FastAPI, Request
-from fastapi.responses import JSONResponse, RedirectResponse
+from fastapi.responses import JSONResponse, RedirectResponse, FileResponse
 from fastapi.middleware.cors import CORSMiddleware
+from fastapi.staticfiles import StaticFiles
 import uvicorn
+import tempfile
+import uuid
+from datetime import datetime, timedelta
+import shutil
 
 # ログ設定
 logging.basicConfig(
@@ -41,6 +46,28 @@ app.add_middleware(
     allow_headers=["*"],
 )
 
+# 画像ファイル保存用ディレクトリ
+IMAGES_DIR = tempfile.mkdtemp(prefix="mcp_images_")
+os.makedirs(IMAGES_DIR, exist_ok=True)
+
+# 静的ファイルのマウント
+app.mount("/images", StaticFiles(directory=IMAGES_DIR), name="images")
+
+# 画像ファイルのクリーンアップ（古いファイルを削除）
+def cleanup_old_images():
+    """24時間以上経過した画像ファイルを削除"""
+    try:
+        now = datetime.now()
+        for filename in os.listdir(IMAGES_DIR):
+            filepath = os.path.join(IMAGES_DIR, filename)
+            if os.path.isfile(filepath):
+                file_time = datetime.fromtimestamp(os.path.getmtime(filepath))
+                if now - file_time > timedelta(hours=24):
+                    os.remove(filepath)
+                    logger.info(f"古い画像を削除: {filename}")
+    except Exception as e:
+        logger.error(f"画像クリーンアップエラー: {str(e)}")
+
 # Gemini APIクライアントの初期化
 def get_gemini_client():
     api_key = os.environ.get("GEMINI_API_KEY")
@@ -210,20 +237,31 @@ async def mcp_call_tool(request: Request):
         if tool_name == "generate_image":
             prompt = arguments.get("prompt", "")
             
+            # 古い画像をクリーンアップ
+            cleanup_old_images()
+            
             # 画像生成を実行
             image, message = generate_image(prompt)
             
             if image:
-                # 画像をbase64エンコード
-                buffered = io.BytesIO()
-                image.save(buffered, format="PNG")
-                img_str = base64.b64encode(buffered.getvalue()).decode()
+                # ユニークなファイル名を生成
+                filename = f"generated_{uuid.uuid4()}.png"
+                filepath = os.path.join(IMAGES_DIR, filename)
+                
+                # 画像を保存
+                image.save(filepath, format="PNG")
+                logger.info(f"画像を保存: {filepath}")
+                
+                # Hugging Face SpacesのベースURLを取得
+                base_url = str(request.base_url).rstrip('/')
+                image_url = f"{base_url}/images/{filename}"
                 
-                logger.info("MCP画像生成成功")
+                logger.info(f"MCP画像生成成功: {image_url}")
                 return JSONResponse({
                     "success": True,
                     "message": message,
-                    "image_base64": img_str
+                    "image_url": image_url,
+                    "image_path": filepath
                 })
             else:
                 logger.warning(f"MCP画像生成失敗: {message}")

mcp_client.py

@@ -21,7 +21,7 @@ class ImageGenMCPClient:
         response = requests.post(f"{self.base_url}/mcp/list_tools")
         return response.json()
     
-    def generate_image(self, prompt: str) -> tuple[Image.Image, str]:
+    def generate_image(self, prompt: str) -> tuple[Image.Image, str, str]:
         """
         画像を生成する
         
@@ -29,7 +29,7 @@ class ImageGenMCPClient:
             prompt: 生成したい画像の説明
             
         Returns:
-            生成された画像とメッセージ
+            生成された画像、メッセージ、画像URL
         """
         payload = {
             "name": "generate_image",
@@ -42,12 +42,13 @@ class ImageGenMCPClient:
         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"]
+            # 画像URLから画像を取得
+            image_url = result["image_url"]
+            img_response = requests.get(image_url)
+            image = Image.open(io.BytesIO(img_response.content))
+            return image, result["message"], image_url
         else:
-            return None, result["message"]
+            return None, result["message"], None
     
     def health_check(self):
         """ヘルスチェック"""
@@ -67,9 +68,10 @@ if __name__ == "__main__":
     print("Available tools:", tools)
     
     # 画像生成
-    image, message = client.generate_image("美しい夕日の風景")
+    image, message, image_url = client.generate_image("美しい夕日の風景")
     if image:
         print(f"Success: {message}")
+        print(f"Image URL: {image_url}")
         image.save("generated_image.png")
     else:
         print(f"Error: {message}")