Update whoop_gradio_server.py

whoop_gradio_server.py

@@ -1,4 +1,3 @@
-#!/usr/bin/env python3
 import os
 from datetime import datetime, timedelta
 import logging
@@ -17,13 +16,6 @@ def initialize_whoop_client_with_input(email, password):
     """
     Authenticates a user using the provided WHOOP email and password.
     Stores credentials in the environment variables and initializes the WhoopClient.
-
-    Args:
-        email (str): WHOOP account email.
-        password (str): WHOOP account password.
-
-    Returns:
-        str: Success or error message indicating authentication status.
     """
     global whoop_client
 
@@ -42,9 +34,6 @@ def initialize_whoop_client_with_input(email, password):
 def get_latest_cycle_gr():
     """
     Retrieves the most recent WHOOP cycle (recovery data) for the authenticated user.
-
-    Returns:
-        str: Summary of latest recovery data or an error message.
     """
     if not whoop_client:
         return "❌ Not authenticated."
@@ -66,12 +55,6 @@ def get_latest_cycle_gr():
 def get_average_strain_gr(days):
     """
     Calculates the average strain over a given number of past days.
-
-    Args:
-        days (int): Number of days to include in the average.
-
-    Returns:
-        str: Average strain value or an error message.
     """
     if not whoop_client:
         return "❌ Not authenticated."
@@ -93,15 +76,51 @@ def get_average_strain_gr(days):
     except Exception as e:
         return f"❌ Error: {e}"
 
-def format_latest_cycle(raw_text):
+def get_workouts_gr():
     """
-    Formats the raw text returned by get_latest_cycle_gr for display.
+    Fetches recent workouts for the authenticated user.
+    """
+    if not whoop_client:
+        return "❌ Not authenticated."
+
+    try:
+        end_date = datetime.now().strftime("%Y-%m-%d")
+        start_date = (datetime.now() - timedelta(days=10)).strftime("%Y-%m-%d")
+        workouts = whoop_client.get_workout_collection(start_date, end_date)
 
-    Args:
-        raw_text (str): Raw output from get_latest_cycle_gr.
+        if not workouts:
+            return "⚠️ No workout data available."
 
-    Returns:
-        Tuple[str, str]: A short summary and detailed cycle data.
+        summary = "\n\n".join([f"🏋️ Workout {i+1}:\nType: {w.get('sport_id')}, Avg HR: {w.get('avg_heart_rate')}, Calories: {w.get('kilojoule')} KJ"
+                               for i, w in enumerate(workouts)])
+        return summary
+    except Exception as e:
+        return f"❌ Error fetching workouts: {e}"
+
+def get_sleeps_gr():
+    """
+    Fetches recent sleep records for the authenticated user.
+    """
+    if not whoop_client:
+        return "❌ Not authenticated."
+
+    try:
+        end_date = datetime.now().strftime("%Y-%m-%d")
+        start_date = (datetime.now() - timedelta(days=10)).strftime("%Y-%m-%d")
+        sleeps = whoop_client.get_sleep_collection(start_date, end_date)
+
+        if not sleeps:
+            return "⚠️ No sleep data available."
+
+        summary = "\n\n".join([f"😴 Sleep {i+1}:\nScore: {s.get('score', {}).get('sleep_score')}, Duration: {s.get('sleep_duration') // 60} min"
+                               for i, s in enumerate(sleeps)])
+        return summary
+    except Exception as e:
+        return f"❌ Error fetching sleeps: {e}"
+
+def format_latest_cycle(raw_text):
+    """
+    Formats the raw text returned by get_latest_cycle_gr for display.
     """
     if raw_text.startswith("❌") or raw_text.startswith("⚠️"):
         return raw_text, ""
@@ -139,10 +158,22 @@ with gr.Blocks(title="Whoop API Explorer") as demo:
             strain_button = gr.Button("Calculate Average Strain")
         average_strain = gr.Label(label="Average Strain")
 
+    with gr.Group():
+        gr.Markdown("## 🏋️ Workout Summary")
+        workout_button = gr.Button("Fetch Recent Workouts")
+        workout_output = gr.Textbox(label="Workout Summary", lines=10)
+
+    with gr.Group():
+        gr.Markdown("## 😴 Sleep Summary")
+        sleep_button = gr.Button("Fetch Recent Sleeps")
+        sleep_output = gr.Textbox(label="Sleep Summary", lines=10)
+
     # Bind actions
     auth_button.click(fn=initialize_whoop_client_with_input, inputs=[email_input, password_input], outputs=auth_output)
     cycle_button.click(fn=lambda: format_latest_cycle(get_latest_cycle_gr()), outputs=[latest_recovery, cycle_details])
     strain_button.click(fn=get_average_strain_gr, inputs=days_input, outputs=average_strain)
+    workout_button.click(fn=get_workouts_gr, outputs=workout_output)
+    sleep_button.click(fn=get_sleeps_gr, outputs=sleep_output)
 
 # Launch app
 if __name__ == "__main__":