Spaces:
Sleeping
Sleeping
use my api key
Browse files
app.py
CHANGED
@@ -1,82 +1,75 @@
|
|
|
|
|
|
1 |
import streamlit as st
|
2 |
-
from streamlit_chat import message
|
3 |
from langchain import OpenAI
|
4 |
from langchain.chains import ConversationChain
|
5 |
-
from langchain.chains.conversation.memory import
|
6 |
-
|
7 |
-
|
8 |
-
|
9 |
-
|
10 |
-
|
11 |
-
|
12 |
-
|
13 |
-
|
14 |
-
|
15 |
-
|
16 |
-
st.session_state['API_Key'] =''
|
17 |
|
18 |
# Setting page title and header
|
19 |
st.set_page_config(page_title="Chat GPT Clone", page_icon=":robot_face:")
|
20 |
-
st.markdown(
|
|
|
|
|
|
|
21 |
|
22 |
|
23 |
st.sidebar.title("😎")
|
24 |
-
st.session_state['API_Key']= st.sidebar.text_input("What's your API key?",type="password")
|
25 |
summarise_button = st.sidebar.button("Summarise the conversation", key="summarise")
|
26 |
if summarise_button:
|
27 |
-
summarise_placeholder = st.sidebar.write(
|
28 |
-
|
29 |
-
|
30 |
-
|
31 |
-
#os.environ["OPENAI_API_KEY"] = "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"
|
32 |
-
|
33 |
-
def getresponse(userInput, api_key):
|
34 |
-
|
35 |
-
if st.session_state['conversation'] is None:
|
36 |
|
|
|
|
|
37 |
llm = OpenAI(
|
38 |
temperature=0,
|
39 |
-
openai_api_key=
|
40 |
-
model_name=
|
41 |
)
|
42 |
|
43 |
-
st.session_state[
|
44 |
-
llm=llm,
|
45 |
-
verbose=True,
|
46 |
-
memory=ConversationSummaryMemory(llm=llm)
|
47 |
)
|
48 |
|
49 |
-
response=st.session_state[
|
50 |
-
print(st.session_state[
|
51 |
-
|
52 |
|
53 |
return response
|
54 |
|
55 |
|
56 |
-
|
57 |
response_container = st.container()
|
58 |
-
# Here we will have a container for user input text box
|
59 |
container = st.container()
|
60 |
|
61 |
|
62 |
with container:
|
63 |
-
with st.form(key=
|
64 |
-
user_input = st.text_area("Your question goes here:", key=
|
65 |
-
submit_button = st.form_submit_button(label=
|
66 |
|
67 |
if submit_button:
|
68 |
-
st.session_state[
|
69 |
-
model_response=getresponse(user_input
|
70 |
-
st.session_state[
|
71 |
-
|
72 |
|
73 |
with response_container:
|
74 |
-
for i in range(len(st.session_state[
|
75 |
-
|
76 |
-
|
77 |
-
|
78 |
-
|
79 |
-
|
80 |
-
|
81 |
-
|
82 |
-
|
|
|
1 |
+
import os
|
2 |
+
|
3 |
import streamlit as st
|
|
|
4 |
from langchain import OpenAI
|
5 |
from langchain.chains import ConversationChain
|
6 |
+
from langchain.chains.conversation.memory import ConversationSummaryMemory
|
7 |
+
from streamlit_chat import message
|
8 |
+
from dotenv import load_dotenv
|
9 |
+
|
10 |
+
|
11 |
+
load_dotenv()
|
12 |
+
|
13 |
+
if "conversation" not in st.session_state:
|
14 |
+
st.session_state["conversation"] = None
|
15 |
+
if "messages" not in st.session_state:
|
16 |
+
st.session_state["messages"] = []
|
|
|
17 |
|
18 |
# Setting page title and header
|
19 |
st.set_page_config(page_title="Chat GPT Clone", page_icon=":robot_face:")
|
20 |
+
st.markdown(
|
21 |
+
"<h1 style='text-align: center;'>How can I assist you? </h1>",
|
22 |
+
unsafe_allow_html=True,
|
23 |
+
)
|
24 |
|
25 |
|
26 |
st.sidebar.title("😎")
|
|
|
27 |
summarise_button = st.sidebar.button("Summarise the conversation", key="summarise")
|
28 |
if summarise_button:
|
29 |
+
summarise_placeholder = st.sidebar.write(
|
30 |
+
"Nice chatting with you my friend ❤️:\n\n"
|
31 |
+
+ st.session_state["conversation"].memory.buffer
|
32 |
+
)
|
|
|
|
|
|
|
|
|
|
|
33 |
|
34 |
+
def getresponse(userInput: str):
|
35 |
+
if st.session_state["conversation"] is None:
|
36 |
llm = OpenAI(
|
37 |
temperature=0,
|
38 |
+
openai_api_key=os.getenv("OPENAI_API_KEY"),
|
39 |
+
model_name="text-davinci-003", # we can also use 'gpt-3.5-turbo'
|
40 |
)
|
41 |
|
42 |
+
st.session_state["conversation"] = ConversationChain(
|
43 |
+
llm=llm, verbose=True, memory=ConversationSummaryMemory(llm=llm)
|
|
|
|
|
44 |
)
|
45 |
|
46 |
+
response = st.session_state["conversation"].predict(input=userInput)
|
47 |
+
print(st.session_state["conversation"].memory.buffer)
|
|
|
48 |
|
49 |
return response
|
50 |
|
51 |
|
|
|
52 |
response_container = st.container()
|
|
|
53 |
container = st.container()
|
54 |
|
55 |
|
56 |
with container:
|
57 |
+
with st.form(key="my_form", clear_on_submit=True):
|
58 |
+
user_input = st.text_area("Your question goes here:", key="input", height=100)
|
59 |
+
submit_button = st.form_submit_button(label="Send")
|
60 |
|
61 |
if submit_button:
|
62 |
+
st.session_state["messages"].append(user_input)
|
63 |
+
model_response = getresponse(user_input)
|
64 |
+
st.session_state["messages"].append(model_response)
|
|
|
65 |
|
66 |
with response_container:
|
67 |
+
for i in range(len(st.session_state["messages"])):
|
68 |
+
if (i % 2) == 0:
|
69 |
+
message(
|
70 |
+
st.session_state["messages"][i],
|
71 |
+
is_user=True,
|
72 |
+
key=str(i) + "_user",
|
73 |
+
)
|
74 |
+
else:
|
75 |
+
message(st.session_state["messages"][i], key=str(i) + "_AI")
|