Update app.py

app.py

@@ -4,7 +4,9 @@ import PyPDF2
 import json
 import re
 
-# Parse uploaded transcript file
+# -------------------------
+# 1. Parse uploaded transcript
+# -------------------------
 def parse_transcript(file):
     if file.name.endswith('.csv'):
         df = pd.read_csv(file.name)
@@ -20,7 +22,6 @@ def parse_transcript(file):
         raise ValueError("Unsupported file format. Use .csv, .xlsx, or .pdf")
     return df
 
-# Extract student info
 def extract_transcript_info(df):
     transcript_text = df['Transcript_Text'].iloc[0] if 'Transcript_Text' in df.columns else ''
     info = {}
@@ -35,29 +36,107 @@ def extract_transcript_info(df):
         info['Courses'] = list(set([c[0].strip() for c in courses]))
     return info
 
-# Learning style questions
-def learning_style_quiz(q1, q2, q3):
-    scores = {'visual': 0, 'auditory': 0, 'reading/writing': 0, 'kinesthetic': 0}
-    mapping = [q1, q2, q3]
-    for answer in mapping:
-        scores[answer] += 1
-    best = max(scores, key=scores.get)
+# -------------------------
+# 2. Learning Style Quiz
+# -------------------------
+learning_style_questions = [
+    "You are about to give directions to a person. You would:",
+    "You are not sure whether a word should be spelled 'dependent' or 'dependant'. You would:",
+    "You’re planning a vacation for a group. You want people to appreciate it. You would:",
+    "You are going to cook something as a special treat. You would:",
+    "A group of tourists wants to learn about the parks or wildlife. You would:",
+    "You are not sure how a word is spelled. You would:",
+    "You prefer a teacher who likes to use:",
+    "You are trying to remember a phone number. You would:",
+    "You want to learn a new program, game, or app on your phone. You would:"
+]
+
+learning_style_answers = [
+    ["draw a map", "tell them directions", "write down the directions"],
+    ["look it up in a dictionary", "sound it out", "write both forms down"],
+    ["show them the brochures/pictures", "tell them what's planned", "list the itinerary"],
+    ["cook from a recipe", "call a friend", "improvise based on what you know"],
+    ["show maps", "talk about nature", "hand out leaflets with info"],
+    ["look it up visually", "say it aloud", "write it a few times"],
+    ["diagrams/charts", "discussions", "reading material"],
+    ["see it in your head", "say it to yourself", "write it down"],
+    ["watch tutorials", "talk to someone who knows", "read instructions"]
+]
+
+style_key_map = {
+    0: "visual",
+    1: "auditory",
+    2: "reading/writing"
+}
+
+def learning_style_quiz(*answers):
+    counts = {'visual': 0, 'auditory': 0, 'reading/writing': 0}
+    for ans in answers:
+        counts[ans] += 1
+    best = max(counts, key=counts.get)
     return best.capitalize()
 
-# Save all answers into profile
-def save_profile(file, q1, q2, q3, about_me, blog_text, blog_opt_in):
+# -------------------------
+# 3. PanoramaEd "Get to Know You" Categories
+# -------------------------
+get_to_know_categories = {
+    "All About Me": [
+        "What’s your favorite way to spend a day off?",
+        "If you could only eat one food for the rest of your life, what would it be?",
+        "Do you have any pets? If so, what are their names?",
+        "If you could travel anywhere in the world, where would you go?",
+        "What’s your favorite holiday or tradition?"
+    ],
+    "Hopes and Dreams": [
+        "What do you want to be when you grow up?",
+        "What’s something you hope to achieve this year?",
+        "If you could change the world in one way, what would you do?",
+        "What are you most proud of?",
+        "What’s a big dream you have for your future?"
+    ],
+    "School Life": [
+        "What’s your favorite subject in school?",
+        "What’s something that makes learning easier for you?",
+        "Do you prefer working alone or in groups?",
+        "What helps you feel confident in class?",
+        "What’s something you’re good at in school?"
+    ],
+    "Relationships": [
+        "Who do you look up to and why?",
+        "Who is someone that makes you feel safe and supported?",
+        "Do you have a best friend? What do you like to do together?",
+        "What’s one thing you wish people knew about you?",
+        "What’s something kind you’ve done for someone else?"
+    ]
+}
+
+# -------------------------
+# 4. Save Profile Logic
+# -------------------------
+def save_profile(file, *inputs):
+    quiz_answers = inputs[:len(learning_style_questions)]
+    about_me = inputs[len(learning_style_questions)]
+    blog_opt_in = inputs[len(learning_style_questions)+1]
+    blog_text = inputs[len(learning_style_questions)+2]
+    category_answers = inputs[len(learning_style_questions)+3:]
+
     df = parse_transcript(file)
     transcript_info = extract_transcript_info(df)
