Update app.py

app.py

@@ -11,29 +11,51 @@ import re
 GROQ_API_KEY = os.getenv("GROQ_API_KEY")
 client = Groq(api_key=GROQ_API_KEY)
 
-# Sample grade, subject, topic mapping
+# Urdu number mapping
+urdu_numbers = ["۰","۱","۲","۳","۴","۵","۶","۷","۸","۹"]
+
+def convert_to_urdu_number(n):
+    return ''.join(urdu_numbers[int(d)] for d in str(n))
+
+# Grades
 grades = [f"Grade {i}" for i in range(1, 13)]
+
+# Subjects by grade according to Pakistani curriculum
 subjects_by_grade = {
-    "Grade 1": ["Mathematics", "English", "Urdu"],
-    "Grade 2": ["Mathematics", "English", "Urdu"],
-    "Grade 3": ["Mathematics", "English", "Urdu", "Science"],
-    "Grade 4": ["Mathematics", "English", "Urdu", "Science"],
+    "Grade 1": ["Mathematics", "English", "Urdu", "Islamiyat"],
+    "Grade 2": ["Mathematics", "English", "Urdu", "Islamiyat"],
+    "Grade 3": ["Mathematics", "English", "Urdu", "Science", "Islamiyat"],
+    "Grade 4": ["Mathematics", "English", "Urdu", "Science", "Islamiyat"],
     "Grade 5": ["Mathematics", "English", "Urdu", "Science", "Islamiyat"],
-    # ... add for all grades
+    "Grade 6": ["Mathematics", "English", "Urdu", "Science", "Islamiyat", "Pakistan Studies"],
+    "Grade 7": ["Mathematics", "English", "Urdu", "Science", "Islamiyat", "Pakistan Studies"],
+    "Grade 8": ["Mathematics", "English", "Urdu", "Science", "Islamiyat", "Pakistan Studies"],
+    "Grade 9": ["Mathematics", "English", "Urdu", "Physics", "Chemistry", "Biology", "Islamiyat", "Pakistan Studies"],
+    "Grade 10": ["Mathematics", "English", "Urdu", "Physics", "Chemistry", "Biology", "Islamiyat", "Pakistan Studies"],
+    "Grade 11": ["Mathematics", "English", "Urdu", "Physics", "Chemistry", "Biology", "Computer Science", "Pakistan Studies", "Islamiyat"],
+    "Grade 12": ["Mathematics", "English", "Urdu", "Physics", "Chemistry", "Biology", "Computer Science", "Pakistan Studies", "Islamiyat"]
 }
+
+# Topics by subject (example topics)
 topics_by_subject = {
-    "Mathematics": ["Addition", "Subtraction", "Multiplication"],
-    "English": ["Grammar", "Vocabulary", "Comprehension"],
+    "Mathematics": ["Algebra", "Geometry", "Fractions"],
+    "English": ["Grammar", "Comprehension", "Vocabulary"],
     "Urdu": ["Harf", "Jumla", "Qissa"],
     "Science": ["Cell Biology", "Human Body Systems", "Plants"],
-    "Islamiyat": ["Seerat", "Ahkam", "Aqaid"]
+    "Physics": ["Mechanics", "Optics", "Electricity"],
+    "Chemistry": ["Elements & Compounds", "Chemical Reactions", "Periodic Table"],
+    "Biology": ["Genetics", "Ecology", "Microbiology"],
+    "Islamiyat": ["Seerat", "Ahkam", "Aqaid"],
+    "Pakistan Studies": ["History", "Geography", "Constitution"],
+    "Computer Science": ["Programming", "Data Structures", "Networks"]
 }
+
 mediums = ["English", "Urdu"]
 
 def generate_mcqs(grade, subject, topic, medium):
     if medium == "Urdu":
         system_msg = "آپ ایک تعلیمی ماہر ہیں۔ پاکستان کے قومی نصاب کے مطابق 20 ایم سی کیو تیار کریں۔"
-        user_msg = f"{grade} {subject} کا موضوع {topic} پر 20 ایم سی کیو تیار کریں۔ ہر سوال میں 4 آپشنز شامل ہوں، درست جواب کو الگ نہ کریں۔"
+        user_msg = f"{grade} {subject} کا موضوع {topic} پر 20 ایم سی کیو تیار کریں۔ ہر سوال میں 4 آپشنز شامل ہوں، جوابات الگ نہ کریں۔"
     else:
         system_msg = "You are an educational expert. Generate 20 MCQs aligned with Pakistani National Curriculum."
         user_msg = f"Create 20 MCQs for {grade} {subject} on the topic {topic}. Each question must have 4 options."
