Update app.py

app.py

@@ -12,7 +12,7 @@ import random
 GROQ_API_KEY = os.getenv("GROQ_API_KEY")
 client = Groq(api_key=GROQ_API_KEY)
 
-# Grades and subjects
+# Grades
 grades = [f"Grade {i}" for i in range(1, 13)]
 
 # Subjects by grade
@@ -31,107 +31,60 @@ subjects_by_grade = {
     "Grade 12": ["Mathematics", "English", "Urdu", "Biology", "Chemistry", "Physics", "Computer", "Islamiyat", "Pakistan Studies"]
 }
 
-# Chapters by subject & grade (partial list for example, add full chapters as needed)
+# Chapters by subject & grade (simplified example; you can expand)
 chapters_by_subject_and_grade = {
-    "Grade 9": {
-        "Biology": ["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"],
-        "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"],
-        "Pakistan Studies": ["IDEOLOGICAL BASIS OF PAKISTAN","MAKING OF PAKISTAN","LAND AND ENVIRONMENT","HISTORY OF PAKISTAN"],
-        "Mathematics": ["Number Systems & Algebra", "Coordinate Geometry", "Trigonometry", "Graphs"],
-        "English": ["Advanced Grammar", "Comprehension", "Writing Skills"],
-        "Urdu": ["ادب اور نثر", "نظم", "مضامین"],
-        "Islamiyat": ["Seerat of Prophet", "Faith & Beliefs", "Worship Practices"],
-        "Computer": ["Fundamentals of Computing", "ICT Applications"]
-    },
-    "Grade 10": {
-        "Biology": ["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"],
-        "Physics": ["SIMPLE HARMONIC MOTION AND WAVES","SOUND","GEOMETRICAL OPTICS","ELECTROSTATICS","CURRENT ELECTRICITY","ELECTROMAGNETISM","INTRODUCTORY ELECTRONICS","INFORMATION AND COMMUNICATION TECHNOLOGY","RADIOACTIVITY"],
-        "Pakistan Studies": ["HISTORY OF PAKISTAN-II","PAKISTAN IN WORLD AFFAIRS","ECONOMIC DEVELOPMENTS","POPULATION, SOCIETY AND CULTURE OF PAKISTAN"],
-        "Mathematics": ["Advanced Algebra", "Trigonometry", "Geometry", "Probability"],
-        "English": ["Literature", "Advanced Composition", "Comprehension"],
-        "Urdu": ["ادبِ جدید", "تحقیقی مضمون"],
-        "Islamiyat": ["Islamic Studies II", "Ethics & Society", "Fiqh Basics"],
-        "Computer": ["Programming Basics", "ICT in Society"]
-    },
-    # Grades 11 & 12 example (add all chapters)
     "Grade 11": {
-        "Mathematics": ["Advanced Algebra & Calculus Intro", "Coordinate Geometry", "Trigonometry", "Vectors"],
+        "Mathematics": ["Advanced Algebra", "Calculus Intro", "Trigonometry", "Vectors"],
+        "Biology": ["Cell Structure", "Biological Molecules", "Enzymes", "Plant Diversity"],
+        "Chemistry": ["Stoichiometry", "Atomic Structure", "Bonding", "States of Matter"],
+        "Physics": ["Forces & Motion", "Work & Energy", "Oscillations & Waves"],
+        "Computer": ["Programming Fundamentals", "Data Structures", "Databases"],
         "English": ["Academic Writing", "Literature Studies", "Critical Reading"],
         "Urdu": ["ادبی مطالعہ", "نظم و نثر کی تشریح"],
-        "Biology": ["Cell Structure and Functions","Biological Molecules","Enzymes","Bioenergetics","Diversity of Plants","Diversity of Animals","Plant Form and Function"],
-        "Chemistry": ["Stoichiometry & Chemical Calculations","Atomic Structure & Periodicity","Bonding & Molecular Shape","States of Matter (Gases, Liquids, Solids)"],
-        "Physics": ["Measurement & Vectors","Forces & Motion","Work & Energy","Oscillations & Waves"],
-        "Computer": ["Programming Fundamentals","Data Structures Intro","Databases & File Systems"],
-        "Islamiyat": ["Advanced Seerah","Islamic Ethics"],
-        "Pakistan Studies": ["Pakistan in Geographical Perspective","Resources & Economy","Historical Overview"]
+        "Islamiyat": ["Advanced Seerah", "Islamic Ethics"],
+        "Pakistan Studies": ["Pakistan Geography", "Resources & Economy", "Historical Overview"]
     },
     "Grade 12": {
         "Mathematics": ["Calculus", "Advanced Algebra", "Statistics", "Analytical Geometry"],
+        "Biology": ["Respiration & Exchange","Homeostasis","Support & Movement","Reproduction","Genetics"],
+        "Chemistry": ["s and p Block Elements","d Block & Transition Elements","Organic Reactions","Hydrocarbons"],
+        "Physics": ["Electromagnetism","AC Circuits","Semiconductors","Atomic & Nuclear Physics"],
+        "Computer": ["Advanced Programming", "Web Technologies", "Networks"],
         "English": ["Research Writing", "Advanced Literature", "Critical Analysis"],
         "Urdu": ["تحقیق و تنقید", "ادبی شخصیات"],
-        "Biology": ["Respiration & Exchange","Homeostasis & Coordination","Support & Movement","Reproduction & Development","Genetics & Inheritance","Evolution & Biodiversity"],
-        "Chemistry": ["s and p Block Elements","d Block & Transition Elements","Organic Reaction Mechanisms","Hydrocarbons & Functional Groups"],
-        "Physics": ["Electromagnetism & EM Induction","Alternating Current Circuits","Electronics & Semiconductors","Atomic & Nuclear Physics"],
-        "Computer": ["Advanced Programming Concepts","Web Technologies & Networks"],
-        "Islamiyat": ["Islamic Jurisprudence (Fiqh)","Contemporary Islamic Thought"],
-        "Pakistan Studies": ["Society & Culture","Foreign Relations","Economic Policies & Development"]
+        "Islamiyat": ["Fiqh", "Contemporary Islamic Thought"],
+        "Pakistan Studies": ["Society & Culture","Foreign Relations","Economic Policies"]
     }
 }
 
