| import streamlit as st |
|
|
| from langchain.chat_models import ChatOpenAI |
| from langchain.schema import ( |
| AIMessage, |
| HumanMessage, |
| SystemMessage |
| ) |
|
|
| |
| st.set_page_config(page_title="Chat AI", page_icon=":robot:") |
| st.header("Hi! Ask me anything!") |
|
|
| if "sessionMessages" not in st.session_state : |
| st.session_state.sessionMessages = [ |
| SystemMessage(content="You are helpful assistant") |
| ] |
|
|
| def load_answer(question): |
| st.session_state.sessionMessages.append( |
| HumanMessage(content=question) |
| ) |
|
|
| assistant_answer = chat(st.session_state.sessionMessages) |
| st.session_state.sessionMessages.append(AIMessage(content=assistant_answer.content)) |
|
|
| return assistant_answer.content |
|
|
| def get_text(): |
| input_text = st.text_input("You:", key = input) |
| return input_text |
|
|
| chat = ChatOpenAI(temperature=0) |
|
|
| user_input = get_text() |
| submit = st.button("send") |
|
|
| if submit: |
| response = load_answer(user_input) |
| st.subheader("answer") |
| st.write(response, key=1) |
| |
