Update app.py

app.py

@@ -4,6 +4,7 @@ import streamlit as st
 import openai
 import time
 from datetime import datetime
+from openai.error import OpenAIError  # Import OpenAIError for handling API errors
 
 # Set your OpenAI API key from Hugging Face Secrets
 openai.api_key = st.secrets["OPENAI_API_KEY"]
@@ -69,17 +70,17 @@ def generate_questions_with_retry(knowledge_material, question_type, cognitive_l
     retries = 0
     while retries < max_retries:
         try:
-            response = client.chat.completions.create(
-                model="GPT-4o mini",
+            response = openai.ChatCompletion.create(
+                model="gpt-3.5-turbo",
                 messages=[
                     {"role": "system", "content": "You are a helpful assistant for generating exam questions."},
                     {"role": "user", "content": prompt}
                 ]
             )
-            return response.choices[0].message.content
-        except openai.error.APIConnectionError:
+            return response.choices[0].message["content"]
+        except OpenAIError as e:
             retries += 1
-            time.sleep(2)  # Wait for 2 seconds before retrying
+            st.error(f"Failed to connect to OpenAI API: {str(e)}")
             if retries == max_retries:
                 st.error("Failed to connect to OpenAI API after several attempts.")
                 return None