Update app.py

app.py

@@ -8,6 +8,19 @@ 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."""
+    try:
+        hours, minutes = map(int, half_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."
+
 @tool
 def check_missing_info() -> str:
     """Checks if required info is missing and asks for it."""
@@ -26,27 +39,6 @@ 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.
-
-    This function saves the user's responses to the memory based on the question asked.
-
-    Args:
-        question (str): A string representing the question asked to the user. For example, "What is your marathon time?".
-        answer (str): A string representing the user's response. For example, "3:30" or "Male".
-
-    Example:
-        save_user_response("What is your marathon time?", "3:30")
-    """
-    if "marathon time" in question.lower():
-        if ":" in answer:
-            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."""
@@ -55,7 +47,10 @@ def fetch_marathon_plan(goal_time: str) -> str:
         return missing_info  # Keep asking until all details are given
 
     # Determine experience level
-    experience = "experienced" if "marathon_time" in user_data else "beginner"
+    if "marathon_time" in user_data:
+        experience = "experienced"
+    else:
+        experience = "beginner"
 
     # Adjust training duration
     training_days = user_data["training_days"]
@@ -72,6 +67,16 @@ 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: