Update app.py

app.py

@@ -3,14 +3,14 @@ import requests
 import json
 
 st.title("OpenAI Chatbot")
-st.write("Interact with OpenAI's GPT-3 models in real-time using your OpenAI API. Choose from a selection of their best models, set the temperature and max tokens, and start a conversation. Delete the conversation at any time to start fresh.")
+st.write("Interact with OpenAI's models in real-time using your OpenAI API. Choose from a selection of their best models, set the temperature and max tokens, and start a conversation. Delete the conversation at any time and start fresh.")
 
 if "history" not in st.session_state:
     st.session_state.history = []
 
 st.sidebar.markdown("## Configuration")
 KEY = st.sidebar.text_input("Enter your OpenAI API key")
-models = ['text-davinci-003', 'text-curie-001', 'text-babbage-001', 'text-ada-001']
+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)
@@ -41,6 +41,9 @@ def generate_answer(prompt):
         "temperature": temperature,
         "max_tokens": max_tokens
     }
+    if not API_KEY:
+        st.warning("Please input your API key")
+        return
     response = requests.post(API_URL, headers=headers, data=json.dumps(data))
     result = response.json()
     if 'choices' in result:
@@ -50,8 +53,6 @@ def generate_answer(prompt):
     else:
         st.error("An error occurred while processing the API response.")
 
-
-
 prompt = st.text_input("Prompt")
 if st.button("Submit"):
     generate_answer(prompt)