Update app.py

app.py

@@ -9,17 +9,17 @@ 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."""
+def estimate_marathon_time(marathon_time: str) -> str:
+    """Estimates a full marathon time based on a given marathon time."""
     try:
-        hours, minutes = map(int, half_time.split(":"))
+        hours, minutes = map(int, marathon_time.split(":"))
         total_minutes = hours * 60 + minutes
         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}"
     except:
-        return "Invalid half-marathon time format. Please use HH:MM."
+        return "Invalid marathon time format. Please use HH:MM."
 
 @tool
 def check_missing_info() -> str:
@@ -39,6 +39,16 @@ def check_missing_info() -> str:
         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."""
@@ -47,10 +57,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"
-    else:
-        experience = "beginner"
+    experience = "experienced" if "marathon_time" in user_data else "beginner"
 
     # Adjust training duration
     training_days = user_data["training_days"]
@@ -67,16 +74,6 @@ def fetch_marathon_plan(goal_time: str) -> str:
     else:
         return f"Could not find an exact match, but you can check Runner's World for more details."
 
-# Collect user responses directly
-def collect_user_responses(question: str, answer: str):
-    """Collects and stores user responses in the user_data dictionary."""
-    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)
-
 final_answer = FinalAnswerTool()
 
 # If the agent does not answer, the model is overloaded, please use another model or the following Hugging Face Endpoint that also contains qwen2.5 coder: