Update app.py

app.py

@@ -10,18 +10,11 @@ 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.
-    """
+    """Estimates a full marathon time based on a given half-marathon time."""
     try:
         hours, minutes = map(int, half_time.split(":"))
         total_minutes = hours * 60 + minutes
-        estimated_full_time = total_minutes * 2.1  # Common estimation factor
+        estimated_full_time = total_minutes * 2.1  # Common factor for estimation
         full_hours = int(estimated_full_time // 60)
         full_minutes = int(estimated_full_time % 60)
         return f"{full_hours}:{full_minutes:02d}"
@@ -29,8 +22,8 @@ def estimate_marathon_time(half_time: str) -> str:
         return "Invalid half-marathon time format. Please use HH:MM."
 
 @tool
-def ask_missing_info() -> str:
-    """Checks which user inputs are missing and asks for them."""
+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:
@@ -43,23 +36,28 @@ def ask_missing_info() -> str:
         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 "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():
+        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():
+        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.
-    
-    Args:
-        goal_time: Desired marathon completion time (e.g., '4:00', '3:30').
-    
-    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
+    """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
 
     # Determine experience level
     if "marathon_time" in user_data: