gdbanassim commited on
Commit
093b75d
·
verified ·
1 Parent(s): a378dea

Create app.py

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