Upload 2 files

app.py

@@ -1,39 +1,45 @@
-#Hello! It seems like you want to import the Streamlit library in Python. Streamlit is a powerful open-source framework used for building web applications with interactive data visualizations and machine learning models. To import Streamlit, you'll need to ensure that you have it installed in your Python environment.
-#Once you have Streamlit installed, you can import it into your Python script using the import statement,
-
 import streamlit as st
+from langchain.chat_models import ChatOpenAI
+from langchain.schema import AIMessage, HumanMessage, SystemMessage
+
+# From here down is all the StreamLit UI.
+st.set_page_config(page_title="LangChain Demo", page_icon=":robot:")
+st.header("Hey, I'm your Chat GPT")
 
 
-from langchain.llms import HuggingFaceHub
+if "sessionMessages" not in st.session_state:
+    st.session_state.sessionMessages = [
+        SystemMessage(content="You are a helpful assistant.")
+    ]
 
 
-#Function to return the response
 def load_answer(question):
-    llm = HuggingFaceHub(repo_id="google/flan-t5-large")
-    answer=llm(question)
-    return answer
 
+    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
 
-#App UI starts here
-st.set_page_config(page_title="LangChain Demo", page_icon=":robot:")
-st.header("LangChain Demo")
 
-#Gets the user input
 def get_text():
-    input_text = st.text_input("You: ", key="input")
+    input_text = st.text_input("You: ", key=input)
     return input_text
 
 
+chat = ChatOpenAI(temperature=0)
+
+
 user_input = get_text()
-submit = st.button('Generate')  
+submit = st.button("Generate")
 
-#If generate button is clicked
 if submit:
 
-    st.subheader("Answer:")
-
-   
     response = load_answer(user_input)
+    st.subheader("Answer:")
 
-    st.write(response)
-
+    st.write(response, key=1)

requirements.txt

@@ -1,3 +1,5 @@
+streamlit
 langchain
-Openai
-Streamlit
+openai
+streamlit-chat
+python-dotenv