Update app.py

app.py

@@ -1,17 +1,19 @@
+# Running fine :)
 import os
 import httpx
 import streamlit as st
-from dotenv import load_dotenv
 from langchain.prompts import PromptTemplate
 from langchain.agents import create_react_agent, AgentExecutor
 from langchain.tools import Tool
 from langchain_huggingface import HuggingFaceEndpoint
 import urllib.parse
 
-# Load environment variables
-load_dotenv()
-OPENWEATHER_API_KEY = os.getenv("OPENWEATHER_API_KEY")
-HF_TOKEN = os.getenv("HF_TOKEN")
+st.title("City Weather Information with AI Review")
+OPENWEATHER_API_KEY = st.sidebar.text_input("Enter Weather API Key", type="password")
+st.sidebar.write("Check out this [Weather API](https://home.openweathermap.org/api_keys) to generate API key")
+HF_TOKEN = st.sidebar.text_input("Enter Hugging Face API Key", type="password")
+st.sidebar.write("Check out this [Hugging Face Token](https://huggingface.co/settings/tokens) to generate token")
+city = st.text_input("Enter the name of a city:")
 
 # Define the tool to fetch weather information
 def fetch_weather(city: str) -> dict:
@@ -19,7 +21,6 @@ def fetch_weather(city: str) -> dict:
     city = city.strip()  # Ensure no leading/trailing spaces
     encoded_city = urllib.parse.quote(city)
     url = f"https://api.openweathermap.org/data/2.5/weather?q={encoded_city}&appid={OPENWEATHER_API_KEY}&units=metric"
-    
     try:
         response = httpx.get(url)
         response.raise_for_status()
@@ -32,13 +33,7 @@ def generate_review(w_info : str) -> str:
     """Generate a review based on the weather information."""
     if "error" in w_info:
         return f"Error fetching weather data: {w_info['error']}"
-
-    # Extract weather details
-    #weather = weather_info['weather'][0]['main']
-    #temperature = weather_info['main']['temp']
-    
     input_text = f"The current weather is {w_info}. Provide a detailed review based on this information."
-    
     # Generate the review using the language model
     review = llm(input_text)
     return review
@@ -67,8 +62,7 @@ generate_review_tool = Tool(
 # Define prompts for the agents
 fetch_weather_prompt = PromptTemplate.from_template("""
 You are an agent that fetches weather information for a given city.
-City: {input} you can use {tools}
-                                                
+City: {input} you can use {tools}                                                
 Question: the input question you must answer
 Thought: you should always think about what to do
 Action: the action to take, should be one of [{tool_names}]
@@ -77,16 +71,14 @@ Observation: the result of the action
 ... (this Thought/Action/Action Input/Observation can repeat N times)
 Thought: I now know the final answer
 Final Answer: the final answer to the original input question
-
 Begin!
 Question: {input}
 Thought:
 {agent_scratchpad}
 """)
 
-
 generate_review_prompt = PromptTemplate.from_template("""
-You are an expert reviewer and use you can use {tools}
+You are an expert reviewer and use you can use {tools} for weather review
 Question: the input question you must answer
 Thought: you should always think about what to do
 Action: the action to take, should be one of [{tool_names}]
@@ -95,7 +87,6 @@ Observation: the result of the action
 ... (this Thought/Action/Action Input/Observation can repeat N times)
 Thought: I now know the final answer
 Final Answer: the final answer to the original input question
-
 Begin!
 Question: {input}
 Thought:
@@ -108,7 +99,6 @@ fetch_weather_agent = create_react_agent(
     tools=[fetch_weather_tool],
     prompt=fetch_weather_prompt
 )
-
 generate_review_agent = create_react_agent(
     llm=llm,
     tools=[generate_review_tool],
@@ -118,34 +108,21 @@ generate_review_agent = create_react_agent(
 fetch_weather_agent_executor = AgentExecutor(agent=fetch_weather_agent, tools=[fetch_weather_tool], verbose=True,handle_parsing_errors=True )
 generate_review_agent_executor = AgentExecutor(agent=generate_review_agent, tools=[generate_review_tool], verbose=True,)
 
-st.title("Weather Information and Review")
-
 # Streamlit UI
-city = st.text_input("Enter the name of a city:")
-
 if st.button("Get Weather Information and Review"):
     with st.spinner("Processing..."):
         try:
             # Fetch weather information
-            weather_info = fetch_weather_agent_executor.invoke({
-                'input': city
-            })
-          #  st.write(weather_info['output'])
+            weather_info = fetch_weather_agent_executor.invoke({'input': city})
             w_info=weather_info['output']
-            # Handle case where the response is not as expected
             if 'error' in weather_info:
                 st.error(f"Error fetching weather data: {weather_info['error']}")
             else:
                 # Generate review based on the weather information
-                review = generate_review_agent_executor.invoke({
-                    'input': w_info
-                })
-                
+                review = generate_review_agent_executor.invoke({'input': w_info})
                 st.subheader("Weather Information")
                 st.write(weather_info['output'])
-                
                 st.subheader("AI Generated Weather Review")
                 st.write(review['output'])
-                
         except Exception as e:
-            st.error(f"Error: {e}")
+            st.error(f"Error: {e}")