Update app.py

app.py

@@ -125,11 +125,17 @@ if submit:
 if st.session_state.submit:
     if uploaded_files:
         client = OpenAI(api_key=openai_api_key)
-        
-        try:    
-            client.Model.list()
-        except:
-            print(openai_api_key)
+
+        # test api key
+        response = client.chat.completions.create(
+            model="gpt-4o-mini",
+            messages=[
+              {"role": "system", "content": "You are a helpful assistant."},
+              {"role": "user", "content": "Say this is a test"}
+            ]
+        )
+        test = response.choices[0].message.content
+        if test == "":
             st.session_state.error = "OpenAI API key is invalid"
             st.session_state.submit = False
             st.rerun()