Update app.py

app.py

@@ -39,11 +39,9 @@ def get_location_from_ip():
         'longitude': response.get('longitude')
     }
 
-
 class WeatherInput(BaseModel):
     city: str = Field(default=None, description="The city to get the weather for.")
 
-
 @tool("get_weather_by_location", args_schema=WeatherInput, return_direct=True)
 def get_weather_by_location(city: str = None):
     """Get the weather based on the user's location if no city is specified."""
@@ -68,29 +66,27 @@ def get_weather_by_location(city: str = None):
     }
 
     response = requests.post(url, json=payload, headers=headers).json()
-
+    
     return format_weather_response(response, city)
 
-
 def format_weather_response(weather_data, city):
     """Format the weather data into a readable string."""
     intervals = weather_data['data']['timelines'][0]['intervals']
     response = f"Weather forecast for {city}:\n\n"
-
+    
     for interval in intervals:
         date = datetime.fromisoformat(interval['startTime']).strftime("%A, %B %d")
         temp = round(interval['values']['temperature'], 1)
         humidity = round(interval['values']['humidity'], 1)
         wind_speed = round(interval['values']['windSpeed'], 1)
-
+        
         response += f"{date}:\n"
         response += f"  Temperature: {temp}°C\n"
         response += f"  Humidity: {humidity}%\n"
         response += f"  Wind Speed: {wind_speed} km/h\n\n"
-
+        
     return response
 
-
 get_weather_tool = Tool(
     name="get_weather_by_location",
     func=get_weather_by_location,
@@ -134,18 +130,18 @@ Remember to be friendly and informative in your responses, and focus on providin
 ])
 
 
-# Initialize tools and agent
-tools = [get_weather_by_location]
 
 memory = ConversationBufferMemory(memory_key="chat_history", return_messages=True)
 
+tools = [get_weather_tool]
+
 agent = create_tool_calling_agent(llm, tools, prompt=prompt)
 
 agent_executor = AgentExecutor(
     agent=agent,
     tools=tools,
     memory=memory,
-    output_parser=PydanticOutputParser(pydantic_object=WeatherOutput)
+    output_parser=parser
 )
 
 
@@ -160,11 +156,11 @@ with gr.Blocks() as demo:
     chatbot = gr.Chatbot()
     with gr.Row():
         txt = gr.Textbox(
-            show_label=False, 
-            placeholder="Ask about the weather...", 
-            lines=1, 
+            show_label=False,
+            placeholder="Ask about the weather...",
+            lines=1,
             container=False
         )
         txt.submit(gradio_interface, inputs=txt, outputs=chatbot)
 
-demo.launch()
+demo.launch(share=True)