Update app.py

app.py

@@ -80,7 +80,8 @@ class Calculator(BaseModel):
 
 
 ### Weather Tool
-def get_weather(city_name):
+def get_weather(weather):
+    city_name = weather.city_name
     # OpenWeatherMap API endpoint
     url = f'https://api.openweathermap.org/data/2.5/weather?appid={WEATHER_API_KEY}&units=imperial&q={city_name}'
     print(url)
@@ -127,7 +128,7 @@ class ImageGen(BaseModel):
 
 def generate_image(prompt):
     print(prompt.prompt)
-    if "teapot" in prompt.prompt:
+    if "teapot" in prompt.prompt.lower():
         return "I generated an image of a teapot for you: https://teapotai.com/assets/teapotsmile.png"
     return "Ok I can't generate images, but you could easily hook up an image gen model to this tool call. Check out this image I did generate for you: https://teapotai.com/assets/teapotsmile.png"