fix tool names

app.py

@@ -1,9 +1,8 @@
 from smolagents import CodeAgent, HfApiModel
-from tools.mood_to_need import mood_to_need
-from tools.need_to_destination import need_to_destination
-from tools.get_weather import get_weather
-from tools.get_flights import get_flights
-from tools.final_wrap import final_wrap
+from tools.mood_to_need import MoodToNeedTool
+from tools.need_to_destination import NeedToDestinationTool
+from tools.weather_tool import WeatherTool
+from tools.find_flight import FlightsFinderTool
 from tools.final_answer import FinalAnswerTool
 from Gradio_UI import GradioUI
 import yaml
@@ -23,12 +22,11 @@ with open("prompts.yaml", "r") as f:
 agent = CodeAgent(
     model=model,
     tools=[
-        mood_to_need,          # Step 1: Mood → Need
-        need_to_destination,   # Step 2: Need → Destination
-        get_weather,           # Step 3: Weather for destination
-        get_flights,           # Step 4: Destination → Flights
-        final_wrap,            # Step 5: Claude wrap
-        FinalAnswerTool()      # Required final output
+        MoodToNeedTool(),          # Step 1: Mood → Need
+        NeedToDestinationTool(),   # Step 2: Need → Destination
+        WeatherTool(),             # Step 3: Weather for destination
+        FlightsFinderTool(),       # Step 4: Destination → Flights           # Step 5: Claude wrap
+        FinalAnswerTool()       # Required final output
     ],
     max_steps=6,
     verbosity_level=1,