Add analytics feature to track daily request counts

- Add analytics.py module for request tracking with file-based storage
- Integrate request logging into search_web function
- Add bar chart visualization showing last 30 days of requests
- Add filelock dependency for thread-safe file operations

🤖 Generated with [Claude Code](https://claude.ai/code)

Co-Authored-By: Claude <[email protected]>

analytics.py

@@ -0,0 +1,38 @@
+# ─── analytics.py ──────────────────────────────────────────────────────────────
+import os
+import json
+from datetime import datetime, timedelta, timezone
+from filelock import FileLock           # pip install filelock
+import pandas as pd                     # already available in HF images
+
+COUNTS_FILE = "/data/request_counts.json"
+LOCK_FILE   = COUNTS_FILE + ".lock"
+
+def _load() -> dict:
+    if not os.path.exists(COUNTS_FILE):
+        return {}
+    with open(COUNTS_FILE) as f:
+        return json.load(f)
+
+def _save(data: dict):
+    with open(COUNTS_FILE, "w") as f:
+        json.dump(data, f)
+
+async def record_request() -> None:
+    """Increment today's counter (UTC) atomically."""
+    today = datetime.now(timezone.utc).strftime("%Y-%m-%d")
+    with FileLock(LOCK_FILE):
+        data = _load()
+        data[today] = data.get(today, 0) + 1
+        _save(data)
+
+def last_n_days_df(n: int = 30) -> pd.DataFrame:
+    """Return a DataFrame with a row for each of the past *n* days."""
+    now = datetime.now(timezone.utc)
+    with FileLock(LOCK_FILE):
+        data = _load()
+    records = []
+    for i in range(n):
+        day = (now - timedelta(days=n - 1 - i)).strftime("%Y-%m-%d")
+        records.append({"date": day, "count": data.get(day, 0)})
+    return pd.DataFrame(records)

app.py

@@ -9,6 +9,7 @@ from dateutil import parser as dateparser
 from limits import parse
 from limits.aio.storage import MemoryStorage
 from limits.aio.strategies import MovingWindowRateLimiter
+from analytics import record_request, last_n_days_df
 
 # Configuration
 SERPER_API_KEY = os.getenv("SERPER_API_KEY")
@@ -25,6 +26,7 @@ rate_limit = parse("360/hour")
 async def search_web(
     query: str, search_type: str = "search", num_results: Optional[int] = 4
 ) -> str:
+    await record_request()
     """
     Search the web for information or fresh news, returning extracted content.
 
@@ -184,6 +186,12 @@ async def search_web(
         return f"Error occurred while searching: {str(e)}. Please try again or check your query."
 
 
+async def search_and_log(query, search_type, num_results):
+    text = await search_web(query, search_type, num_results)
+    chart_df = last_n_days_df()        # default = 30 days
+    return text, chart_df
+
+
 # Create Gradio interface
 with gr.Blocks(title="Web Search MCP Server") as demo:
     gr.HTML(
@@ -259,6 +267,17 @@ with gr.Blocks(title="Web Search MCP Server") as demo:
         info="The extracted article content will appear here",
     )
 
+    # ════════════════════════════  NEW  ════════════════════════════
+    requests_plot = gr.BarPlot(
+        value=last_n_days_df(),     # initial empty/old data
+        x="date",
+        y="count",
+        title="Requests per Day (Last 30 days)",
+        tooltip="all",
+        height=320,                 # tweak to taste
+    )
+    # ═══════════════════════════════════════════════════════════════
+
     # Add examples
     gr.Examples(
         examples=[
@@ -275,9 +294,9 @@ with gr.Blocks(title="Web Search MCP Server") as demo:
     )
 
     search_button.click(
-        fn=search_web,
+        fn=search_and_log,                          # wrapper
         inputs=[query_input, search_type_input, num_results_input],
-        outputs=output,
+        outputs=[output, requests_plot],            # update both
     )
 
 

requirements.txt

@@ -2,4 +2,5 @@ gradio
 httpx
 trafilatura
 python-dateutil
-limits
+limits
+filelock