Update app.py

app.py

@@ -9,11 +9,19 @@ openAiKey = st.text_input(label="Input the openai key", type="password")
 
 try:
     openai.api_key = openAiKey
+    # Test request to check if the key is valid
+    openai.ChatCompletion.create(
+        model="gpt-3.5-turbo",
+        messages=[
+            {"role": "system", "content": "You are a helpful assistant."},
+            {"role": "user", "content": "Test prompt to check OpenAI key validity."}
+        ]
+    )
     st.success("OpenAI key is valid!")
 except Exception as e:
     st.error(f"Invalid OpenAI key: {e}")
     st.stop()
-    
+
 def get_chat_completion(prompt, model="gpt-3.5-turbo"):
     try:
         response = openai.ChatCompletion.create(
@@ -26,6 +34,8 @@ def get_chat_completion(prompt, model="gpt-3.5-turbo"):
         return response['choices'][0]['message']['content']
     except Exception as e:
         return str(e)
+
+    
 def main():
     st.sidebar.markdown("""
     <style>