Update app.py

app.py

@@ -1,21 +1,18 @@
 import os
 from dotenv import load_dotenv
 import httpx
-import streamlit as st
+import gradio as gr
 from langchain.prompts import PromptTemplate
 from langchain_huggingface import HuggingFaceEndpoint
 from langchain_core.messages import BaseMessage, HumanMessage
 from langgraph.graph import MessageGraph, END
 from typing import Sequence
 
+# Load environment variables
 load_dotenv()
 HF_TOKEN = os.getenv("HF_TOKEN")
 WEATHER_TOKEN = os.getenv("WEATHER_TOKEN")
 
-# streamlit app
-st.title("City Weather Information with AI Review")
-city = st.text_input("Enter the name of a city:")
-
 # Initialize the HuggingFace inference endpoint
 llm = HuggingFaceEndpoint(
     repo_id="mistralai/Mistral-7B-Instruct-v0.3",
@@ -25,8 +22,7 @@ llm = HuggingFaceEndpoint(
 )
 
 # Define nodes
-def fetch_weather_node(state: Sequence[BaseMessage]) -> str:
-    city = state[0].content.strip()
+def fetch_weather_node(city: str) -> str:
     url = f"https://api.openweathermap.org/data/2.5/weather?q={city}&appid={WEATHER_TOKEN}&units=metric"
     
     try:
@@ -39,15 +35,14 @@ def fetch_weather_node(state: Sequence[BaseMessage]) -> str:
     except Exception as e:
         return f"Error: {e}"
 
-def generate_review_node(state: Sequence[BaseMessage]) -> str:
-    input_text = state[0].content
-    response = llm(input_text)
+def generate_review_node(weather_info: str) -> str:
+    response = llm(weather_info)
     return response
 
 # Define the prompt template for generating weather reviews
 review_prompt_template = """
 You are an expert weather analyst. Based on the provided weather information, generate a detailed and insightful review. 
-Weather Information: {weather_info[1]}
+Weather Information: {weather_info}
 Your review should include an analysis of the weather conditions and finish in 150 words. 
 Review:
 """
@@ -67,27 +62,33 @@ builder.set_finish_point("generate_review")
 # Compile the graph
 graph = builder.compile()
 
-# Streamlit app
-if st.button("Get Weather Information and Review"):
+# Define the Gradio interface
+def get_weather_and_review(city: str) -> str:
     if city:
-        with st.spinner("Processing..."):
-            try:
-                # Prepare the input for the graph
-                weather_info = graph.invoke(HumanMessage(content=city))
-                st.write(weather_info[1].content)
-                # Generate the review using the refined prompt
-                review_input = review_prompt_template.format(weather_info=weather_info)
-                review = graph.invoke(HumanMessage(content=review_input))
-                
-                st.subheader("AI Generated Weather Review")
-                st.write(review[2].content)
-                st.subheader("Mermaid Graph")
-                st.write("Check out this [mermaid link](https://mermaid.live/) to display a graph with following data")
-                #st.write(graph.get_graph().draw_mermaid())
-                mermaid_code = graph.get_graph().draw_mermaid()
-                st.markdown(f"```mermaid\n{mermaid_code}\n```", unsafe_allow_html=True)
+        try:
+            # Prepare the input for the graph
+            weather_info = graph.invoke(HumanMessage(content=city))
+            weather_info_text = weather_info[1].content
+            
+            # Generate the review using the refined prompt
+            review_input = review_prompt_template.format(weather_info=weather_info_text)
+            review = graph.invoke(HumanMessage(content=review_input))
+            review_text = review[2].content
+
+            return f"**Weather Information:**\n{weather_info_text}\n\n**AI Generated Weather Review:**\n{review_text}"
 
-            except Exception as e:
-                st.error(f"Error generating weather review: {e}")
+        except Exception as e:
+            return f"Error generating weather review: {e}"
     else:
-        st.warning("Please enter a city name.")
+        return "Please enter a city name."
+
+interface = gr.Interface(
+    fn=get_weather_and_review,
+    inputs=gr.Textbox(lines=2, placeholder="Enter the name of a city:", label="City"),
+    outputs="text",
+    title="City Weather Information with AI Review",
+    description="Enter the name of a city to get current weather information and an AI-generated review based on that information."
+)
+
+if __name__ == "__main__":
+    interface.launch()