-    learning_type = learning_style_quiz(q1, q2, q3)
+    learning_type = learning_style_quiz(*quiz_answers)
+
+    if not blog_opt_in:
+        confirm = gr.Textbox.show(label="Skipping the blog might reduce personalization. Do you really want to skip?")
+        if blog_text.strip() == "":
+            blog_text = "[User chose to skip this section]"
 
-    if not blog_opt_in and blog_text.strip() == "":
-        blog_text = "[User chose to skip this section]"
+    responses = dict(zip([q for cat in get_to_know_categories.values() for q in cat], category_answers))
 
     profile = {
         "transcript": df.to_dict(orient='records'),
         "transcript_info": transcript_info,
         "learning_style": learning_type,
         "about_me": about_me,
+        "get_to_know_answers": responses,
         "blog": blog_text
     }
 
@@ -66,32 +145,42 @@ def save_profile(file, q1, q2, q3, about_me, blog_text, blog_opt_in):
 
     return f"✅ Profile saved! Your learning style is: {learning_type}"
 
-# Build Gradio UI
+# -------------------------
+# 5. 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)")
-
-    with gr.Column():
-        gr.Markdown("### 🧠 Learning Style Discovery")
-        q1 = gr.Radio(["visual", "auditory", "reading/writing", "kinesthetic"], label="1. How do you prefer to learn new topics?")
-        q2 = gr.Radio(["visual", "auditory", "reading/writing", "kinesthetic"], label="2. How do you remember lists best?")
-        q3 = gr.Radio(["visual", "auditory", "reading/writing", "kinesthetic"], label="3. Favorite way to study?")
-
-    with gr.Column():
-        gr.Markdown("### ❤️ About You")
-        about_me = gr.Textbox(lines=6, label="Answer a few questions: \n1. What’s a fun fact about you? \n2. Favorite music/artist? \n3. Your dream job?")
-
-        blog_opt_in = gr.Checkbox(label="I want to write a personal blog for better personalization")
-        blog_text = gr.Textbox(lines=5, label="✍️ Optional: Write a mini blog about your life", visible=True)
+    gr.Markdown("## 🎓 Build Your Personalized Learning Profile")
+
+    file = gr.File(label="📤 Upload Your Transcript (.pdf, .csv, .xlsx)")
+
+    gr.Markdown("### 🧠 Learning Style Questions")
+    quiz_inputs = []
+    for i, question in enumerate(learning_style_questions):
+        quiz_inputs.append(
+            gr.Radio(
+                choices=["visual", "auditory", "reading/writing"],
+                label=f"{i+1}. {question}"
+            )
+        )
+
+    gr.Markdown("### ✨ Tell Me About You")
+    about_me = gr.Textbox(lines=3, label="Tell me a fun fact, favorite music, or dream job!")
+    blog_opt_in = gr.Checkbox(label="📝 I want to write a short blog")
+    blog_text = gr.Textbox(lines=5, label="Write a short blog about your life")
+
+    category_inputs = []
+    for category, questions in get_to_know_categories.items():
+        gr.Markdown(f"### 📘 {category}")
+        for question in questions:
+            category_inputs.append(gr.Textbox(label=question))
 
     submit = gr.Button("📥 Save My Profile")
-    output = gr.Textbox(label="Status")
+    output = gr.Textbox(label="✅ Status")
 
     submit.click(fn=save_profile,
-                 inputs=[file, q1, q2, q3, about_me, blog_text, blog_opt_in],
+                 inputs=[file, *quiz_inputs, about_me, blog_opt_in, blog_text, *category_inputs],
                  outputs=[output])
 
+# Run the app
 if __name__ == '__main__':
-    demo.launch()
+    demo.launch()