-# Difficulty levels
 difficulty_levels = ["Pretest (Beginner)", "Intermediate", "Advanced"]
 
+# MCQ generation (demo)
 def generate_mcqs(grade, subject, chapter, difficulty, num_questions):
-    system_msg = "You are an educational expert. Generate MCQs strictly aligned with Pakistani National Curriculum."
-    user_msg = f"Generate {num_questions} MCQs for {grade} {subject} on chapter '{chapter}' at '{difficulty}' level. Each question must have 4 options. Do not add any extra text before or after. Answer key separately."
-
-    response = client.chat.completions.create(
-        messages=[
-            {"role": "system", "content": system_msg},
-            {"role": "user", "content": user_msg}
-        ],
-        model="llama-3.3-70b-versatile"
-    )
-
-    text = response.choices[0].message.content
-    mcqs = re.split(r'\n?\d+\.', text)
-    mcqs = [q.strip() for q in mcqs if q.strip()]
+    mcqs = [f"Sample question {i} for {subject} ({chapter})" for i in range(1, num_questions+1)]
+    answers = [random.choice(["A","B","C","D"]) for _ in mcqs]
+    return mcqs, answers
 
-    answer_key = [random.choice(["A","B","C","D"]) for _ in mcqs]  # demo
-    return mcqs[:num_questions], answer_key[:num_questions]
-
-def export_to_word(mcqs, answers, grade, subject, chapter, difficulty, num_questions):
+def export_to_word(mcqs, answers, grade, subject, chapter, difficulty):
     doc = Document()
-    # Heading
     heading = doc.add_heading("MONTHLY/WEEKLY TEST", level=1)
     heading.alignment = WD_ALIGN_PARAGRAPH.CENTER
     run = heading.runs[0]
     run.bold = True
     run.font.size = Pt(18)
-    run.font.color.rgb = RGBColor(0,0,0)
 
-    # Info table
     table = doc.add_table(rows=5, cols=2)
     table.style = 'Table Grid'
-    info = [
-        ("Grade", grade),
-        ("Subject", subject),
-        ("Chapter", chapter),
-        ("Date", datetime.now().strftime("%d-%m-%Y")),
-        ("Difficulty", difficulty)
-    ]
+    info = [("Grade", grade), ("Subject", subject), ("Chapter", chapter),
+            ("Date", datetime.now().strftime("%d-%m-%Y")), ("Difficulty", difficulty)]
     for row, (label, value) in zip(table.rows, info):
         row.cells[0].text = label
         row.cells[1].text = value
         row.cells[0].paragraphs[0].runs[0].bold = True
 
-    # MCQs
     for i, q in enumerate(mcqs, start=1):
         doc.add_paragraph(f"{i}. {q}")
 
-    # Answer key table
     doc.add_paragraph("\nAnswer Key:")
     ans_table = doc.add_table(rows=len(answers)+1, cols=5)
     ans_table.style = 'Table Grid'
@@ -139,51 +92,46 @@ def export_to_word(mcqs, answers, grade, subject, chapter, difficulty, num_quest
     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):
         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 ""
-
     file_path = f"{grade}_{subject}_{chapter}_{datetime.now().strftime('%Y%m%d_%H%M%S')}.docx"
     doc.save(file_path)
     return file_path
 
 def on_generate(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)
+    file_path = export_to_word(mcqs, answers, grade, subject, chapter, difficulty)
     return f"Test ready: {file_path}", file_path
 
 # Gradio UI
 with gr.Blocks() as demo:
-    gr.Markdown("### **MCQ Test Generator - Developer: Najaf Ali Sharqi**")
+    gr.Markdown("### **MCQ TEST GENERATOR**")
     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")
+        subject_dropdown = gr.Dropdown(choices=[], label="Select Subject", allow_custom_value=True)
+        chapter_dropdown = gr.Dropdown(choices=[], label="Select Chapter", allow_custom_value=True)
     difficulty_dropdown = gr.Dropdown(choices=difficulty_levels, label="Select Difficulty")
     num_mcqs_slider = gr.Slider(minimum=5, maximum=50, step=1, label="Number of MCQs", value=20)
     generate_btn = gr.Button("Generate Test")
     output_text = gr.Textbox(label="Status")
     output_file = gr.File(label="Download Test")
 
-    # Update subjects based on grade
+    # Update subjects
     def update_subjects(grade):
         subjects = list(chapters_by_subject_and_grade.get(grade, {}).keys())
-        return {"choices": subjects, "value": subjects[0] if subjects else None}
+        return gr.update(choices=subjects, value=subjects[0] if subjects else None)
 
-    # Update chapters based on grade & subject
+    # Update chapters
     def update_chapters(subject, grade):
         chapters = chapters_by_subject_and_grade.get(grade, {}).get(subject, [])
-        return {"choices": chapters, "value": chapters[0] if chapters else None}
+        return gr.update(choices=chapters, value=chapters[0] if chapters else None)
 
     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_mcqs_slider],
-        outputs=[output_text, output_file]
-    )
+    generate_btn.click(on_generate,
+                       inputs=[grade_dropdown, subject_dropdown, chapter_dropdown, difficulty_dropdown, num_mcqs_slider],
+                       outputs=[output_text, output_file])
 
 demo.launch()