Update app.py

app.py

@@ -10,7 +10,7 @@ if "history" not in st.session_state:
 
 st.sidebar.markdown("## Configuration")
 API_KEY = st.sidebar.text_input("Enter your OpenAI API key")
-models = ['text-davinci-003', 'text-curie-001', 'text-babbage-001 (WIP)', 'text-ada-001 (WIP)']
+models = ['text-davinci-003', 'text-curie-001 (WIP)', 'text-babbage-001 (WIP)', 'text-ada-001 (WIP)']
 model = st.sidebar.selectbox("Select a model", models, index=0)
 
 temperature = st.sidebar.slider("Temperature", 0.0, 1.0, 0.7)
@@ -18,10 +18,8 @@ max_tokens = st.sidebar.slider("Max Tokens", 0, 4000, 2024)
 
 if st.sidebar.button("Delete Conversation"):
     st.session_state.history = []
-
 st.sidebar.markdown("## GPT-3")
 st.sidebar.markdown("OpenAI's GPT-3 models can understand and generate natural language. They offer four main models with different levels of power suitable for different tasks. Davinci is the most capable model, and Ada is the fastest.")
-
 st.sidebar.markdown("text-davinci-003 | 4,000 max tokens")
 st.sidebar.markdown("text-curie-001 | 2,048 max tokens")
 st.sidebar.markdown("text-babbage-001 | 2,048 max tokens")
@@ -45,9 +43,12 @@ def generate_answer(prompt):
     }
     response = requests.post(API_URL, headers=headers, data=json.dumps(data))
     result = response.json()
-    message_bot = result['choices'][0]['text']
-    st.session_state.history.append({"message": prompt, "is_user": True})
-    st.session_state.history.append({"message": message_bot, "is_user": False})
+    if 'choices' in result:
+        message_bot = result['choices'][0]['text'].strip()
+        st.session_state.history.append({"message": prompt, "is_user": True})
+        st.session_state.history.append({"message": message_bot, "is_user": False})
+    else:
+        st.error("An error occurred while processing the API response.")