Update app.py

app.py

@@ -9,39 +9,62 @@ from Gradio_UI import GradioUI
 
 #Marathon trining plan tool
 @tool
-def fetch_marathon_plan(goal_time: str) -> str:
-    """Fetches a marathon training plan based on a goal time.
+def estimate_marathon_time(half_time: str) -> str:
+    """Estimates a full marathon time based on a given half-marathon time.
+    
+    Args:
+        half_time: Half-marathon completion time in HH:MM format.
+    
+    Returns:
+        Estimated full marathon time in HH:MM format.
+    """
+    try:
+        hours, minutes = map(int, half_time.split(":"))
+        total_minutes = hours * 60 + minutes
+        estimated_full_time = total_minutes * 2.1  # Common estimation factor
+        full_hours = int(estimated_full_time // 60)
+        full_minutes = int(estimated_full_time % 60)
+        return f"{full_hours}:{full_minutes:02d}"
+    except:
+        return "Invalid half-marathon time format. Please use HH:MM."
+
+@tool
+def fetch_marathon_plan(goal_time: str, gender: str, training_days: int, current_time: str = None, half_time: str = None) -> str:
+    """Fetches a marathon training plan based on goal time, gender, and weekly commitment.
     
     Args:
         goal_time: Desired marathon completion time (e.g., '4:00', '3:30').
+        gender: 'male' or 'female' (affects training pace).
+        training_days: Days per week the user can train.
+        current_time: User's current marathon time (if available).
+        half_time: User's half-marathon time (if available).
     
     Returns:
-        A link to a training plan or summary of key details.
+        A recommended training duration and plan.
     """
+    # Determine experience level
+    if current_time:
+        experience = "experienced"
+    elif half_time:
+        estimated_full = estimate_marathon_time(half_time)
+        experience = f"estimated full marathon time of {estimated_full}"
+    else:
+        experience = "beginner"
+
+    # Adjust training duration
+    if training_days >= 5:
+        weeks_needed = 8
+    else:
+        weeks_needed = 12
+
     search_tool = DuckDuckGoSearchTool()
-    query = f"marathon training plan {goal_time} site:runnersworld.com"
+    query = f"{gender} marathon training plan {goal_time} {weeks_needed}-week site:runnersworld.com"
     results = search_tool.search(query)
     
     if results:
-        return f"Here is a recommended training plan for {goal_time}: {results[0]['url']}"
+        return f"Based on your response ({experience}, {gender}, training {training_days} days/week), you need a {weeks_needed}-week plan. Here’s a suggested plan: {results[0]['url']}"
     else:
-        return f"Could not find a specific plan for {goal_time}, but you can check Runner's World for more details."
-
-@tool
-def get_current_time_in_timezone(timezone: str) -> str:
-    """A tool that fetches the current local time in a specified timezone.
-    Args:
-        timezone: A string representing a valid timezone (e.g., 'America/New_York').
-    """
-    try:
-        # Create timezone object
-        tz = pytz.timezone(timezone)
-        # Get current time in that timezone
-        local_time = datetime.datetime.now(tz).strftime("%Y-%m-%d %H:%M:%S")
-        return f"The current local time in {timezone} is: {local_time}"
-    except Exception as e:
-        return f"Error fetching time for timezone '{timezone}': {str(e)}"
-
+        return f"Could not find an exact match, but you can check Runner's World for more details."
 
 final_answer = FinalAnswerTool()