Update app.py

app.py

@@ -9,67 +9,55 @@ from Gradio_UI import GradioUI
 
 #Marathon training plan
 @tool
-def estimate_marathon_time(marathon_time: str) -> str:
-    """
-    Estimate a full marathon time from the given marathon time.
-
-    marathon_time (str): The marathon time in "HH:MM" format (e.g., "3:45").
+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, marathon_time.split(":"))
+        hours, minutes = map(int, half_time.split(":"))
         total_minutes = hours * 60 + minutes
-        estimated_full_time = total_minutes * 2.1  # Common factor for estimation
+        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 marathon time format. Please use HH:MM."
+        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 = []
+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.
     
-    if "marathon_time" not in user_data:
-        missing.append("What is your best marathon 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).
     
-    if "gender" not in user_data:
-        missing.append("Are you male or female? This affects the training plan.")
-    
-    if "training_days" not in user_data:
-        missing.append("How many days per week can you commit to training?")
-    
-    if missing:
-        return "I need more details before creating your plan:\n" + "\n".join(missing)
-    return "All information collected. Ready to generate your plan!"
-
-@tool
-def save_user_response(question: str, answer: str):
-    """Saves user responses to memory."""
-    if "marathon time" in question.lower():
-        user_data["marathon_time"] = answer
-    elif "male" in answer.lower() or "female" in answer.lower():
-        user_data["gender"] = answer.lower()
-    elif answer.isdigit():
-        user_data["training_days"] = int(answer)
-
-@tool
-def fetch_marathon_plan(goal_time: str) -> str:
-    """Generates a marathon training plan ONLY IF all required inputs are provided."""
-    missing_info = check_missing_info()
-    if "I need more details" in missing_info:
-        return missing_info  # Keep asking until all details are given
-
+    Returns:
+        A recommended training duration and plan.
+    """
     # Determine experience level
-    experience = "experienced" if "marathon_time" in user_data else "beginner"
+    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
-    training_days = user_data["training_days"]
-    weeks_needed = 8 if training_days >= 5 else 12
+    if training_days >= 5:
+        weeks_needed = 8
+    else:
+        weeks_needed = 12
 
-    # Search for a plan
     search_tool = DuckDuckGoSearchTool()
-    gender = user_data["gender"]
     query = f"{gender} marathon training plan {goal_time} {weeks_needed}-week site:runnersworld.com"
     results = search_tool.search(query)