Update app.py

app.py

@@ -1,325 +1,81 @@
-# app.py
 import os
-import re
-import random
 import gradio as gr
 from groq import Groq
 from datetime import datetime
 from docx import Document
 from docx.shared import Pt, RGBColor
 from docx.enum.text import WD_ALIGN_PARAGRAPH
+import re
+import random
 
-# ---------------------------
-# Initialize GROQ client (reads API key from env variable)
-# ---------------------------
+# Initialize GROQ client
 GROQ_API_KEY = os.getenv("GROQ_API_KEY")
 client = Groq(api_key=GROQ_API_KEY)
 
-# ---------------------------
-# Grades and subjects (UI choices)
-# ---------------------------
+# Grades
 grades = [f"Grade {i}" for i in range(1, 13)]
-subjects_by_grade = {
-    "Grade 1": ["Mathematics", "English", "Urdu", "Islamiyat", "General Knowledge"],
-    "Grade 2": ["Mathematics", "English", "Urdu", "Islamiyat", "General Knowledge"],
-    "Grade 3": ["Mathematics", "English", "Urdu", "Science", "Islamiyat", "General Knowledge"],
-    "Grade 4": ["Mathematics", "English", "Urdu", "Science", "Islamiyat", "Social Studies"],
-    "Grade 5": ["Mathematics", "English", "Urdu", "Science", "Islamiyat", "Social Studies"],
-    "Grade 6": ["Mathematics", "English", "Urdu", "Science", "Islamiyat", "Social Studies"],
-    "Grade 7": ["Mathematics", "English", "Urdu", "Science", "Islamiyat", "Social Studies", "Computer"],
-    "Grade 8": ["Mathematics", "English", "Urdu", "Science", "Islamiyat", "Social Studies", "Computer"],
-    "Grade 9": ["Mathematics", "English", "Urdu", "Biology", "Chemistry", "Physics", "Islamiyat", "Pakistan Studies", "Computer"],
-    "Grade 10": ["Mathematics", "English", "Urdu", "Biology", "Chemistry", "Physics", "Islamiyat", "Pakistan Studies", "Computer"],
-    "Grade 11": ["Mathematics", "English", "Urdu", "Biology", "Chemistry", "Physics", "Computer", "Islamiyat", "Pakistan Studies"],
-    "Grade 12": ["Mathematics", "English", "Urdu", "Biology", "Chemistry", "Physics", "Computer", "Islamiyat", "Pakistan Studies"]
-}
 
-# ---------------------------
-# Chapters by subject and grade (complete, 1-12)
-# Based on National Curriculum (2006) where available.
-# For Grades 9-10 biological/chem/phys/pkstudies, kept your exact lists.
-# For Grades 1-8 and other subjects, chapter/unit headings derive from curriculum SLO groupings.
-# ---------------------------
-chapters_by_subject_and_grade = {
-    # ---------- Grades 1-5 (competency-based curriculum — grouped into practical unit headings) ----------
-    "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"]
-    },
-    "Grade 2": {
-        "Mathematics": ["Place Value", "Addition & Subtraction Facts", "Money", "Time (clock)", "Basic Shapes"],
-        "English": ["Reading Short Texts", "Simple Grammar (nouns, verbs)", "Vocabulary Building", "Speaking Activities"],
-        "Urdu": ["روزمرہ الفاظ", "مختصر کہانیاں", "قرأتِ آسان"],
-        "Islamiyat": ["Stories of Prophets", "Good Deeds", "Worship Basics"],
-        "General Knowledge": ["Local Community", "Transportation & Safety", "Plants & Animals"]
-    },
-    "Grade 3": {
-        "Mathematics": ["Multiplication & Division", "Fractions (intro)", "Geometry Basics", "Measurement", "Patterns"],
-        "English": ["Tenses (simple)", "Comprehension Passages", "Vocabulary Development", "Sentence Writing"],
-        "Urdu": ["قواعد كا تعارف", "کہیں کہانیاں"],
-        "Science": ["Plants & Animals", "Human Body (basic)", "Materials & Uses"],
-        "Islamiyat": ["Seerat Stories & Morals", "Basic Akhlaq"]
-    },
-    "Grade 4": {
-        "Mathematics": ["Fractions & Decimals (intro)", "Geometry (angles, shapes)", "Data & Graphs", "Word Problems"],
-        "English": ["Grammar (parts of speech)", "Paragraph Writing", "Comprehension", "Listening & Speaking"],
-        "Urdu": ["نظم و نثر کا مطالعہ", "قواعد"],
-        "Science": ["Ecosystems", "Forces & Simple Machines", "Heat & Energy"],
-        "Islamiyat": ["Pillars of Islam", "Seerah Highlights"],
-        "Social Studies": ["Local Geography", "Community & Careers", "Citizenship"]
-    },
-    "Grade 5": {
-        "Mathematics": ["Number Operations", "Percentage & Ratio (intro)", "Geometry & Area", "Statistics (intro)"],
-        "English": ["Advanced Comprehension", "Essay & Letter Writing", "Grammar Consolidation"],
-        "Urdu": ["ادبی مطالعہ اور تحریر", "قواعد"],
-        "Science": ["Cells (intro)", "Matter & its Properties", "Energy & Sound"],
-        "Islamiyat": ["Quranic Stories & Morals"],
-        "Social Studies": ["History (local & national)", "Map Skills", "Basic Economics"]
-    },
+# Difficulty levels
+difficulty_levels = ["Pretest (Beginner)", "Intermediate", "Advanced"]
 
-    # ---------- Grades 6-8 ----------
-    "Grade 6": {
-        "Mathematics": ["Number System & Factors", "Algebraic Expressions (intro)", "Geometry Basics", "Data Handling"],
-        "English": ["Grammar & Usage", "Comprehension", "Writing Paragraphs", "Poetry & Prose"],
-        "Urdu": ["نثر و نظم", "قواعد ِاردو", "تحریر"],
-        "Science": ["Atoms & Molecules (intro)", "Human Body Systems", "Plants & Ecology"],
-        "Islamiyat": ["Seerah & Ahkam (intro)"],
-        "Social Studies": ["Physical Geography", "Early Civilizations", "Citizenship"]
-    },
-    "Grade 7": {
-        "Mathematics": ["Algebra (linear)", "Geometry (polygons, areas)", "Mensuration", "Graphs"],
-        "English": ["Advanced Grammar", "Comprehension Passages", "Composition"],
-        "Urdu": ["ادبی اور تنقیدی مطالعہ", "قواعد"],
-        "Science": ["Cell Biology", "Electricity & Magnetism (basic)", "Properties of Matter"],
-        "Islamiyat": ["Islamic History & Ethics"],
-        "Social Studies": ["Regional Geography", "Historical Events & Movements"],
-        "Computer": ["ICT Basics", "Introductory Coding Concepts"]
-    },
-    "Grade 8": {
-        "Mathematics": ["Linear Equations & Inequalities", "Coordinate Geometry", "Pythagoras & Trigonometry (intro)", "Probability"],
-        "English": ["Critical Reading", "Extended Writing", "Grammar Review"],
-        "Urdu": ["ادبی تنقید", "تحریر"],
-        "Science": ["Forces & Motion", "Heat & Temperature", "Waves & Optics (intro)"],
-        "Islamiyat": ["Fiqh Basics & Seerah"],
-        "Social Studies": ["Civics & Government", "Economic Concepts"],
-        "Computer": ["Algorithms & Flowcharts", "Application Software"]
-    },
+# MCQ count range
+min_mcq, max_mcq = 5, 50
 
-    # ---------- Grade 9 (kept as you provided earlier for science subjects) ----------
+# Example chapters database (fill official chapters later)
+chapters_by_grade_subject = {
     "Grade 9": {
-        "Mathematics": ["Number Systems & Algebra", "Coordinate Geometry", "Trigonometry", "Graphs"],
-        "English": ["Advanced Grammar", "Comprehension", "Writing Skills"],
-        "Urdu": ["ادب اور نثر", "نظم", "مضامین"],
         "Biology": [
-            "INTRODUCTION TO BIOLOGY",
-            "SOLVING A BIOLOGICAL PROBLEM",
-            "BIODIVERSITY",
-            "CELLS AND TISSUES",
-            "CELL CYCLE",
-            "ENZYMES",
-            "BIOENERGETICS",
-            "NUTRITION",
-            "TRANSPORT"
+            "INTRODUCTION TO BIOLOGY","SOLVING A BIOLOGICAL PROBLEM","BIODIVERSITY",
+            "CELLS AND TISSUES","CELL CYCLE","ENZYMES","BIOENERGETICS",
+            "NUTRITION","TRANSPORT"
         ],
         "Chemistry": [
-            "FUNDAMENTALS OF CHEMISTRY",
-            "STRUCTURE OF ATOMS",
-            "PERIODIC TABLE AND PERIODICITY OF PROPERTIES",
-            "STRUCTURE OF MOLECULES",
-            "PHYSICAL STATES OF MATTER",
-            "SOLUTIONS",
-            "ELECTROCHEMISTRY",
-            "CHEMICAL REACTIVITY"
+            "FUNDAMENTALS OF CHEMISTRY","STRUCTURE OF ATOMS","PERIODIC TABLE AND PERIODICITY OF PROPERTIES",
+            "STRUCTURE OF MOLECULES","PHYSICAL STATES OF MATTER","SOLUTIONS",
+            "ELECTROCHEMISTRY","CHEMICAL REACTIVITY"
         ],
         "Physics": [
-            "PHYSICAL QUANTITIES AND MEASUREMENT",
-            "KINEMATICS",
-            "DYNAMICS",
-            "TURNING EFFECT OF FORCES",
-            "GRAVITATION",
-            "WORK AND ENERGY",
-            "PROPERTIES OF MATTER",
-            "THERMAL PROPERTIES OF MATTER",
-            "TRANSFER OF HEAT"
+            "PHYSICAL QUANTITIES AND MEASUREMENT","KINEMATICS","DYNAMICS",
+            "Turning Effect of Forces","GRAVITATION","WORK AND ENERGY",
+            "PROPERTIES OF MATTER","THERMAL PROPERTIES OF MATTER","TRANSFER OF HEAT"
         ],
-        "Islamiyat": ["Seerat of Prophet", "Faith & Beliefs", "Worship Practices"],
         "Pakistan Studies": [
-            "IDEOLOGICAL BASIS OF PAKISTAN",
-            "MAKING OF PAKISTAN",
-            "LAND AND ENVIRONMENT",
-            "HISTORY OF PAKISTAN"
+            "IDEOLOGICAL BASIS OF PAKISTAN","MAKING OF PAKISTAN",
+            "LAND AND ENVIRONMENT","HISTORY OF PAKISTAN"
         ],
-        "Computer": ["Fundamentals of Computing", "ICT Applications"]
+        "Mathematics": ["Chapter 1", "Chapter 2"],  # Placeholder
+        "English": ["Chapter 1", "Chapter 2"],
+        "Urdu": ["Chapter 1", "Chapter 2"]
     },
-
-    # ---------- Grade 10 (kept as you provided earlier for science subjects) ----------
     "Grade 10": {
-        "Mathematics": ["Advanced Algebra", "Trigonometry", "Geometry", "Probability"],
-        "English": ["Literature", "Advanced Composition", "Comprehension"],
-        "Urdu": ["ادبِ جدید", "تحقیقی مضمون"],
         "Biology": [
-            "GASEOUS EXCHANGE",
-            "HOMEOSTASIS",
-            "COORDINATION",
-            "SUPPORT AND MOVEMENT",
-            "REPRODUCTION",
-            "INHERITANCE",
-            "MAN AND HIS ENVIRONMENT",
-            "BIOTECHNOLOGY",
-            "PHARMA"
+            "GASEOUS EXCHANGE","HOMEOSTASIS","COORDINATION","SUPPORT AND MOVEMENT",
+            "REPRODUCTION","INHERITANCE","MAN AND HIS ENVIRONMENT","BIOTECHNOLOGY","PHARMA"
         ],
         "Chemistry": [
-            "CHEMICAL EQUILIBRIUM",
-            "ACIDS, BASES AND SALTS",
-            "ORGANIC CHEMISTRY",
-            "HYDROCARBONS",
-            "BIOCHEMISTRY",
-            "ENVIRONMENTAL CHEMISTRY I: THE ATMOSPHERE",
-            "ENVIRONMENTAL CHEMISTRY II: WATER",
-            "CHEMICAL INDUSTRIES"
+            "CHEMICAL EQUILIBRIUM","ACIDS, BASES AND SALTS","ORGANIC CHEMISTRY",
+            "HYDROCARBONS","BIOCHEMISTRY","ENVIRONMENTAL CHEMISTRY I: THE ATMOSPHERE",
+            "ENVIRONMENTAL CHEMISTRY II: WATER","CHEMICAL INDUSTRIES"
         ],
         "Physics": [
-            "SIMPLE HARMONIC MOTION AND WAVES",
-            "SOUND",
-            "GEOMETRICAL OPTICS",
-            "ELECTROSTATICS",
-            "CURRENT ELECTRICITY",
-            "ELECTROMAGNETISM",
-            "INTRODUCTORY ELECTRONICS",
-            "INFORMATION AND COMMUNICATION TECHNOLOGY",
-            "RADIOACTIVITY"
+            "SIMPLE HARMONIC MOTION AND WAVES","SOUND","GEOMETRICAL OPTICS","ELECTROSTATICS",
+            "CURRENT ELECTRICITY","ELECTROMAGNETISM","INTRODUCTORY ELECTRONICS",
+            "INFORMATION AND COMMUNICATION TECHNOLOGY","RADIOACTIVITY"
         ],
-        "Islamiyat": ["Islamic Studies II", "Ethics & Society", "Fiqh Basics"],
         "Pakistan Studies": [
-            "HISTORY OF PAKISTAN-II",
-            "PAKISTAN IN WORLD AFFAIRS",
-            "ECONOMIC DEVELOPMENTS",
+            "HISTORY OF PAKISTAN-II","PAKISTAN IN WORLD AFFAIRS","ECONOMIC DEVELOPMENTS",
             "POPULATION, SOCIETY AND CULTURE OF PAKISTAN"
         ],
-        "Computer": ["Programming Basics", "ICT in Society"]
-    },
-
-    # ---------- Grade 11 (Higher Secondary — XI) ----------
-    "Grade 11": {
-        "Mathematics": ["Advanced Algebra & Calculus Intro", "Coordinate Geometry", "Trigonometry", "Vectors"],
-        "English": ["Academic Writing", "Literature Studies", "Critical Reading"],
-        "Urdu": ["ادبی مطالعہ", "نظم و نثر کی تشریح"],
-        "Biology": [
-            "Cell Structure and Functions",
-            "Biological Molecules",
-            "Enzymes",
-            "Bioenergetics",
-            "Acellular Life & Microorganisms",
-            "Diversity of Plants",
-            "Diversity of Animals",
-            "Plant Form and Function",
-            "Digestion and Nutrition",
-            "Circulation and Respiration",
-            "Immunity"
-        ],
-        "Chemistry": [
-            "Stoichiometry & Chemical Calculations",
-            "Atomic Structure & Periodicity",
-            "Bonding & Molecular Shape",
-            "States of Matter (Gases, Liquids, Solids)",
-            "Solutions & Colligative Properties",
-            "Thermochemistry",
-            "Chemical Kinetics",
-            "Electrochemistry & Redox"
-        ],
-        "Physics": [
-            "Measurement & Vectors",
-            "Forces & Motion",
-            "Work & Energy",
-            "Rotational Motion",
-            "Fluid Mechanics",
-            "Oscillations & Waves",
-            "Geometrical Optics",
-            "Thermodynamics",
-            "Electrostatics",
-            "Current Electricity"
-        ],
-        "Computer": ["Programming Fundamentals", "Data Structures Intro", "Databases & File Systems"],
-        "Islamiyat": ["Advanced Seerah", "Islamic Ethics"],
-        "Pakistan Studies": [
-            "Pakistan in Geographical Perspective",
-            "Resources & Economy",
-            "Historical Overview",
-            "Political & Constitutional Development",
-            "Human Rights & Citizenship"
-        ]
+        "Mathematics": ["Chapter 1", "Chapter 2"],  # Placeholder
+        "English": ["Chapter 1", "Chapter 2"],
+        "Urdu": ["Chapter 1", "Chapter 2"]
     },
-
-    # ---------- Grade 12 (Higher Secondary — XII) ----------
-    "Grade 12": {
-        "Mathematics": ["Calculus", "Advanced Algebra", "Statistics", "Analytical Geometry"],
-        "English": ["Research Writing", "Advanced Literature", "Critical Analysis"],
-        "Urdu": ["تحقیق و تنقید", "ادبی شخصیات"],
-        "Biology": [
-            "Respiration & Exchange",
-            "Homeostasis & Coordination (Nervous & Chemical)",
-            "Support, Movement & Musculoskeletal System",
-            "Reproduction & Development",
-            "Genetics & Inheritance",
-            "Evolution & Biodiversity",
-            "Man and His Environment",
-            "Biotechnology & Its Applications",
-            "Biology and Human Welfare"
-        ],
-        "Chemistry": [
-            "s and p Block Elements",
-            "d Block & Transition Elements",
-            "Organic Reaction Mechanisms",
-            "Hydrocarbons & Functional Groups",
-            "Alcohols, Phenols, Ethers",
-            "Aldehydes & Ketones",
-            "Carboxylic Acids & Derivatives",
-            "Biochemistry (Carbohydrates, Proteins, Lipids)",
-            "Analytical & Industrial Chemistry",
-            "Environmental Chemistry"
-        ],
-        "Physics": [
-            "Electromagnetism & EM Induction",
-            "Alternating Current Circuits",
-            "Electronics & Semiconductors",
-            "Atomic & Nuclear Physics",
-            "Modern Physics Topics",
-            "Optics & Waves (advanced)"
-        ],
-        "Computer": ["Advanced Programming Concepts", "Web Technologies & Networks"],
-        "Islamiyat": ["Islamic Jurisprudence (Fiqh)", "Contemporary Islamic Thought"],
-        "Pakistan Studies": [
-            "Society & Culture",
-            "Foreign Relations",
-            "Economic Policies & Development",
-            "Contemporary Challenges & National Identity"
-        ]
-    }
+    # Placeholder for other grades (1–8, 11–12)
 }
 
-# ---------------------------
-# Mediums (kept English as before)
-# ---------------------------
-mediums = ["English"]
-
-# ---------------------------
-# MCQ generation function: uses grade, subject, chapter, difficulty, and num_questions
-# Behavior otherwise remains same (GROQ model call, parsing, docx export)
-# ---------------------------
 def generate_mcqs(grade, subject, chapter, difficulty, num_questions):
-    system_msg = "You are an expert educational test developer. Follow instructions strictly."
-    user_msg = (
-        f"Create {num_questions} MCQs for {grade} - {subject} - Chapter: '{chapter}'.\n"
-        f"Difficulty: {difficulty}.\n\n"
-        "Rules:\n"
-        "- ONLY output the MCQ items (numbered). For each MCQ provide stem and four options labeled A, B, C, D.\n"
-        "- At the end, provide an 'Answer Key' table listing question numbers and correct option letters.\n"
-        "- Do NOT include any other explanatory or introductory text.\n"
-        "- Align questions with the Pakistani National Curriculum (2006) learning outcomes for this grade.\n"
-    )
+    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 answer key separately."
 
     response = client.chat.completions.create(
         messages=[
@@ -330,29 +86,23 @@ def generate_mcqs(grade, subject, chapter, difficulty, num_questions):
     )
 
     text = response.choices[0].message.content
-    # Parse MCQs roughly by splitting on question numbers like "1." or newline number.
-    mcqs = re.split(r'\n\s*\d+\s*\.', text)
+    mcqs = re.split(r'\n?\d+\.', text)
     mcqs = [q.strip() for q in mcqs if q.strip()]
+    # Random answers as placeholder
+    answer_key = [random.choice(["A","B","C","D"]) for _ in mcqs]
 
-    # Keep your previous behavior: if generation returns fewer questions, still handle gracefully
-    # Random answer key (demo) — leave unchanged to preserve behavior
-    answer_key = [random.choice(["A", "B", "C", "D"]) for _ in mcqs]
-
-    return mcqs[:num_questions], answer_key[:num_questions], text
+    return mcqs[:num_questions], answer_key[:num_questions]
 
-# ---------------------------
-# DOCX export (MS Word) — unchanged layout but includes difficulty & MCQ count
-# ---------------------------
 def export_to_word(mcqs, answers, grade, subject, chapter, difficulty, num_questions):
     doc = Document()
-
     heading = doc.add_heading("WEEKLY/MONTHLY TEST", level=1)
     heading.alignment = WD_ALIGN_PARAGRAPH.CENTER
     run = heading.runs[0]
     run.bold = True
-    run.font.color.rgb = RGBColor(0, 128, 0)
+    run.font.color.rgb = RGBColor(0,128,0)
     run.font.size = Pt(18)
 
+    # Info table
     table = doc.add_table(rows=6, cols=2)
     table.style = 'Table Grid'
     info = [
@@ -368,84 +118,60 @@ def export_to_word(mcqs, answers, grade, subject, chapter, difficulty, num_quest
         row.cells[1].text = value
         row.cells[0].paragraphs[0].runs[0].bold = True
 
-    doc.add_paragraph("")  # spacing
-
-    # Add MCQs
+    # MCQs
     for i, q in enumerate(mcqs, start=1):
         p = doc.add_paragraph(f"{i}. {q}")
         p.alignment = WD_ALIGN_PARAGRAPH.LEFT
 
-    # Answer key table
+    # Answer Key
     doc.add_paragraph("\nAnswer Key:")
-    rows_needed = max(1, len(answers) + 1)
-    ans_table = doc.add_table(rows=rows_needed, cols=5)
+    ans_table = doc.add_table(rows=num_questions+1, cols=5)
     ans_table.style = 'Table Grid'
     headers = ["Q#", "A", "B", "C", "D"]
-    for idx, text_h in enumerate(headers):
-        ans_table.cell(0, idx).text = text_h
+    for idx, text in enumerate(headers):
+        ans_table.cell(0, idx).text = text
         ans_table.cell(0, idx).paragraphs[0].runs[0].bold = True
 
     for r, ans in enumerate(answers, start=1):
-        if r >= ans_table.rows:
-            # extend table if necessary
-            ans_table.add_row()
         ans_table.cell(r, 0).text = str(r)
-        for c, option in enumerate(["A", "B", "C", "D"], start=1):
-            ans_table.cell(r, c).text = "✔" if ans == option else ""
+        for c, option in enumerate(["A","B","C","D"], start=1):
+            ans_table.cell(r, c).text = "✔" if ans==option else ""
 
-    safe_chapter = re.sub(r'[\\/*?:"<>|]', "_", chapter)
-    file_path = f"{grade}_{subject}_{safe_chapter}_{datetime.now().strftime('%Y%m%d_%H%M%S')}.docx"
+    file_path = f"{grade}_{subject}_{chapter}_{datetime.now().strftime('%Y%m%d_%H%M%S')}.docx"
     doc.save(file_path)
     return file_path
 
-# ---------------------------
-# Handler called by the UI
-# ---------------------------
 def on_generate(grade, subject, chapter, difficulty, num_questions):
-    if not (grade and subject and chapter):
-        return "Please select Grade, Subject, and Chapter.", None
-
-    mcqs, answers, raw_text = generate_mcqs(grade, subject, chapter, difficulty, num_questions)
+    mcqs, answers = generate_mcqs(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
-# ---------------------------
 with gr.Blocks() as demo:
-    gr.Markdown("### **MCQ Test Generator — National Curriculum (2006) — Grades 1–12**")
+    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")
-
-    difficulty_dropdown = gr.Dropdown(
-        choices=["Absolute Beginner (Pretest)", "Intermediate", "Advanced"],
-        value="Intermediate",
-        label="Select Difficulty Level"
-    )
-
-    num_questions_slider = gr.Slider(minimum=5, maximum=60, step=1, value=20, label="Number of MCQs")
-
+    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 / Generated MCQs (preview)", lines=18)
-    output_file = gr.File(label="Download Test (.docx)")
+    output_text = gr.Textbox(label="Status")
+    output_file = gr.File(label="Download Test")
 
     def update_subjects(selected_grade):
-        return gr.update(choices=subjects_by_grade.get(selected_grade, []), value=None)
+        return gr.update(choices=list(chapters_by_grade_subject.get(selected_grade, {}).keys()), value=None)
+    grade_dropdown.change(update_subjects, inputs=grade_dropdown, outputs=subject_dropdown)
 
     def update_chapters(selected_subject, selected_grade):
-        chapters = chapters_by_subject_and_grade.get(selected_grade, {}).get(selected_subject, [])
-        return gr.update(choices=chapters, value=None)
-
-    grade_dropdown.change(update_subjects, inputs=grade_dropdown, outputs=subject_dropdown)
+        return gr.update(choices=chapters_by_grade_subject.get(selected_grade, {}).get(selected_subject, []), value=None)
     subject_dropdown.change(update_chapters, inputs=[subject_dropdown, grade_dropdown], outputs=chapter_dropdown)
-    grade_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_questions_slider],
+        on_generate, 
+        inputs=[grade_dropdown, subject_dropdown, chapter_dropdown, difficulty_dropdown, num_mcq_slider], 
         outputs=[output_text, output_file]
     )
 
-demo.launch(share=True, ssr_mode=False)
+demo.launch()