Update app.py

app.py

@@ -8,34 +8,13 @@ from tools.final_answer import FinalAnswerTool
 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.
-
-    Parameters:
-        half_time (str): The recorded half-marathon time in the format 'HH:MM'.
-
-    Returns:
-        str: The 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 ValueError:
-        return "Invalid half-marathon time format. Please use 'HH:MM'."
-
 @tool
 def check_missing_info() -> str:
     """Checks if required info is missing and asks for it."""
     missing = []
     
-    if "marathon_time" not in user_data and "half_marathon_time" not in user_data:
-        missing.append("What is your best marathon time? If you don’t have one, provide your best half-marathon time.")
+    if "marathon_time" not in user_data:
+        missing.append("What is your best marathon time?")
     
     if "gender" not in user_data:
         missing.append("Are you male or female? This affects the training plan.")
@@ -53,8 +32,6 @@ def save_user_response(question: str, answer: str):
     if "marathon time" in question.lower():
         if ":" in answer:
             user_data["marathon_time"] = answer
-        elif "half" in question.lower():
-            user_data["half_marathon_time"] = answer
     elif "male" in answer.lower() or "female" in answer.lower():
         user_data["gender"] = answer.lower()
     elif answer.isdigit():
@@ -68,13 +45,7 @@ def fetch_marathon_plan(goal_time: str) -> str:
         return missing_info  # Keep asking until all details are given
 
     # Determine experience level
-    if "marathon_time" in user_data:
-        experience = "experienced"
-    elif "half_marathon_time" in user_data:
-        estimated_full = estimate_marathon_time(user_data["half_marathon_time"])
-        experience = f"estimated full marathon time of {estimated_full}"
-    else:
-        experience = "beginner"
+    experience = "experienced" if "marathon_time" in user_data else "beginner"
 
     # Adjust training duration
     training_days = user_data["training_days"]