Fixing sidebar

app.py

@@ -11,7 +11,7 @@ st.set_page_config(initial_sidebar_state="collapsed")
 def login_page():
     # st.title("API Selection")
 
-    
+
     api_options = ["Gemini API", "OpenAI API"]
     selected_api = st.selectbox("### Select API:", api_options)
     api_key = st.text_input(f"Enter your {selected_api} Key:", type="password")
@@ -45,7 +45,7 @@ def login_page():
 def quiz_app():
     st.title("QuizLM 🧠")
 
-    # Initialize session state for quiz data and progress (if not already initialized during login)
+    # Initialize session state for quiz data and progress
     if 'quiz_data' not in st.session_state:
         st.session_state.quiz_data = None
     if 'current_question_index' not in st.session_state:
@@ -56,8 +56,11 @@ def quiz_app():
         st.session_state.quiz_completed = False
     if 'score' not in st.session_state:
         st.session_state.score = 0
-    if 'quiz_started' not in st.session_state: # Track if quiz generation has started
+    if 'quiz_started' not in st.session_state:
         st.session_state.quiz_started = False
+    if 'quiz_topic' not in st.session_state:
+        st.session_state.quiz_topic = "" # Initialize quiz topic
+
 
     # API Configuration based on selection - Moved inside quiz_app after login
     if st.session_state.selected_api == "Gemini API":
@@ -202,20 +205,16 @@ def quiz_app():
                 st.markdown(f"<font color='{color}'>{result_text}</font>", unsafe_allow_html=True)
 
             st.divider()
-            
+
             percentage_correct = (correct_count / len(st.session_state.quiz_data)) * 100
-            
+
             if percentage_correct > 75:
                 border_color = "#9AD8E1"
                 delta_text_color = "normal"
             else:
                 border_color = "#E098B7"
                 delta_text_color = "inverse"
-            
-            # if st.get_option("theme.base") == "dark":
-            #     background_color = "#292D34"  # Dark background for dark mode
-            # else:
-            #     background_color = "#FFF"      # White background for light mode
+
 
             st.write(
                 """
@@ -227,24 +226,27 @@ def quiz_app():
                 """,
                 unsafe_allow_html=True,
             )
-            
+
             st.metric(label=f"**Final Score**", value=f"{correct_count} out of {len(st.session_state.quiz_data)}", delta=f"{percentage_correct:.0f}%", delta_color=delta_text_color)
             style_metric_cards(border_left_color=border_color, background_color="DEFAULT_BACKGROUND_COLOR")
-            
-            # st.markdown(f"### Final Score: {correct_count} out of {len(st.session_state.quiz_data)} correct ({percentage_correct:.2f}%)")
+
             st.session_state.score = correct_count # Store score in session state
 
 
-    # User input for topic - Conditional placement
-    if not st.session_state.quiz_started:
-        topic = st.text_input("Enter the topic for your quiz:")
-    else:
-        topic = st.sidebar.text_input("Quiz Topic", value=st.session_state.get('quiz_topic', '')) # Move to sidebar after quiz starts, keep topic value
+    # User input for topic in sidebar
+    topic = st.sidebar.text_input("Quiz Topic", value=st.session_state.get('quiz_topic', ''))
 
-    if topic and not st.session_state.quiz_started: # Only process topic and button if quiz hasn't started yet for this topic
-        if st.button("Generate Quiz"):
-            st.session_state.quiz_started = True # Set quiz started to true when generate button is clicked
-            st.session_state.quiz_topic = topic # Store topic in session state to keep it in sidebar input
+    if st.sidebar.button("Generate Quiz"): # Button in sidebar to generate quiz
+        st.session_state.quiz_started = True # Set quiz started to true when generate button is clicked
+        st.session_state.quiz_topic = topic # Store topic in session state to keep it in sidebar input
+        # Reset quiz state when generating a new quiz
+        st.session_state.quiz_data = None
+        st.session_state.current_question_index = 0
+        st.session_state.user_answers = []
+        st.session_state.quiz_completed = False
+        st.session_state.score = 0
+
+        if topic:
             with st.spinner(f"Generating quiz on '{topic}'..."):
                 try:
                     prompt = f"""
@@ -307,10 +309,6 @@ def quiz_app():
                     if quiz_content: # Check if quiz_content was generated successfully (outer if)
                         if parsed_quiz_data: # Check if parsing was successful (inner if)
                             st.session_state.quiz_data = parsed_quiz_data
-                            st.session_state.current_question_index = 0
-                            st.session_state.user_answers = []
-                            st.session_state.quiz_completed = False
-                            st.session_state.score = 0
                             st.success(f"Quiz on '{topic}' generated successfully! Let's begin.")
                         else: # else associated with inner if parsed_quiz_data
                             st.error("Failed to parse quiz content. Please try generating again.")
@@ -321,7 +319,10 @@ def quiz_app():
                 except Exception as e:
                     st.error(f"An error occurred: {e}")
                     st.error("Please check your API key and network connection. If the problem persists, try a different topic or try again later.")
-            st.rerun() # Rerun to move topic input to sidebar
+        else:
+            st.warning("Please enter a topic to generate a quiz.")
+        st.rerun() # Rerun to update the quiz display based on new topic/generation
+
 
     # Quiz Display Logic
     if st.session_state.quiz_data:
@@ -329,8 +330,11 @@ def quiz_app():
             display_question()
         else:
             display_results()
-    elif topic and not st.session_state.quiz_data and not st.session_state.quiz_completed and not st.session_state.quiz_started: # Message if topic entered but quiz not generated yet and quiz not started
-        st.info("Click 'Generate Quiz' to start the quiz.")
+    elif topic and not st.session_state.quiz_data and not st.session_state.quiz_completed and st.session_state.quiz_started == False: # Message if topic entered but quiz not generated yet and quiz not started
+        st.info("Enter a topic and click 'Generate Quiz' in the sidebar to start.")
+    elif not topic and not st.session_state.quiz_data and not st.session_state.quiz_completed and st.session_state.quiz_started == False:
+        st.info("Enter a topic in the sidebar to generate a quiz.")
+
 
 # --- Main App Flow Control ---
 if 'logged_in' not in st.session_state: