allow user input to decide API key in-session

app.py

@@ -1,83 +1,109 @@
 """
-Gradio Space – Brawl Stars battle-log viewer
-Now shows the server’s public IP so you can whitelist it
+MCP-ready Gradio app
+  • Tool 1: save_brawlstars_key(api_key)        → str
+  • Tool 2: get_recent_battles(player_tag)      → dict
+UI: two tabs so humans can still use the app.
 """
-import os, socket, requests, gradio as gr, brawlstats
+import gradio as gr
+import brawlstats                      # pip install brawlstats
 
+# ───────────────────────────────────────────────
+# GLOBAL STATE  (shared by all tool invocations)
+# ───────────────────────────────────────────────
+BS_API_KEY = None           # will hold the token after Tool-1 runs
 
-# ────────────────────────────────────────────────────────────────
-# 0  Utility – discover the IP our outbound requests will use
-# ────────────────────────────────────────────────────────────────
-def get_public_ip() -> str:
+
+# ───────────────────────────────────────────────
+# TOOL 1  –  save the key (no echo)
+# ───────────────────────────────────────────────
+def save_brawlstars_key(api_key: str) -> str:
     """
-    Try https://api.ipify.org first (tiny plain-text service).
-    Fall back to http://ifconfig.me or container interface.
+    Store a Brawl Stars API key in server memory for later use.
+
+    This should be the **first** tool called in a session.  Once the key
+    is saved, subsequent calls to `get_recent_battles` can access the API
+    without requiring the LLM (or user) to resend the long token.
+
+    Args:
+        api_key (str): Your personal token generated at
+            https://developer.brawlstars.com.  Do **not** share this
+            publicly—send it only once via this tool.
+
+    Returns:
+        str: Status message —
+            • "✅ API key saved in server memory" on success  
+            • "❌ No key provided" if the argument was empty
     """
-    for url in ("https://api.ipify.org", "https://ifconfig.me/ip"):
-        try:
-            resp = requests.get(url, timeout=3)
-            if resp.ok:
-                return resp.text.strip()
-        except requests.RequestException:
-            pass
-    # last-ditch: internal docker address (may be 10.x/172.x)
-    return socket.gethostbyname(socket.gethostname())
+    global BS_API_KEY
+    BS_API_KEY = api_key.strip()
+    if BS_API_KEY:
+        return "✅ API key saved in server memory"
+    return "❌ No key provided"
 
 
-SERVER_IP = get_public_ip()
+# ───────────────────────────────────────────────
+# TOOL 2  –  fetch recent battles using saved key
+# ───────────────────────────────────────────────
+def get_recent_battles(player_tag: str) -> dict:
+    """
+    Retrieve a player's 25 most-recent Brawl Stars battles.
 
+    Call `save_brawlstars_key` **once** beforehand to cache the API key.
 
-# ────────────────────────────────────────────────────────────────
-# 1  Gradio callback
-# ────────────────────────────────────────────────────────────────
-def recent_battles(api_key: str, player_tag: str) -> dict:
-    api_key = api_key.strip()
-    tag     = player_tag.strip().upper().lstrip("#")
+    Args:
+        player_tag (str): The player’s in-game tag, with or without the
+            leading '#', e.g. "#V2LQY9UY" or "V2LQY9UY".
 
-    if not api_key:
-        return {"error": "API key is empty!"}
+    Returns:
+        dict: On success –
+            {
+              "player":   "#TAG",
+              "count":    <int>,            # number of battles returned
+              "battles":  <list[dict]>      # raw JSON from Supercell
+            }
 
+            On failure –
+            { "error": "<human-readable message>" }
+    """
+    if not BS_API_KEY:
+        return {"error": "API key not set - call save_brawlstars_key first"}
+
+    tag = player_tag.strip().upper().lstrip("#")
     try:
-        client = brawlstats.Client(api_key)
-        logs   = client.get_battle_logs(tag)
+        client  = brawlstats.Client(BS_API_KEY)
+        logs    = client.get_battle_logs(tag)           # BattleLog object
     except brawlstats.Forbidden:
         return {"error": "Invalid API key"}
     except brawlstats.NotFound:
         return {"error": f"Player #{tag} not found"}
     except brawlstats.RequestError as e:
-        return {"error": f"Request failed: {e}"}
+        return {"error": f"API request failed: {e}"}
 
     return {
-        "server_ip": SERVER_IP,
-        "player":    f"#{tag}",
-        "count":     len(logs),
-        "battles":   logs.raw_data
+        "player":   f"#{tag}",
+        "count":    len(logs),
+        "battles":  logs.raw_data                      # serialisable list
     }
 
 
-# ────────────────────────────────────────────────────────────────
-# 2  Interface
-# ────────────────────────────────────────────────────────────────
-with gr.Blocks(title="Brawl Stars – Recent Matches") as demo:
-    gr.Markdown(
-        f"### Server public IP: `{SERVER_IP}`  \n"
-        "Add this address to **Allowed IPs** when you generate a Brawl Stars API key."
-    )
-
-    api_box = gr.Textbox(
-        label="API Key",
-        type="password",
-        placeholder="Paste your API token…"
-    )
-    tag_box = gr.Textbox(
-        label="Player Tag",
-        placeholder="#V2LQY9UY"
-    )
-    out_json = gr.JSON(label="Battle Log")
-
-    api_box.change(lambda *_: None, inputs=None, outputs=out_json)
-    tag_box.submit(recent_battles, [api_box, tag_box], out_json)
+# ───────────────────────────────────────────────
+# HUMAN-FRIENDLY UI  (optional)
+# ───────────────────────────────────────────────
+with gr.Blocks(title="Brawl Stars MCP Tools") as demo:
+    with gr.Tab("🔑 Save API Key"):
+        api_box   = gr.Textbox(type="password", label="API Key")
+        save_btn  = gr.Button("Save")
+        save_out  = gr.Textbox(label="Status", interactive=False)
+        save_btn.click(save_brawlstars_key, inputs=api_box, outputs=save_out)
 
-if __name__ == "__main__":
-    demo.launch()
+    with gr.Tab("📜 Get Recent Battles"):
+        tag_box   = gr.Textbox(label="Player Tag", placeholder="#V2LQY9UY")
+        fetch_btn = gr.Button("Fetch")
+        json_out  = gr.JSON(label="Battle Log")
+        fetch_btn.click(get_recent_battles, inputs=tag_box, outputs=json_out)
 
+# ───────────────────────────────────────────────
+# LAUNCH AS MCP SERVER
+# ───────────────────────────────────────────────
+if __name__ == "__main__":
+    demo.launch(mcp_server=True)