barkha167 commited on
Commit
6b572e0
1 Parent(s): 93ca373

Upload app.py

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