Update app.py

app.py

@@ -16,6 +16,9 @@ if "messages" not in st.session_state:
 for msg in st.session_state.messages:
     st.chat_message(msg["role"]).write(msg["content"])
 
+# Select model using a dropdown
+model_choice = st.selectbox('Choose a model:', ['gpt-3.5-turbo', 'gpt-4-turbo'])
+
 # Input for new prompts
 prompt = st.chat_input("Enter your question:")
 if prompt:
@@ -30,10 +33,9 @@ if prompt:
     # Use a spinner to indicate that the model is generating a response
     with st.spinner('Thinking...'):
         client = OpenAI(api_key=openai_api_key)
-        response = client.chat.completions.create(model="gpt-3.5-turbo", messages=st.session_state.messages)
+        response = client.chat.completions.create(model=model_choice, messages=st.session_state.messages)
         msg = response.choices[0].message.content
 
     # Append and display the assistant's response
     st.session_state.messages.append({"role": "assistant", "content": msg})
     st.chat_message("assistant").write(msg)
-