Updates

app.py

@@ -163,26 +163,29 @@ if selected_category:
             selected_module_data = module_data
             break
 
-    # Display black background title with PDF report button
-    st.markdown(
-        f"""
-        <div style="background-color: #333; padding: 10px; border-radius: 5px; margin-top: 20px;">
-        <span style='font-size: 18px; color: white;'>{selected_category} > {selected_module_data['title']}</span>
-        </div>
-        """, 
-        unsafe_allow_html=True
-    )
+    # Create a horizontal layout with the category title and the PDF download button
+    col1, col2 = st.columns([3, 1])
+    with col1:
+        st.markdown(
+            f"""
+            <div style="background-color: #333; padding: 10px; border-radius: 5px; margin-top: 20px;">
+            <span style='font-size: 18px; color: white;'>{selected_category} > {selected_module_data['title']}</span>
+            </div>
+            """, 
+            unsafe_allow_html=True
+        )
     
-    # Show PDF report button, initially disabled until a question is answered
-    pdf_disabled = len(st.session_state.questions) == 0
-    st.download_button(
-        label="📄 Download PDF Report",
-        data=st.session_state.pdf_data if not pdf_disabled else b'',
-        file_name="quiz_report.pdf",
-        mime="application/pdf",
-        disabled=pdf_disabled,
-        key="pdf_download_button"
-    )
+    with col2:
+        # Show PDF report button, initially disabled until a question is answered
+        pdf_disabled = len(st.session_state.questions) == 0
+        st.download_button(
+            label="📄 Download PDF Report",
+            data=st.session_state.pdf_data if not pdf_disabled else b'',
+            file_name="quiz_report.pdf",
+            mime="application/pdf",
+            disabled=pdf_disabled,
+            key="pdf_download_button"
+        )
 
     if selected_module != st.session_state.current_module:
         st.session_state.current_module = selected_module
@@ -258,6 +261,9 @@ if submit_button:
         # Generate the PDF report data after each submission
         st.session_state.pdf_data = generate_pdf_report()
 
+        # Generate a new question for the next round
+        st.session_state.current_question = generate_new_question(selected_category, selected_module, selected_module_data)
+
         # Refresh the PDF button state
         st.download_button(
             label="📄 Download PDF Report",