Update app.py

app.py

@@ -9,29 +9,17 @@ from Gradio_UI import GradioUI
 
 # Below is an example of a tool that does nothing. Amaze us with your creativity !
 @tool
-def current_day(weekday:str)-> str: #it's import to specify the return type
+def capital_country(country:str)-> str: #it's import to specify the return type
     #Keep this format for the description / args / args description but feel free to modify the tool
-    """A tool that does nothing yet 
+    """A tool that give the name of capital of Poland
     Args:
-        weekday: 
+        country: 
        
     """
-    dni_tygodnia = {
-        0: "Poniedziałek",
-        1: "Wtorek", 
-        2: "Środa",
-        3: "Czwartek",
-        4: "Piątek",
-        5: "Sobota",
-        6: "Niedziela"
-    }
-    
-    # Pobieramy aktualną datę
-    dzisiaj = datetime.now()
+
     
-    # Pobieramy numer dnia tygodnia (0=poniedziałek, 6=niedziela)
-    numer_dnia = dzisiaj.weekday()
-    return f"Dzisiaj jest {dni_tygodnia[numer_dnia]}"
+
+    return f"Stolicą Polski jest Warszawa."
 
 @tool
 def get_current_time_in_timezone(timezone: str) -> str:
@@ -70,7 +58,7 @@ with open("prompts.yaml", 'r') as stream:
     
 agent = CodeAgent(
     model=model,
-    tools=[final_answer, get_current_time_in_timezone, current_day], ## add your tools here (don't remove final answer)
+    tools=[final_answer, get_current_time_in_timezone, capital_country], ## add your tools here (don't remove final answer)
     max_steps=6,
     verbosity_level=1,
     grammar=None,