Upload index.py

index.py

@@ -135,7 +135,20 @@ with tab2:
     st.title("AI-Powered Registration System")
 
     if st.session_state.summary:
-        # [Summary code remains unchanged]
+        st.success("Registration Complete!")
+        st.subheader("Summary")
+        for key, value in st.session_state.summary.items():
+            st.write(f"**{key}**: {value}")
+            new_value = st.text_input(f"Edit {key}", key=f"edit_{key}")
+            if st.button(f"Update {key}", key=f"update_{key}"):
+                edit_field(key, new_value)
+        col1, col2 = st.columns(2)
+        with col1:
+            if st.button("Next Registration", key="next_reg", on_click=start_registration):
+                pass
+        with col2:
+            if st.button("End Session", key="end_sess", on_click=start_registration):
+                pass
     else:
         if st.session_state.current_question:
             with st.form(key=f"question_form_{st.session_state.question_number}"):