update to streamlit

app.py

@@ -1,4 +1,4 @@
-import gradio as gr
+import streamlit as st
 from modules.data_class import DataState
 from modules.tools import data_node
 from modules.nodes import chatbot_with_tools, human_node, maybe_exit_human_node, maybe_route_to_tools
@@ -33,28 +33,27 @@ graph_builder.add_edge(START, "chatbot_healthassistant")
 # Compile the graph
 graph_with_order_tools = graph_builder.compile()
 
-# Function to handle conversation
-def chat_interface(user_input, history):
-    if not history:
-        history = []
+# Streamlit UI
+st.title("LangGraph Chatbot")
+st.markdown("Chat with an AI-powered health assistant.")
 
-    # Create the initial state
-    state = DataState(messages=history, data={}, finished=False)
+# Initialize session state
+if "messages" not in st.session_state:
+    st.session_state.messages = []
 
-    # Run the chatbot
-    for output in graph_with_order_tools.stream(state):
-        response = output["messages"][-1]  # Extract the latest chatbot response
-        history.append(("User: " + user_input, "Bot: " + response))
+user_input = st.text_input("You:", key="input")
 
-    return "", history  # Return empty input and updated chat history
+if st.button("Send"):
+    if user_input:
+        # Add user input to history
+        st.session_state.messages.append(("User", user_input))
 
-# Launch Gradio UI
-iface = gr.ChatInterface(
-    fn=chat_interface,
-    title="LangGraph Chatbot",
-    description="A chatbot powered by LangGraph and hosted on Hugging Face.",
-    theme="compact"
-)
+        # Run LangGraph chatbot
+        state = DataState(messages=st.session_state.messages, data={}, finished=False)
+        for output in graph_with_order_tools.stream(state):
+            response = output["messages"][-1]  # Get the last chatbot response
+            st.session_state.messages.append(("Bot", response))
 
-if __name__ == "__main__":
-    iface.launch()
+# Display chat history
+for sender, message in st.session_state.messages:
+    st.write(f"**{sender}:** {message}")

requirements.txt

@@ -1,4 +1,4 @@
-gradio
+streamlit
 aiohappyeyeballs==2.4.4
 aiohttp==3.11.11
 aiosignal==1.3.2