Update app.py

app.py

@@ -1,18 +1,21 @@
 import os
 from dotenv import load_dotenv
 import httpx
-import gradio as gr
+import streamlit as st
 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",
@@ -22,7 +25,8 @@ llm = HuggingFaceEndpoint(
 )
 
 # Define nodes
-def fetch_weather_node(city: str) -> str:
+def fetch_weather_node(state: Sequence[BaseMessage]) -> str:
+    city = state[0].content.strip()
     url = f"https://api.openweathermap.org/data/2.5/weather?q={city}&appid={WEATHER_TOKEN}&units=metric"
     
     try:
@@ -35,14 +39,15 @@ def fetch_weather_node(city: str) -> str:
     except Exception as e:
         return f"Error: {e}"
 
-def generate_review_node(weather_info: str) -> str:
-    response = llm(weather_info)
+def generate_review_node(state: Sequence[BaseMessage]) -> str:
+    input_text = state[0].content
+    response = llm(input_text)
     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}
+Weather Information: {weather_info[1]}
 Your review should include an analysis of the weather conditions and finish in 150 words. 
 Review:
 """
@@ -62,33 +67,29 @@ builder.set_finish_point("generate_review")
 # Compile the graph
 graph = builder.compile()
 
-# Define the Gradio interface
-def get_weather_and_review(city: str) -> str:
+# Streamlit app
+if st.button("Get Weather Information and Review"):
     if city:
-        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}"
+        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)
 
-        except Exception as e:
-            return f"Error generating weather review: {e}"
+            except Exception as e:
+                st.error(f"Error generating weather review: {e}")
     else:
-        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()
+        st.warning("Please enter a city name.")