Update app.py

app.py

@@ -10,9 +10,6 @@ genai.configure(api_key=API_KEY)
 # Initialize the Generative Model
 model = genai.GenerativeModel('gemini-pro')
 
-# Global variable to store the generated quiz
-generated_quiz = ""
-
 def extract_text_from_pdf(pdf_file):
     """Extract text from the given PDF file."""
     try:
@@ -61,7 +58,7 @@ def answer_question_from_quiz(question):
     try:
         prompt = (f"Based on the following quiz, answer the user's question. The quiz is formatted with "
                   "questions followed by answers:\n\n"
-                  f"{generated_quiz}\n\n"
+                  f"{st.session_state.generated_quiz}\n\n"
                   f"User's question: {question}\n\n"
                   "Please provide a detailed answer based on the quiz content.")
         response = model.generate_content(prompt)
@@ -72,15 +69,13 @@ def answer_question_from_quiz(question):
 def process_pdf(pdf_file, difficulty):
     """Process the uploaded PDF to generate and explain quiz questions."""
     try:
-        global generated_quiz
-        
         # Extract text from the PDF
         extracted_text = extract_text_from_pdf(pdf_file)
         
         # Generate a new quiz based on the extracted text
-        generated_quiz = generate_quiz(extracted_text, difficulty)
+        st.session_state.generated_quiz = generate_quiz(extracted_text, difficulty)
         
-        return generated_quiz
+        return st.session_state.generated_quiz
     except Exception as e:
         return f"Error processing PDF: {e}"
 
@@ -88,20 +83,22 @@ def main():
     st.title("AI Quiz Generator")
     st.write("Upload an exam or quiz PDF to generate a new quiz with the same style and complexity. You can also ask questions based on the generated quiz.")
     
+    if 'generated_quiz' not in st.session_state:
+        st.session_state.generated_quiz = ""
+
     uploaded_file = st.file_uploader("Upload PDF", type=["pdf"])
     difficulty = st.selectbox("Difficulty Level", ['easy', 'medium', 'hard'])
     
     if st.button("Generate Quiz"):
         if uploaded_file is not None:
-            global generated_quiz
-            generated_quiz = process_pdf(uploaded_file, difficulty)
+            st.session_state.generated_quiz = process_pdf(uploaded_file, difficulty)
             
             st.subheader("Generated Quiz")
-            st.text(generated_quiz)
+            st.text(st.session_state.generated_quiz)
         else:
             st.error("Please upload a PDF file.")
     
-    if generated_quiz:
+    if st.session_state.generated_quiz:
         question = st.text_input("Ask a question about the quiz", "")
         if st.button("Ask"):
             if question: