Update app.py

app.py

@@ -9,14 +9,24 @@ st.set_page_config(page_title="📚 Multiple Choice Test Creator", layout="wide"
 if 'tests' not in st.session_state:
     st.session_state['tests'] = []
 
-# Function to add a new test
-def add_test(test_title, questions):
-    test = {
-        'title': test_title,
-        'questions': questions
-    }
-    st.session_state['tests'].append(test)
-    st.success(f"Test '{test_title}' created successfully!")
+# Initialize session state for current questions being added
+if 'current_questions' not in st.session_state:
+    st.session_state['current_questions'] = []
+
+# Function to add a new question
+def add_question():
+    if len(st.session_state['current_questions']) < 10:
+        st.session_state['current_questions'].append({
+            'question': '',
+            'options': {'A': '', 'B': '', 'C': '', 'D': ''},
+            'correct': 'A'
+        })
+    else:
+        st.warning("You can only add up to 10 questions.")
+
+# Function to remove a question
+def remove_question(index):
+    st.session_state['current_questions'].pop(index)
 
 # Function to generate HTML content for the test
 def generate_html(test_title, questions):
@@ -77,24 +87,6 @@ with tabs[0]:
     # Input for Test Title
     test_title = st.text_input("Enter Test Title", "")
 
-    # Container for questions
-    questions_container = st.container()
-
-    # Initialize session state for questions if not present
-    if 'current_questions' not in st.session_state:
-        st.session_state['current_questions'] = []
-
-    # Function to add a new question
-    def add_question():
-        if len(st.session_state['current_questions']) < 10:
-            st.session_state['current_questions'].append({
-                'question': '',
-                'options': {'A': '', 'B': '', 'C': '', 'D': ''},
-                'correct': 'A'
-            })
-        else:
-            st.warning("You can only add up to 10 questions.")
-
     # Button to add a new question
     st.button("➕ Add a Question", on_click=add_question)
 
@@ -102,41 +94,28 @@ with tabs[0]:
     for idx, q in enumerate(st.session_state['current_questions']):
         with st.expander(f"Question {idx + 1}", expanded=True):
             # Question Text
-            st.text_input("Question Text", value=q['question'], key=f"question_{idx}", 
-                         on_change=lambda i=idx: st.session_state['current_questions'].__setitem__(
-                             i, 
-                             {'question': st.session_state[f"question_{i}"], 
-                              'options': st.session_state['current_questions'][i]['options'], 
-                              'correct': st.session_state['current_questions'][i]['correct']}
-                         ))
+            question_key = f"question_{idx}"
+            question_text = st.text_input("Question Text", value=q['question'], key=question_key)
+            st.session_state['current_questions'][idx]['question'] = question_text
 
             # Answer Options
             for option in ['A', 'B', 'C', 'D']:
-                st.text_input(f"Answer {option}", value=q['options'][option], key=f"question_{idx}_option_{option}", 
-                             on_change=lambda o=option, i=idx: st.session_state['current_questions'].__setitem__(
-                                 i, 
-                                 {
-                                     'question': st.session_state[f"question_{i}"],
-                                     'options': {**st.session_state['current_questions'][i]['options'], o: st.session_state[f"question_{i}_option_{o}"]},
-                                     'correct': st.session_state['current_questions'][i]['correct']
-                                 }
-                             ))
+                option_key = f"question_{idx}_option_{option}"
+                option_text = st.text_input(f"Answer {option}", value=q['options'][option], key=option_key)
+                st.session_state['current_questions'][idx]['options'][option] = option_text
 
             # Correct Answer Selector
-            correct_option = st.selectbox("Select Correct Answer", options=['A', 'B', 'C', 'D'], 
+            correct_option_key = f"question_{idx}_correct"
+            correct_option = st.selectbox("Select Correct Answer", options=['A', 'B', 'C', 'D'],
                                          index=['A','B','C','D'].index(q['correct']),
-                                         key=f"question_{idx}_correct", 
-                                         on_change=lambda c='A', i=idx: st.session_state['current_questions'].__setitem__(
-                                             i, 
-                                             {
-                                                 'question': st.session_state[f"question_{i}"],
-                                                 'options': st.session_state['current_questions'][i]['options'],
-                                                 'correct': st.session_state[f"question_{i}_correct"]
-                                             }
-                                         ))
-            # Update the correct answer in session_state
+                                         key=correct_option_key)
             st.session_state['current_questions'][idx]['correct'] = correct_option
 
+            # Remove Question Button
+            if st.button("🗑️ Remove", key=f"remove_{idx}"):
+                remove_question(idx)
+                st.experimental_rerun()
+
     # Button to create the test
     if st.button("✅ Create Test"):
         if not test_title:
@@ -152,17 +131,26 @@ with tabs[0]:
                     incomplete = True
                     break
             if not incomplete:
-                add_test(test_title, st.session_state['current_questions'])
+                # Add the test to the session state
+                test = {
+                    'title': test_title,
+                    'questions': st.session_state['current_questions'].copy()
+                }
+                st.session_state['tests'].append(test)
+                st.success(f"Test '{test_title}' created successfully!")
+
                 # Clear current questions after creation
                 st.session_state['current_questions'] = []
+
                 # Provide download button
-                html_content = generate_html(test_title, st.session_state['tests'][-1]['questions'])
+                html_content = generate_html(test_title, test['questions'])
                 st.download_button(
                     label="📄 Download Test as HTML",
                     data=html_content,
                     file_name=f"{test_title.replace(' ', '_')}.html",
                     mime="text/html"
                 )
+
                 # Optionally, display the test
                 st.markdown("---")
                 st.header("📝 Your Test")
@@ -186,18 +174,27 @@ with tabs[1]:
             # Initialize session state for user answers if not present
             if 'user_answers' not in st.session_state:
                 st.session_state['user_answers'] = {}
-            
+
             with st.form("take_test_form"):
-                for idx, q in enumerate(selected_test['questions'], 1):
+                for idx, q in enumerate(selected_test['questions'], start=1):
                     st.markdown(f"**{idx}. {q['question']}**")
+                    # Prepare options with letters and text
+                    option_list = [
+                        f"A. {q['options']['A']}",
+                        f"B. {q['options']['B']}",
+                        f"C. {q['options']['C']}",
+                        f"D. {q['options']['D']}"
+                    ]
+                    # Create radio buttons with the full option text
                     answer = st.radio(
                         f"Select an answer for Question {idx}",
-                        options=['A', 'B', 'C', 'D'],
+                        options=option_list,
                         key=f"test_{selected_test_title}_q{idx}"
                     )
-                    st.session_state['user_answers'][idx] = answer
+                    # Store the selected answer's letter
+                    st.session_state['user_answers'][idx] = answer.split(".")[0].strip()
                     st.markdown("---")
-                
+
                 # Submit button
                 submit = st.form_submit_button("Submit Test")
 
@@ -205,20 +202,25 @@ with tabs[1]:
                 # Grade the test
                 score = 0
                 total = len(selected_test['questions'])
-                for idx, q in enumerate(selected_test['questions'], 1):
-                    if st.session_state['user_answers'][idx] == q['correct']:
+                feedback = []
+                for idx, q in enumerate(selected_test['questions'], start=1):
+                    user_ans = st.session_state['user_answers'].get(idx, None)
+                    correct_ans = q['correct']
+                    is_correct = user_ans == correct_ans
+                    if is_correct:
                         score += 1
+                        feedback.append(f"**Question {idx}**: Correct ✅")
+                    else:
+                        feedback.append(f"**Question {idx}**: Incorrect ❌ (Correct Answer: {correct_ans})")
 
                 # Display the result
                 st.success(f"You scored {score} out of {total}.")
 
-                # Optionally, display correct answers
-                with st.expander("View Correct Answers"):
-                    for idx, q in enumerate(selected_test['questions'], 1):
-                        user_ans = st.session_state['user_answers'][idx]
-                        correct_ans = q['correct']
-                        is_correct = user_ans == correct_ans
-                        result = "✅ Correct" if is_correct else f"❌ Incorrect (Correct Answer: {correct_ans})"
-                        st.markdown(f"**{idx}. {q['question']}**")
-                        st.markdown(f"Your Answer: {user_ans} - {result}")
-                        st.markdown("---")
+                # Display feedback
+                with st.expander("View Feedback"):
+                    for fb in feedback:
+                        st.markdown(fb)
+
+                # Optionally, clear answers for next test
+                st.session_state['user_answers'] = {}
+