Update app.py

app.py

@@ -284,7 +284,10 @@ def main():
         - Educational content works better than narrative
         """)
     
-    # Generate Quiz button
+    # Generate Quiz button with automatic rerun logic
+    if "quiz_generated" not in st.session_state:
+        st.session_state.quiz_generated = False
+    
     if st.button("🧠 Generate Quiz"):
         if passage and len(passage) > 50:
             # Loading the model (with the cached resource)
@@ -313,7 +316,7 @@ def main():
                         st.session_state.user_answers = {}
                         st.session_state.quiz_submitted = False
                         st.session_state.show_explanations = False
-                        st.experimental_rerun()
+                        st.session_state.quiz_generated = True
                     else:
                         st.error("Couldn't create valid quiz questions. Please try a different text or add more content.")
                 else:
@@ -381,11 +384,11 @@ def main():
             
             # Restart button
             if st.button("Generate Another Quiz"):
-                # Clear session state and rerun
-                for key in ["quiz_items", "user_answers", "quiz_submitted", "show_explanations"]:
+                # Clear session state
+                for key in ["quiz_items", "user_answers", "quiz_submitted", "show_explanations", "quiz_generated"]:
                     if key in st.session_state:
                         del st.session_state[key]
-                st.experimental_rerun()
+                # No need for rerun as page will refresh naturally with next event
 
 if __name__ == "__main__":
     main()