Update app.py

app.py

@@ -78,13 +78,19 @@ def generate_questions_with_retry(knowledge_material, question_type, cognitive_l
                 temperature=0.7  # Controls randomness
             )
             return response.choices[0].text  # Extract the text from the API response
-        except openai.error.APIError as e:
+        except openai.APIConnectionError 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
+        except openai.RateLimitError as e:
+            st.error(f"Rate limit reached: {str(e)}")
+            return None
+        except openai.APIError as e:
+            st.error(f"API Error: {str(e)}")
+            return None
 
 # Main logic for the app
 if 'username' not in st.session_state: