captain-awesome's picture
Update app.py
6ad6f7e verified
raw
history blame
1.35 kB
import streamlit as st
from langchain_core import AIMessage, HumanMessage
def get_response(user_input):
return "I dont know"
# app config
st.set_page_config(page_title= "Chat with Websites", page_icon="🤖")
st.title("Chat with Websites")
if "chat_history" not in st.session_state:
st.session_state.chat_history = [
AIMessage(content = "Hello, I am a bot. How can I help you"),
]
#sidebar
with st.sidebar:
st.header("Settings")
website_url = st.text_input("Website URL")
openai_apikey = st.text_input("Enter your OpenAI API key")
#user_input
user_query = st.chat_input("Type your message here...")
if user_query is not None and user_query !="":
response = get_response(user_query)
st.session_state.chat_history .append(HumanMessage(content=user_query))
st.session_state.chat_history .append(AIMessageMessage(content=response))
#conversation
for message in st.session_state.chat_history:
if isinstance(message, AIMessage): # checking if the messsage is the instance of an AI message
with st.chat_message("AI"):
st.write(message.content)
elif isinstance(message, HumanMessage): # checking if the messsage is the instance of a Human
with st.chat_message("Human"):
st.write(message.content)