Update app.py

app.py

@@ -7,7 +7,7 @@ from tools.final_answer import FinalAnswerTool
 
 from Gradio_UI import GradioUI
 
-#Marathon trining plan tool
+#Marathon training plan
 @tool
 def estimate_marathon_time(half_time: str) -> str:
     """Estimates a full marathon time based on a given half-marathon time.
@@ -29,35 +29,54 @@ def estimate_marathon_time(half_time: str) -> str:
         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 ask_missing_info() -> str:
+    """Checks which user inputs are missing and asks for them."""
+    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 "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 "Before generating your plan, I need some details:\n" + "\n".join(missing)
+    return "All information collected. Ready to generate your plan!"
+
+@tool
+def fetch_marathon_plan(goal_time: str) -> str:
+    """Generates a marathon training plan only if all required inputs are provided.
     
     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.
     """
+    # Check if any info is missing
+    missing_info = ask_missing_info()
+    if "Before generating" in missing_info:
+        return missing_info  # Ask for missing details
+
     # Determine experience level
-    if current_time:
+    if "marathon_time" in user_data:
         experience = "experienced"
-    elif half_time:
-        estimated_full = estimate_marathon_time(half_time)
+    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"
 
     # Adjust training duration
-    if training_days >= 5:
-        weeks_needed = 8
-    else:
-        weeks_needed = 12
+    training_days = user_data["training_days"]
+    weeks_needed = 8 if training_days >= 5 else 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)