import os
import streamlit as st
from langchain.schema import HumanMessage, SystemMessage, AIMessage
from langchain_community.chat_models import ChatOpenAI

st.set_page_config(page_title="Convorsational QA Chatbot")
st.header("Hey, Type your queries")

from dotenv import load_dotenv
load_dotenv()

chat = ChatOpenAI(temperature=0.5)

if 'flownMessages' not in st.session_state:
   st.session_state['flownMessages'] = [
      SystemMessage(content="You are a simple therapist")
   ]

def get_chatmodel_response(query):
   st.session_state['flownMessages'].append(HumanMessage(content=query))
   answer = chat(st.session_state['flownMessages'])
   st.session_state['flownMessages'].append(AIMessage(answer.content))
   return answer.content

input = st.text_input("Input: ", key="input")
response = get_chatmodel_response(input)

submit = st.button("Ask the question")

if submit:
   st.subheader("The Response is: ")
   st.write(response)