Update app.py

app.py

@@ -4,9 +4,7 @@ import PyPDF2
 import json
 import re
 
-# -------------------------
-# 1. Parse uploaded transcript
-# -------------------------
+# Parse uploaded transcript file
 def parse_transcript(file):
     if file.name.endswith('.csv'):
         df = pd.read_csv(file.name)
@@ -22,6 +20,7 @@ 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 = {}
@@ -36,83 +35,80 @@ def extract_transcript_info(df):
         info['Courses'] = list(set([c[0].strip() for c in courses]))
     return info
 
-# -------------------------
-# 2. Learning Style Quiz
-# -------------------------
+# Learning style questions - from educationplanner.org
 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:"
+    "When you are learning something new, you prefer to:",
+    "When you are at home, you like to:",
+    "When you spell a word, you remember it by:",
+    "When you read, you:",
+    "When you write, you:",
+    "When you listen to music, you:",
+    "When you work at solving a problem, you:",
+    "When you give someone directions, you:",
+    "When you are concentrating, you:",
+    "When you meet someone new, you remember them by:"
 ]
 
 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"]
+    ["Watch a demonstration", "Listen to instructions", "Read about it"],
+    ["Watch TV or play games", "Listen to music or talk", "Read books or write"],
+    ["Seeing the word", "Hearing the word", "Writing the word"],
+    ["See the characters in your mind", "Hear the words in your mind", "Focus on the meaning of the words"],
+    ["Prefer diagrams or charts", "Prefer to discuss ideas", "Prefer to write detailed explanations"],
+    ["Enjoy the sound and rhythm", "Remember lyrics easily", "Analyze the meaning"],
+    ["Visualize the problem", "Talk it through", "Write out steps"],
+    ["Use maps or draw it", "Give verbal directions", "Write down directions"],
+    ["Create mental pictures", "Repeat things aloud", "Take notes or read quietly"],
+    ["Facial features", "Voice or name", "Spelling or written form"]
 ]
 
-style_key_map = {
-    0: "visual",
-    1: "auditory",
-    2: "reading/writing"
+style_count_map = {
+    0: 'visual',
+    1: 'auditory',
+    2: 'reading/writing'
 }
 
+# Quiz logic to analyze learning style
 def learning_style_quiz(*answers):
-    counts = {'visual': 0, 'auditory': 0, 'reading/writing': 0}
+    scores = {'visual': 0, 'auditory': 0, 'reading/writing': 0}
     for ans in answers:
-        counts[ans] += 1
-    best = max(counts, key=counts.get)
+        scores[ans] += 1
+    best = max(scores, key=scores.get)
     return best.capitalize()
 
-# -------------------------
-# 3. PanoramaEd "Get to Know You" Categories
-# -------------------------
+# PanoramaEd categories and multiple choice questions
 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?"
+        ("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?"
+        ("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?"
+        ("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?"
+        ("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
-# -------------------------
+# Save all answers into profile
 def save_profile(file, *inputs):
     quiz_answers = inputs[:len(learning_style_questions)]
     about_me = inputs[len(learning_style_questions)]
@@ -124,12 +120,11 @@ def save_profile(file, *inputs):
     transcript_info = extract_transcript_info(df)
     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))
+    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'),
@@ -145,42 +140,40 @@ def save_profile(file, *inputs):
 
     return f"✅ Profile saved! Your learning style is: {learning_type}"
 
-# -------------------------
-# 5. Gradio UI
-# -------------------------
+# Build Gradio UI
 with gr.Blocks() as demo:
-    gr.Markdown("## 🎓 Build Your Personalized Learning Profile")
+    gr.Markdown("## 🎓 Personalized AI Student Assistant")
 
-    file = gr.File(label="📤 Upload Your Transcript (.pdf, .csv, .xlsx)")
+    with gr.Row():
+        file = gr.File(label="📄 Upload Your Transcript (.csv, .xlsx, .pdf)")
 
-    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"],
+    with gr.Column():
+        gr.Markdown("### 🧠 Learning Style Discovery")
+        quiz_components = []
+        for i, (question, options) in enumerate(zip(learning_style_questions, learning_style_answers)):
+            quiz_components.append(gr.Radio(
+                choices=options,
                 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")
+    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)
 
     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))
+        for q_text, _ in questions:
+            category_inputs.append(gr.Textbox(label=q_text))
 
     submit = gr.Button("📥 Save My Profile")
-    output = gr.Textbox(label="✅ Status")
+    output = gr.Textbox(label="Status")
 
     submit.click(fn=save_profile,
-                 inputs=[file, *quiz_inputs, about_me, blog_opt_in, blog_text, *category_inputs],
+                 inputs=[file, *quiz_components, about_me, blog_opt_in, blog_text, *category_inputs],
                  outputs=[output])
 
-# Run the app
 if __name__ == '__main__':
-    demo.launch()
+    demo.launch()