Update app.py
Browse files
app.py
CHANGED
@@ -17,6 +17,12 @@ st.header("Hey, I'm EcoCoinBot. I'm a sustainable cryptocurrency advisor. I'm yo
|
|
17 |
# Input OpenAI API key
|
18 |
openai_api_key = st.text_input("Enter your OpenAI API Key:", type="password")
|
19 |
|
|
|
|
|
|
|
|
|
|
|
|
|
20 |
# Create sessionMessages if not already exists
|
21 |
if "sessionMessages" not in st.session_state:
|
22 |
st.session_state.sessionMessages = [
|
@@ -24,9 +30,10 @@ if "sessionMessages" not in st.session_state:
|
|
24 |
]
|
25 |
|
26 |
# Function to load answer
|
27 |
-
def load_answer(question):
|
28 |
st.session_state.sessionMessages.append(HumanMessage(content=question))
|
29 |
try:
|
|
|
30 |
assistant_answer = chat.invoke(st.session_state.sessionMessages)
|
31 |
st.session_state.sessionMessages.append(AIMessage(content=assistant_answer.content))
|
32 |
return assistant_answer.content
|
@@ -34,7 +41,7 @@ def load_answer(question):
|
|
34 |
if "Incorrect API key" in str(e):
|
35 |
return "The provided OpenAI API Key is incorrect. Please try again."
|
36 |
else:
|
37 |
-
return "An error occurred. Please try again later."
|
38 |
|
39 |
# Function to get user input
|
40 |
def get_text():
|
@@ -43,12 +50,11 @@ def get_text():
|
|
43 |
|
44 |
# Proceed if OpenAI API key is provided
|
45 |
if openai_api_key:
|
46 |
-
chat = ChatOpenAI(temperature=0, openai_api_key=openai_api_key)
|
47 |
user_input = get_text()
|
48 |
submit = st.button('Generate')
|
49 |
|
50 |
if submit:
|
51 |
-
response = load_answer(user_input)
|
52 |
st.subheader("Answer:")
|
53 |
st.write(response)
|
54 |
else:
|
|
|
17 |
# Input OpenAI API key
|
18 |
openai_api_key = st.text_input("Enter your OpenAI API Key:", type="password")
|
19 |
|
20 |
+
# Dropdown to select the model
|
21 |
+
model_choice = st.selectbox(
|
22 |
+
"Select the model:",
|
23 |
+
["gpt-4", "gpt-4-0", "gpt-3.5", "gpt-3.5-turbo"]
|
24 |
+
)
|
25 |
+
|
26 |
# Create sessionMessages if not already exists
|
27 |
if "sessionMessages" not in st.session_state:
|
28 |
st.session_state.sessionMessages = [
|
|
|
30 |
]
|
31 |
|
32 |
# Function to load answer
|
33 |
+
def load_answer(question, model):
|
34 |
st.session_state.sessionMessages.append(HumanMessage(content=question))
|
35 |
try:
|
36 |
+
chat = ChatOpenAI(temperature=0, openai_api_key=openai_api_key, model=model)
|
37 |
assistant_answer = chat.invoke(st.session_state.sessionMessages)
|
38 |
st.session_state.sessionMessages.append(AIMessage(content=assistant_answer.content))
|
39 |
return assistant_answer.content
|
|
|
41 |
if "Incorrect API key" in str(e):
|
42 |
return "The provided OpenAI API Key is incorrect. Please try again."
|
43 |
else:
|
44 |
+
return f"An error occurred: {str(e)}. Please try again later."
|
45 |
|
46 |
# Function to get user input
|
47 |
def get_text():
|
|
|
50 |
|
51 |
# Proceed if OpenAI API key is provided
|
52 |
if openai_api_key:
|
|
|
53 |
user_input = get_text()
|
54 |
submit = st.button('Generate')
|
55 |
|
56 |
if submit:
|
57 |
+
response = load_answer(user_input, model_choice)
|
58 |
st.subheader("Answer:")
|
59 |
st.write(response)
|
60 |
else:
|