Update app.py

app.py

@@ -143,7 +143,6 @@ Your dreams are powerful: {'; '.join(hopes) if hopes else 'You are filled with p
 Believe in yourself and keep moving forward.
 """
 
-# Save all answers into profile
 def save_profile(file, *inputs):
     if not file:
         return "⚠️ Please upload your transcript."
@@ -191,7 +190,6 @@ def save_profile(file, *inputs):
 
     return f"✅ Profile saved! Your learning style is: {learning_type}"
 
-# Build Gradio UI
 with gr.Blocks() as demo:
     gr.Markdown("## 🎓 Personalized AI Student Assistant")
 
@@ -215,15 +213,38 @@ with gr.Blocks() as demo:
 
     blog_checkbox = gr.Checkbox(label="📝 I'd like to write a mini blog about myself")
     blog_text = gr.Textbox(lines=5, label="✍️ Mini Blog", visible=False)
-
     blog_checkbox.change(fn=lambda x: gr.update(visible=x), inputs=blog_checkbox, outputs=blog_text)
 
     submit = gr.Button("🗕️ Save My Profile")
     output = gr.Textbox(label="Status")
 
+    # New UI elements for summary and confirmation
+    summary_markdown = gr.Markdown(visible=False)
+    confirm_button = gr.Button("✅ Confirm", visible=False)
+    edit_button = gr.Button("✏️ Edit", visible=False)
+
+    def handle_post_save(message):
+        with open("student_summary.md", "r") as f:
+            summary = f.read()
+        return {
+            output: gr.update(value=message),
+            summary_markdown: gr.update(value=summary, visible=True),
+            confirm_button: gr.update(visible=True),
+            edit_button: gr.update(visible=True),
+            file: gr.update(visible=False),
+            blog_checkbox: gr.update(visible=False),
+            blog_text: gr.update(visible=False),
+            **{c: gr.update(visible=False) for c in quiz_components + category_inputs}
+        }
+
     submit.click(fn=save_profile,
                  inputs=[file, *quiz_components, blog_checkbox, blog_text, *category_inputs],
-                 outputs=[output])
+                 outputs=[output]).then(
+        fn=handle_post_save,
+        inputs=None,
+        outputs=[output, summary_markdown, confirm_button, edit_button, file, blog_checkbox, blog_text, *quiz_components, *category_inputs]
+    )
 
 if __name__ == '__main__':
-    demo.launch()
+    demo.launch()
+