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)