Update app.py

app.py

@@ -17,6 +17,12 @@ st.header("Hey, I'm EcoCoinBot. I'm a sustainable cryptocurrency advisor. I'm yo
 # Input OpenAI API key
 openai_api_key = st.text_input("Enter your OpenAI API Key:", type="password")
 
+# Dropdown to select the model
+model_choice = st.selectbox(
+    "Select the model:",
+    ["gpt-4", "gpt-4-0", "gpt-3.5", "gpt-3.5-turbo"]
+)
+
 # Create sessionMessages if not already exists
 if "sessionMessages" not in st.session_state:
     st.session_state.sessionMessages = [
@@ -24,9 +30,10 @@ if "sessionMessages" not in st.session_state:
     ]
 
 # Function to load answer
-def load_answer(question):
+def load_answer(question, model):
     st.session_state.sessionMessages.append(HumanMessage(content=question))
     try:
+        chat = ChatOpenAI(temperature=0, openai_api_key=openai_api_key, model=model)
         assistant_answer = chat.invoke(st.session_state.sessionMessages)
         st.session_state.sessionMessages.append(AIMessage(content=assistant_answer.content))
         return assistant_answer.content
@@ -34,7 +41,7 @@ def load_answer(question):
         if "Incorrect API key" in str(e):
             return "The provided OpenAI API Key is incorrect. Please try again."
         else:
-            return "An error occurred. Please try again later."
+            return f"An error occurred: {str(e)}. Please try again later."
 
 # Function to get user input
 def get_text():
@@ -43,12 +50,11 @@ def get_text():
 
 # Proceed if OpenAI API key is provided
 if openai_api_key:
-    chat = ChatOpenAI(temperature=0, openai_api_key=openai_api_key)
     user_input = get_text()
     submit = st.button('Generate')  
 
     if submit:
-        response = load_answer(user_input)
+        response = load_answer(user_input, model_choice)
         st.subheader("Answer:")
         st.write(response)
 else: