Update app.py

app.py

@@ -16,11 +16,19 @@ grades = [f"Grade {i}" for i in range(1, 13)]
 difficulty_levels = ["Pretest (Beginner)", "Intermediate", "Advanced"]
 min_mcq, max_mcq = 5, 50
 
-# Subjects and chapters (Grades 1–12)
+# -------------------------------
+# Subjects and chapters by grade
+# -------------------------------
 chapters_by_subject_and_grade = {
-    # Grades 1–10 already included (omitted here for brevity; same as previous code)
-    # ----------
-    # Grades 11
+    # Grades 1–12 (Grades 1–10 abbreviated here for brevity; 11-12 full as given)
+    "Grade 1": {
+        "Mathematics": ["Counting & Number Recognition", "Basic Addition", "Basic Subtraction", "Shapes & Space", "Measurement (Length, Weight)"],
+        "English": ["Alphabet & Phonics", "Listening & Speaking", "Basic Vocabulary", "Simple Sentences", "Early Reading"],
+        "Urdu": ["حروف اور آوازیں", "الفاظ", "سادہ جملے", "پڑھائی کا تعارف", "چھوٹی کہانیاں"],
+        "Islamiyat": ["Iman Basics", "Prophets' Stories (selected)", "Good Manners", "Simple Prayers & Worship"],
+        "General Knowledge": ["My Home & Family", "Plants & Animals", "Seasons & Weather", "My School"]
+    },
+    # ... (Grades 2–10 omitted for brevity, include all previous chapters as before)
     "Grade 11": {
         "Mathematics": ["Advanced Algebra & Calculus Intro", "Coordinate Geometry", "Trigonometry", "Vectors"],
         "English": ["Academic Writing", "Literature Studies", "Critical Reading"],
@@ -37,7 +45,6 @@ chapters_by_subject_and_grade = {
         "Islamiyat": ["Advanced Seerah","Islamic Ethics"],
         "Pakistan Studies": ["Pakistan in Geographical Perspective","Resources & Economy","Historical Overview","Political & Constitutional Development","Human Rights & Citizenship"]
     },
-    # Grades 12
     "Grade 12": {
         "Mathematics": ["Calculus","Advanced Algebra","Statistics","Analytical Geometry"],
         "English": ["Research Writing","Advanced Literature","Critical Analysis"],
@@ -56,6 +63,9 @@ chapters_by_subject_and_grade = {
     }
 }
 
+# -------------------------------
+# Generate MCQs using GROQ
+# -------------------------------
 def generate_mcqs(grade, subject, chapter, difficulty, num_questions):
     system_msg = "You are an educational expert. Generate MCQs strictly aligned with the Pakistani National Curriculum."
     user_msg = f"Create {num_questions} MCQs for {grade} {subject} on the chapter '{chapter}' at '{difficulty}' level. Each question must have 4 options. Provide only questions and options. Do not add any introductory text."
@@ -69,17 +79,19 @@ def generate_mcqs(grade, subject, chapter, difficulty, num_questions):
     )
 
     text = response.choices[0].message.content
-    # Extract questions and options only, removing any prefatory text
     mcqs = re.findall(r'\d+\.\s.*?(?=\n\d+\.|\Z)', text, re.DOTALL)
     mcqs = [q.strip() for q in mcqs if q.strip()]
     answer_key = [random.choice(["A","B","C","D"]) for _ in mcqs]
 
     return mcqs[:num_questions], answer_key[:num_questions]
 
+# -------------------------------
+# Export to MS Word
+# -------------------------------
 def export_to_word(mcqs, answers, grade, subject, chapter, difficulty, num_questions):
     doc = Document()
     
-    # Main heading
+    # Heading
     heading = doc.add_paragraph("MONTHLY/WEEKLY TEST")
     heading.alignment = WD_ALIGN_PARAGRAPH.CENTER
     heading.runs[0].bold = True
@@ -128,25 +140,37 @@ def on_generate(grade, subject, chapter, difficulty, num_questions):
     file_path = export_to_word(mcqs, answers, grade, subject, chapter, difficulty, num_questions)
     return f"Test ready: {file_path}", file_path
 
+# -------------------------------
 # Gradio UI
+# -------------------------------
+def update_subjects(grade):
+    subjects = list(chapters_by_subject_and_grade.get(grade, {}).keys())
+    return gr.Dropdown.update(choices=subjects, value=subjects[0] if subjects else None)
+
+def update_chapters(subject, grade):
+    chapters = chapters_by_subject_and_grade.get(grade, {}).get(subject, [])
+    return gr.Dropdown.update(choices=chapters, value=chapters[0] if chapters else None)
+
 with gr.Blocks() as demo:
     gr.Markdown("### **MCQ Test Generator - Developer: Najaf Ali Sharqi**")
+    
     with gr.Row():
         grade_dropdown = gr.Dropdown(choices=grades, label="Select Grade")
         subject_dropdown = gr.Dropdown(choices=[], label="Select Subject")
         chapter_dropdown = gr.Dropdown(choices=[], label="Select Chapter")
+        
     with gr.Row():
         difficulty_dropdown = gr.Dropdown(choices=difficulty_levels, label="Select Difficulty")
         num_mcq_slider = gr.Slider(minimum=min_mcq, maximum=max_mcq, value=20, step=1, label="Number of MCQs")
+        
     generate_btn = gr.Button("Generate Test")
     output_text = gr.Textbox(label="Status")
     output_file = gr.File(label="Download Test")
-
-    # Update subjects on grade change
-    grade_dropdown.change(lambda g: gr.update(choices=list(chapters_by_subject_and_grade.get(g, {}).keys()), value=None), grade_dropdown, subject_dropdown)
-    # Update chapters on subject change
-    subject_dropdown.change(lambda s, g: gr.update(choices=chapters_by_subject_and_grade.get(g, {}).get(s, []), value=None), [subject_dropdown, grade_dropdown], chapter_dropdown)
-
+    
+    # Dropdown dependencies
+    grade_dropdown.change(update_subjects, inputs=grade_dropdown, outputs=subject_dropdown)
+    subject_dropdown.change(update_chapters, inputs=[subject_dropdown, grade_dropdown], outputs=chapter_dropdown)
+    
     generate_btn.click(on_generate, inputs=[grade_dropdown, subject_dropdown, chapter_dropdown, difficulty_dropdown, num_mcq_slider], outputs=[output_text, output_file])
 
 demo.launch()