Update app.py

app.py

@@ -1,35 +1,47 @@
 import os
 import gradio as gr
-from groq import Groq
 from datetime import datetime
 from docx import Document
 from docx.shared import Pt, Inches
 from docx.enum.text import WD_ALIGN_PARAGRAPH
 
 # -------------------------
-# Initialize GROQ client
+# Grades and Difficulty
 # -------------------------
-GROQ_API_KEY = os.getenv("GROQ_API_KEY")
-client = Groq(api_key=GROQ_API_KEY)
-
 grades = [f"Grade {i}" for i in range(1, 13)]
 difficulty_levels = ["Pretest (Beginner)", "Intermediate", "Advanced"]
 min_mcq, max_mcq = 5, 50
 
 # -------------------------
-# Chapters by grade & subject (use your full dictionary here)
+# Subjects and Chapters
+# Use full dictionary for grades 1-12
 # -------------------------
 chapters_by_subject_and_grade = {
-    # Paste the full dictionary from your previous code (Grades 1-12 with all subjects & chapters)
+    # ----------------- Grades 1-8 (example) -----------------
+    "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"]
+    },
+    # ... add all remaining grades 3-12 using your previous dictionary
 }
 
 # -------------------------
-# Generate MCQs (dummy placeholder or Groq call)
+# Dummy MCQ generator (replace with actual AI API)
 # -------------------------
 def generate_mcqs(grade, subject, chapter, difficulty, num_questions):
     mcqs = [f"Sample question {i} for {subject} chapter {chapter}" for i in range(1, num_questions+1)]
-    options_list = [["Option A", "Option B", "Option C", "Option D"] for _ in range(num_questions)]
-    answers = ["A" for _ in range(num_questions)]
+    options_list = [["Option A", "Option B", "Option C", "Option D"] for _ in mcqs]
+    answers = ["A"] * num_questions
     return mcqs, options_list, answers
 
 # -------------------------
@@ -38,41 +50,42 @@ def generate_mcqs(grade, subject, chapter, difficulty, num_questions):
 def export_to_word(mcqs, options_list, answers, grade, subject, chapter, difficulty):
     doc = Document()
     style = doc.styles['Normal']
-    font = style.font
-    font.name = 'Times New Roman'
-    font.size = Pt(12)
-    
+    style.font.name = 'Times New Roman'
+    style.font.size = Pt(12)
+
+    # Heading
     heading = doc.add_paragraph("MONTHLY/WEEKLY TEST")
     heading.alignment = WD_ALIGN_PARAGRAPH.CENTER
     heading.runs[0].bold = True
     heading.runs[0].font.size = Pt(16)
-    
-    doc.add_paragraph()
-    
+
+    doc.add_paragraph()  # empty line
+
+    # Info Table
     table = doc.add_table(rows=6, cols=2)
     table.style = 'Table Grid'
     info = [
-        ("Grade", grade),
-        ("Subject", subject),
-        ("Chapter", chapter),
-        ("Difficulty", difficulty),
-        ("Number of Questions", str(len(mcqs))),
+        ("Grade", grade), ("Subject", subject), ("Chapter", chapter),
+        ("Difficulty", difficulty), ("Number of Questions", str(len(mcqs))),
         ("Date", datetime.now().strftime("%d-%m-%Y"))
     ]
     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
-    
-    doc.add_paragraph()
-    
+
+    doc.add_paragraph()  # empty line
+
+    # MCQs
     for i, (q, opts) in enumerate(zip(mcqs, options_list), start=1):
         doc.add_paragraph(f"{i}. {q}")
         for letter, text in zip(["A","B","C","D"], opts):
             p = doc.add_paragraph(f"{letter}. {text}")
             p.paragraph_format.left_indent = Inches(0.25)
-    
-    doc.add_paragraph()
+
+    doc.add_paragraph()  # empty line
+
+    # Answer Key
     doc.add_paragraph("ANSWER KEY")
     ans_table = doc.add_table(rows=len(answers)+1, cols=5)
     ans_table.style = 'Table Grid'
@@ -84,7 +97,7 @@ def export_to_word(mcqs, options_list, answers, grade, subject, chapter, difficu
         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
@@ -102,35 +115,33 @@ def on_generate(grade, subject, chapter, difficulty, num_questions):
 # -------------------------
 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")
-    
-    # -------------------------
-    # Dynamic dropdown updates
-    # -------------------------
-    def update_subjects(selected_grade):
-        subjects = list(chapters_by_subject_and_grade.get(selected_grade, {}).keys())
+
+    # Dropdown updates
+    def update_subjects(grade):
+        subjects = list(chapters_by_subject_and_grade.get(grade, {}).keys())
         return gr.update(choices=subjects, value=subjects[0] if subjects else None)
-    
-    def update_chapters(selected_subject, selected_grade):
-        chapters = chapters_by_subject_and_grade.get(selected_grade, {}).get(selected_subject, [])
+
+    def update_chapters(subject, grade):
+        chapters = chapters_by_subject_and_grade.get(grade, {}).get(subject, [])
         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, 
+
+    grade_dropdown.change(update_subjects, grade_dropdown, subject_dropdown)
+    subject_dropdown.change(update_chapters, [subject_dropdown, grade_dropdown], chapter_dropdown)
+
+    generate_btn.click(on_generate,
                        inputs=[grade_dropdown, subject_dropdown, chapter_dropdown, difficulty_dropdown, num_mcq_slider],
                        outputs=[output_text, output_file])