oarabbus/GenAITDMApp

1 2import streamlit as st3 4 5from langchain.chat_models import ChatOpenAI6from langchain.schema import (7    AIMessage,8    HumanMessage,9    SystemMessage10)11 12# From here down is all the StreamLit UI13st.set_page_config(page_title="LangChain Demo", page_icon=":robot:")14st.header("Hey, I'm your Chat GPT")15 16 17 18if "sessionMessages" not in st.session_state:19     st.session_state.sessionMessages = [20        SystemMessage(content="You are a helpful assistant.")21    ]22 23 24 25def load_answer(question):26 27    st.session_state.sessionMessages.append(HumanMessage(content=question))28 29    assistant_answer  = chat(st.session_state.sessionMessages )30 31    st.session_state.sessionMessages.append(AIMessage(content=assistant_answer.content))32 33    return assistant_answer.content34 35 36def get_text():37    input_text = st.text_input("You: ", key= input)38    return input_text39 40 41chat = ChatOpenAI(temperature=0)42 43 44 45 46user_input=get_text()47submit = st.button('Generate')  48 49if submit:50    51    response = load_answer(user_input)52    st.subheader("Answer:")53 54    st.write(response,key= 1)55 56