Update app.py

app.py

@@ -1,34 +1,69 @@
 # app.py
+import os
 import gradio as gr
 
+# --- コア処理（ここに本来のツール処理を書く） ---
+def handle_request(q: str, user_label: str) -> str:
+    # 実際にはここから外部API呼び出し・DB操作などを行う
+    return f"[{user_label}] You said: {q}"
+
+# --- 認証＋実行：キーが無いときは guest ---
 def my_tool(q: str, dev_token: str, request: gr.Request) -> str:
-    # 1) MCP/HTTP ヘッダから取得
+    # 1) MCP/HTTPのAuthorizationヘッダ
     auth = request.headers.get("authorization")
 
-    # 2) ヘッダが無ければ UI入力欄から（開発用）
+    # 2) UIのトークン入力（開発用）
     if not auth and dev_token:
         auth = f"Bearer {dev_token.strip()}"
 
-    # 3) 最終的にキーがなければ guest として扱う
+    # 3) 環境変数（任意の開発用）
     if not auth:
-        user = "guest"
+        env_token = os.getenv("DEMO_TOKEN", "").strip()
+        if env_token:
+            auth = f"Bearer {env_token}"
+
+    # ラベル決定（guest or user）
+    if not auth or not auth.lower().startswith("bearer "):
+        user_label = "guest"
     else:
-        # "Bearer <token>" をパース
-        token = auth.split(" ", 1)[1] if " " in auth else auth
-        user = f"user:{token[:6]}..."  # トークン先頭数文字だけ表示
+        token = auth.split(" ", 1)[1].strip()
+        short = token[:6] + "..." if len(token) > 6 else token
+        user_label = f"user:{short}"
+
+    # ここから本来の処理へ
+    return handle_request(q, user_label)
 
-    return f"Hello {user}, you said: {q}"
+# ---- Gradio UI ----
+examples = [
+    ["Hello world", ""],                    # 1) guest
+    ["Check my status", "sk_test_123456"], # 2) ダミーAPIキー
+    ["Fetch HF data", "hf_dummy_token_abcdef"],  # 3) HFトークン例
+]
 
 with gr.Blocks() as demo:
-    gr.Markdown("### Gradio MCP 認証デモ")
-    q = gr.Textbox(label="Query")
+    gr.Markdown("### Gradio MCP 認証デモ（キー無し=guest）")
+    q = gr.Textbox(label="Query", placeholder="Type something")
     dev_token = gr.Textbox(
         label="API Token (開発用オプション)",
         type="password",
         placeholder="sk_xxx / hf_xxx"
     )
-    out = gr.Textbox(label="Result")
+    out = gr.Textbox(label="Result", lines=3)
     run = gr.Button("Run")
     run.click(my_tool, inputs=[q, dev_token], outputs=[out])
 
-demo.launch(mcp_server=True)
+    gr.Examples(
+        examples=examples,
+        inputs=[q, dev_token],
+        outputs=[out],
+        fn=my_tool,
+        cache_examples=False,
+    )
+
+# MCPサーバを同時起動（SSEは非推奨。/gradio_api/mcp/ を使用）
+demo.launch(
+    mcp_server=True,
+    ssr_mode=True,          # SSRログの⚡表示用（不要ならFalse）
+    server_name="0.0.0.0",
+    server_port=int(os.getenv("PORT", "7860")),
+)