Rohini08 commited on
Commit
92f96fc
·
verified ·
1 Parent(s): 72d7503

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +53 -0
app.py CHANGED
@@ -0,0 +1,53 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import streamlit as st
2
+
3
+ #os.environ["OPENAI_API_KEY"] = "sk-None-afwHPTfbX723JEK8wq1bT3BlbkFJShAGH88BOZbVLhvIFQBr"
4
+ from langchain_openai import ChatOpenAI
5
+ from langchain.schema import (
6
+ AIMessage,
7
+ HumanMessage,
8
+ SystemMessage
9
+ )
10
+
11
+ # From here down is all the StreamLit UI
12
+ st.set_page_config(page_title="LangChain Demo", page_icon=":robot:")
13
+ st.header("Hey, I'm your Chat GPT")
14
+
15
+
16
+
17
+ if "sessionMessages" not in st.session_state:
18
+ st.session_state.sessionMessages = [
19
+ SystemMessage(content="You are a helpful assistant.")
20
+ ]
21
+
22
+
23
+
24
+ def load_answer(question):
25
+
26
+ st.session_state.sessionMessages.append(HumanMessage(content=question))
27
+
28
+ assistant_answer = chat.invoke(st.session_state.sessionMessages )
29
+
30
+ st.session_state.sessionMessages.append(AIMessage(content=assistant_answer.content))
31
+
32
+ return assistant_answer.content
33
+
34
+
35
+ def get_text():
36
+ input_text = st.text_input("You: ")
37
+ return input_text
38
+
39
+
40
+ chat = ChatOpenAI(temperature=0)
41
+
42
+
43
+
44
+
45
+ user_input=get_text()
46
+ submit = st.button('Generate')
47
+
48
+ if submit:
49
+
50
+ response = load_answer(user_input)
51
+ st.subheader("Answer:")
52
+
53
+ st.write(response)