Upload app.py

app.py

@@ -425,7 +425,7 @@ with gr.Blocks(title="NanoBanana Gemini Image Generator V9", theme=gr.themes.Sof
         dataset_folder: str = "",
         custom_filename: str = "",
         return_image_data: bool = False
-    ):
+    ) -> dict:
         """API endpoint for image generation"""
         try:
             if not prompt:
@@ -509,52 +509,27 @@ with gr.Blocks(title="NanoBanana Gemini Image Generator V9", theme=gr.themes.Sof
             return {"error": str(e), "success": False}
 
     # API専用エンドポイントとして公開（UIには表示されない）
-    demo.api(
-        fn=api_generate,
-        inputs=[
-            gr.Textbox(label="prompt"),
-            gr.Textbox(label="gemini_api_key"),
-            gr.Textbox(label="model", default="gemini-2.5-flash-image"),
-            gr.Textbox(label="size", default="1024x1024"),
-            gr.Textbox(label="style", default="Default"),
-            gr.Checkbox(label="save_to_dataset", default=True),
-            gr.Textbox(label="dataset_folder", default=""),
-            gr.Textbox(label="custom_filename", default=""),
-            gr.Checkbox(label="return_image_data", default=False)
-        ],
-        outputs=gr.JSON(),
-        api_name="generate"
-    )
+    gr.api(api_generate, api_name="generate")
 
     # Health check endpoint
-    def api_health():
+    def api_health() -> dict:
         """Health check endpoint"""
         from datetime import datetime
         return {
             "status": "healthy",
-            "timestamp": datetime.utcnow().isoformat(),
+            "timestamp": datetime.utcnow().isoformat() + "Z",
             "version": "9.0.0",
             "available_models": AVAILABLE_MODELS
         }
 
-    demo.api(
-        fn=api_health,
-        inputs=[],
-        outputs=gr.JSON(),
-        api_name="health"
-    )
+    gr.api(api_health, api_name="health")
 
     # Models endpoint
-    def api_models():
+    def api_models() -> dict:
         """Get available models"""
         return {"models": AVAILABLE_MODELS}
 
-    demo.api(
-        fn=api_models,
-        inputs=[],
-        outputs=gr.JSON(),
-        api_name="models"
-    )
+    gr.api(api_models, api_name="models")
 
     # Footer
     gr.Markdown(