Update app.py

app.py

@@ -9,62 +9,23 @@ from Gradio_UI import GradioUI
 
 #Marathon training plan
 @tool
-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.
+def fetch_marathon_plan(goal_time: str) -> str:
+    """Fetches a marathon training plan based on a goal time.
     
     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 recommended training duration and plan.
+        A link to a training plan or summary of key details.
     """
-    # 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"{gender} marathon training plan {goal_time} {weeks_needed}-week site:runnersworld.com"
+    query = f"marathon training plan {goal_time} site:runnersworld.com"
     results = search_tool.search(query)
     
     if results:
-        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']}"
+        return f"Here is a recommended training plan for {goal_time}: {results[0]['url']}"
     else:
-        return f"Could not find an exact match, but you can check Runner's World for more details."
+        return f"Could not find a specific plan for {goal_time}, but you can check Runner's World for more details."
 
 final_answer = FinalAnswerTool()