Upload whoop_gradio_server.py

whoop_gradio_server.py

@@ -0,0 +1,116 @@
+#!/usr/bin/env python3
+import os
+from pathlib import Path
+from datetime import datetime, timedelta
+import logging
+
+import gradio as gr
+from dotenv import load_dotenv
+from whoop import WhoopClient
+
+# Configure logging
+logging.basicConfig(level=logging.INFO)
+logger = logging.getLogger(__name__)
+
+# Global Whoop client
+whoop_client = None
+
+def initialize_whoop_client():
+    global whoop_client
+    env_path = Path(__file__).parent / 'config' / '.env'
+
+    if not env_path.exists():
+        return "❌ .env file not found."
+
+    load_dotenv(dotenv_path=env_path)
+    email = os.getenv("WHOOP_EMAIL")
+    password = os.getenv("WHOOP_PASSWORD")
+
+    if not email or not password:
+        return "❌ Missing WHOOP_EMAIL or WHOOP_PASSWORD in environment variables."
+
+    try:
+        whoop_client = WhoopClient(username=email, password=password)
+        return "✅ Successfully authenticated with Whoop."
+    except Exception as e:
+        return f"❌ Authentication failed: {e}"
+
+def get_latest_cycle_gr():
+    if not whoop_client:
+        return "❌ Not authenticated."
+
+    try:
+        end_date = datetime.now().strftime("%Y-%m-%d")
+        start_date = (datetime.now() - timedelta(days=1)).strftime("%Y-%m-%d")
+        cycles = whoop_client.get_cycle_collection(start_date, end_date)
+        if not cycles:
+            return "⚠️ No cycle data available."
+
+        latest = cycles[0]
+        score = latest.get("score", {}).get("recovery", "N/A")
+        return f"🌀 Latest Cycle:\nRecovery Score: {score}\n\nFull Data:\n{latest}"
+    except Exception as e:
+        return f"❌ Error: {e}"
+
+def get_average_strain_gr(days):
+    if not whoop_client:
+        return "❌ Not authenticated."
+
+    try:
+        end_date = datetime.now().strftime("%Y-%m-%d")
+        start_date = (datetime.now() - timedelta(days=int(days))).strftime("%Y-%m-%d")
+        cycles = whoop_client.get_cycle_collection(start_date, end_date)
+
+        if not cycles:
+            return "⚠️ No cycle data available."
+
+        strains = [c['score']['strain'] for c in cycles if c.get('score') and c['score'].get('strain') is not None]
+        if not strains:
+            return "⚠️ No strain data available."
+
+        avg = sum(strains) / len(strains)
+        return f"📊 Average Strain over {days} days: {avg:.2f} (from {len(strains)} entries)"
+    except Exception as e:
+        return f"❌ Error: {e}"
+
+def get_auth_status_gr():
+    if not whoop_client:
+        return "🔒 Not authenticated."
+
+    try:
+        profile = whoop_client.get_profile()
+        return f"🔓 Authenticated.\nUser: {profile.get('name', 'Unknown')}\nEmail: {profile.get('email', 'Unknown')}"
+    except Exception as e:
+        return f"❌ Error checking auth: {e}"
+
+# UI definition
+with gr.Blocks(title="Whoop API Explorer") as demo:
+    gr.Markdown("# 🧠 WHOOP API Dashboard")
+    gr.Markdown("Authenticate and explore your recovery and strain data.")
+
+    with gr.Row():
+        auth_output = gr.Textbox(label="Auth Status", interactive=False)
+        auth_button = gr.Button("Authenticate with Whoop")
+
+    with gr.Row():
+        cycle_output = gr.Textbox(label="Latest Cycle", lines=10, interactive=False)
+        cycle_button = gr.Button("Get Latest Cycle")
+
+    with gr.Row():
+        days_input = gr.Number(value=7, label="Days for Avg Strain")
+        strain_output = gr.Textbox(label="Average Strain", lines=4, interactive=False)
+        strain_button = gr.Button("Get Average Strain")
+
+    with gr.Row():
+        status_output = gr.Textbox(label="Whoop Profile Info", lines=4, interactive=False)
+        status_button = gr.Button("Check Auth Status")
+
+    # Bind actions
+    auth_button.click(fn=initialize_whoop_client, outputs=auth_output)
+    cycle_button.click(fn=get_latest_cycle_gr, outputs=cycle_output)
+    strain_button.click(fn=get_average_strain_gr, inputs=days_input, outputs=strain_output)
+    status_button.click(fn=get_auth_status_gr, outputs=status_output)
+
+# Launch app
+if __name__ == "__main__":
+    demo.launch(mcp_server=True)