Update app.py

app.py

@@ -66,7 +66,6 @@ style_count_map = {0: "visual", 1: "auditory", 2: "reading/writing"}
 
 def learning_style_quiz(*answers):
     scores = {'visual': 0, 'auditory': 0, 'reading/writing': 0}
-
     for i, ans in enumerate(answers):
         if i < len(learning_style_answers):
             options = learning_style_answers[i]
@@ -74,13 +73,10 @@ def learning_style_quiz(*answers):
                 index = options.index(ans)
                 style = style_count_map[index]
                 scores[style] += 1
-
     max_score = max(scores.values())
     best_styles = [style.capitalize() for style, score in scores.items() if score == max_score]
-
     return ", ".join(best_styles)
 
-# PanoramaEd categories and multiple choice questions
 get_to_know_categories = {
     "All About Me": [
         ("What’s your favorite way to spend a day off?", []),
@@ -116,7 +112,6 @@ get_to_know_categories = {
     ]
 }
 
-# Generators for output summaries
 def generate_learning_plan(info):
     level = info.get("Grade_Level", "unknown")
     courses = info.get("Courses", [])
@@ -143,32 +138,25 @@ 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."
-
+        return "⚠️ Please upload your transcript.", False
     quiz_answers = inputs[:len(learning_style_questions)]
     if any(ans is None for ans in quiz_answers):
-        return "⚠️ Please answer all the learning style questions."
-
+        return "⚠️ Please answer all the learning style questions.", False
     blog_checkbox = inputs[len(learning_style_questions)]
     blog_text = inputs[len(learning_style_questions)+1]
     category_answers = inputs[len(learning_style_questions)+2:]
-
     if any(ans.strip() == "" for ans in category_answers):
-        return "⚠️ Please complete all 'Get to Know You' sections before saving."
-
+        return "⚠️ Please complete all 'Get to Know You' sections before saving.", False
     if blog_checkbox and blog_text.strip() == "":
-        return "⚠️ You checked the blog option but didn’t write anything. Please write your mini blog or uncheck the option."
+        return "⚠️ You checked the blog option but didn’t write anything. Please write your mini blog or uncheck the option.", False
 
     df = parse_transcript(file)
     transcript_info = extract_transcript_info(df)
     learning_type = learning_style_quiz(*quiz_answers)
-
     question_texts = [q for cat in get_to_know_categories.values() for q, _ in cat]
     responses = dict(zip(question_texts, category_answers))
-
     profile = {
         "transcript": df.to_dict(orient='records'),
         "transcript_info": transcript_info,
@@ -176,25 +164,28 @@ def save_profile(file, *inputs):
         "get_to_know_answers": responses,
         "blog": blog_text if blog_checkbox else "[User chose to skip this section]"
     }
-
     summary = {
         "Learning_Plan": generate_learning_plan(transcript_info),
         "Style_Summary": generate_learning_style_summary(learning_type),
         "Motivation": generate_motivation_section(responses)
     }
-
     with open("student_profile.json", "w") as f:
         json.dump(profile, f, indent=4)
-
     with open("student_summary.md", "w") as f:
         f.write(summary["Learning_Plan"] + '\n' + summary["Style_Summary"] + '\n' + summary["Motivation"])
+    return f"✅ Profile saved! Your learning style is: {learning_type}", True
 
-    return f"✅ Profile saved! Your learning style is: {learning_type}"
+def hide_ui():
+    updates = []
+    updates.append(gr.update(visible=False))  # for file upload
+    updates.extend([gr.update(visible=False) for _ in range(len(learning_style_questions))])
+    updates.append(gr.update(visible=False))  # blog checkbox
+    updates.append(gr.update(visible=False))  # blog text
+    updates.extend([gr.update(visible=False) for _ in range(sum(len(v) for v in get_to_know_categories.values()))])
+    return updates
 
-# Build Gradio UI
 with gr.Blocks() as demo:
     gr.Markdown("## 🎓 Personalized AI Student Assistant")
-
     with gr.Row():
         file = gr.File(label="📄 Upload Your Transcript (.csv, .xlsx, .pdf)")
 
@@ -215,7 +206,6 @@ 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")
@@ -223,7 +213,11 @@ with gr.Blocks() as demo:
 
     submit.click(fn=save_profile,
                  inputs=[file, *quiz_components, blog_checkbox, blog_text, *category_inputs],
-                 outputs=[output])
+                 outputs=[output, gr.State()])
+
+    submit.click(fn=hide_ui,
+                 inputs=[],
+                 outputs=[file, *quiz_components, blog_checkbox, blog_text, *category_inputs])
 
 if __name__ == '__main__':
-    demo.launch()
+    demo.launch()