@@ -45,15 +67,16 @@ def generate_mcqs(grade, subject, topic, medium):
         ],
         model="llama-3.3-70b-versatile"
     )
+
     text = response.choices[0].message.content
     # Split questions by number pattern
     mcqs = re.split(r'\n?\d+\.', text)
     mcqs = [q.strip() for q in mcqs if q.strip()]
-    return mcqs
+    return mcqs[:20]  # Ensure exactly 20 questions
 
 def export_to_word(mcqs, grade, subject, topic, medium):
     doc = Document()
-    # Add heading
+    # Heading
     heading = doc.add_heading("WEEKLY/MONTHLY TEST", level=1)
     heading.alignment = WD_ALIGN_PARAGRAPH.CENTER
     run = heading.runs[0]
@@ -61,7 +84,7 @@ def export_to_word(mcqs, grade, subject, topic, medium):
     run.font.color.rgb = RGBColor(0, 128, 0)
     run.font.size = Pt(18)
 
-    # Add info table
+    # Info table
     table = doc.add_table(rows=5, cols=2)
     table.style = 'Table Grid'
     info = [
@@ -78,13 +101,26 @@ def export_to_word(mcqs, grade, subject, topic, medium):
         if medium == "Urdu":
             row.cells[1].paragraphs[0].alignment = WD_ALIGN_PARAGRAPH.RIGHT
 
-    # Add MCQs
+    # MCQs
     for i, q in enumerate(mcqs, start=1):
-        p = doc.add_paragraph(f"{i}. {q}")
         if medium == "Urdu":
+            number = convert_to_urdu_number(i)
+            p = doc.add_paragraph(f"{q}  {number}.")
             p.alignment = WD_ALIGN_PARAGRAPH.RIGHT
+        else:
+            p = doc.add_paragraph(f"{i}. {q}")
+
+    # Answer key table (5 columns)
+    ans_table = doc.add_table(rows=5, cols=5)
+    ans_table.style = 'Table Grid'
+    for i in range(20):
+        row = ans_table.rows[i % 5]
+        col = i % 5
+        row.cells[col].text = f"Q{i+1}: ____"
+        if medium == "Urdu":
+            row.cells[col].paragraphs[0].alignment = WD_ALIGN_PARAGRAPH.RIGHT
 
-    # Save document
+    # Save file
     file_path = f"{grade}_{subject}_{topic}_{datetime.now().strftime('%Y%m%d_%H%M%S')}.docx"
     doc.save(file_path)
     return file_path
@@ -94,6 +130,7 @@ def on_generate(grade, subject, topic, medium):
     file_path = export_to_word(mcqs, grade, subject, topic, medium)
     return f"Test ready: {file_path}", file_path
 
+# Gradio UI
 with gr.Blocks() as demo:
     gr.Markdown("### **MCQ Test Generator - Developer: Najaf Ali Sharqi**")
     with gr.Row():
@@ -106,16 +143,17 @@ with gr.Blocks() as demo:
     output_text = gr.Textbox(label="Status")
     output_file = gr.File(label="Download Test")
 
-    # Update subjects when grade changes
+    # Update subjects dynamically
     def update_subjects(selected_grade):
-        return gr.update(choices=subjects_by_grade.get(selected_grade, []), value=None)
+        return gr.Dropdown.update(choices=subjects_by_grade.get(selected_grade, []), value=None)
     grade_dropdown.change(update_subjects, inputs=grade_dropdown, outputs=subject_dropdown)
 
-    # Update topics when subject changes
+    # Update topics dynamically
     def update_topics(selected_subject):
-        return gr.update(choices=topics_by_subject.get(selected_subject, []), value=None)
+        return gr.Dropdown.update(choices=topics_by_subject.get(selected_subject, []), value=None)
     subject_dropdown.change(update_topics, inputs=subject_dropdown, outputs=topic_dropdown)
 
-    generate_btn.click(on_generate, inputs=[grade_dropdown, subject_dropdown, topic_dropdown, medium_dropdown], outputs=[output_text, output_file])
+    generate_btn.click(on_generate, inputs=[grade_dropdown, subject_dropdown, topic_dropdown, medium_dropdown],
+                       outputs=[output_text, output_file])
 
 demo.launch(share=True, ssr_mode=False)