Update app.py

app.py

@@ -3,9 +3,7 @@ import gradio as gr
 import requests
 import inspect
 import pandas as pd
-from dateutil import parser as date_parser
-import calendar
-import re
+from agent import agent_executor
 
 # (Keep Constants as is)
 # --- Constants ---
@@ -15,27 +13,15 @@ DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 # ----- THIS IS WERE YOU CAN BUILD WHAT YOU WANT ------
 class BasicAgent:
     def __init__(self):
-        print("Tool-based Agent initialized (Math + Date)")
+        print("LLM Tool-Enhanced Agent initialized.")
 
     def __call__(self, question: str) -> str:
-        question_lower = question.lower()
-
-        # 1. Mathefragen erkennen
-        if any(word in question_lower for word in ["calculate", "sum", "add", "subtract", "multiply", "divide", "+", "-", "*", "/"]):
-            try:
-                return self.solve_math(question)
-            except Exception as e:
-                print(f"[Math Error] {e}")
-
-        # 2. Datumsfragen erkennen
-        if "what day" in question_lower or "which day" in question_lower or "day of the week" in question_lower:
-            try:
-                return self.solve_date(question)
-            except Exception as e:
-                print(f"[Date Error] {e}")
-
-        # 3. Fallback
-        return "I don't know."
+        try:
+            result = agent_executor.run(question)
+            return result.strip()
+        except Exception as e:
+            print(f"Agent error: {e}")
+            return "I don't know."
 
     def solve_math(self, question: str) -> str:
         print(f"Solving math: {question}")