Update app.py

app.py

@@ -3,8 +3,9 @@ import gradio as gr
 from groq import Groq
 from datetime import datetime
 from docx import Document
-from docx.shared import Pt
+from docx.shared import Pt, Inches
 from docx.enum.text import WD_ALIGN_PARAGRAPH
+from docx.oxml.ns import qn
 
 # -------------------------
 # Initialize GROQ client
@@ -17,16 +18,14 @@ difficulty_levels = ["Pretest (Beginner)", "Intermediate", "Advanced"]
 min_mcq, max_mcq = 5, 50
 
 # -------------------------
-# Chapters by grade & subject
-# Use your complete chapters_by_subject_and_grade dictionary here (same as previous code)
+# Chapters by grade & subject (use your full dictionary here)
 # -------------------------
 chapters_by_subject_and_grade = {
-    # Full dictionary from previous code including Grades 1-12
-    # ...
+    # Paste full dictionary from your previous code (Grades 1-12)
 }
 
 # -------------------------
-# Function to generate MCQs from Groq LLaMA-3
+# Generate MCQs from Groq LLaMA-3
 # -------------------------
 def generate_mcqs(grade, subject, chapter, difficulty, num_questions):
     prompt = f"""
@@ -48,14 +47,13 @@ def generate_mcqs(grade, subject, chapter, difficulty, num_questions):
         import json
         mcq_list = json.loads(response.text)
     except:
-        # fallback: placeholder MCQs if Groq fails
         mcq_list = [{"question": f"Sample question {i} for {subject} chapter {chapter}",
                      "options": ["Option A","Option B","Option C","Option D"],
                      "answer": "A"} for i in range(1, num_questions+1)]
     
     questions = [q["question"] for q in mcq_list]
-    answers = [q["answer"] for q in mcq_list]
     options_list = [q["options"] for q in mcq_list]
+    answers = [q["answer"] for q in mcq_list]
     
     return questions, options_list, answers
 
@@ -64,13 +62,22 @@ def generate_mcqs(grade, subject, chapter, difficulty, num_questions):
 # -------------------------
 def export_to_word(mcqs, options_list, answers, grade, subject, chapter, difficulty):
     doc = Document()
+    
+    # Set document default font (Times New Roman)
+    style = doc.styles['Normal']
+    font = style.font
+    font.name = 'Times New Roman'
+    font.size = Pt(12)
+    
+    # Heading
     heading = doc.add_paragraph("MONTHLY/ WEEKLY TEST")
     heading.alignment = WD_ALIGN_PARAGRAPH.CENTER
-    run = heading.runs[0]
-    run.bold = True
-    run.font.size = Pt(16)
+    heading.runs[0].bold = True
+    heading.runs[0].font.size = Pt(16)
     
-    # Info table
+    doc.add_paragraph()  # empty line
+    
+    # Info Table
     table = doc.add_table(rows=6, cols=2)
     table.style = 'Table Grid'
     info = [
@@ -82,15 +89,22 @@ def export_to_word(mcqs, options_list, answers, grade, subject, chapter, difficu
         row.cells[1].text = value
         row.cells[0].paragraphs[0].runs[0].bold = True
     
+    doc.add_paragraph()  # empty line
+    
     # MCQs
     for i, (q, opts) in enumerate(zip(mcqs, options_list), start=1):
-        para = doc.add_paragraph(f"{i}. {q}")
+        q_para = doc.add_paragraph(f"{i}. {q}")
+        q_para.paragraph_format.space_after = Pt(2)
         for letter, text in zip(["A","B","C","D"], opts):
-            doc.add_paragraph(f"{letter}. {text}", style='List Number')
+            opt_para = doc.add_paragraph(f"{letter}. {text}")
+            opt_para.paragraph_format.left_indent = Inches(0.25)
+            opt_para.paragraph_format.space_after = Pt(1)
     
-    # Answer key table
-    doc.add_paragraph("\nAnswer Key:")
-    ans_table = doc.add_table(rows=len(mcqs)+1, cols=5)
+    doc.add_paragraph()  # empty line
+    
+    # Answer Key Table
+    doc.add_paragraph("ANSWER KEY")
+    ans_table = doc.add_table(rows=len(answers)+1, cols=5)
     ans_table.style = 'Table Grid'
     headers = ["Q#", "A", "B", "C", "D"]
     for idx, text in enumerate(headers):
@@ -106,7 +120,7 @@ def export_to_word(mcqs, options_list, answers, grade, subject, chapter, difficu
     return file_path
 
 # -------------------------
-# Generate test function
+# Generate test
 # -------------------------
 def on_generate(grade, subject, chapter, difficulty, num_questions):
     mcqs, options_list, answers = generate_mcqs(grade, subject, chapter, difficulty, num_questions)
@@ -118,26 +132,30 @@ 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")
-
+    
     # Dropdown updates
     def update_subjects(grade):
         return gr.update(choices=list(chapters_by_subject_and_grade.get(grade, {}).keys()), value=None)
+    
     def update_chapters(subject, grade):
         return gr.update(choices=chapters_by_subject_and_grade.get(grade, {}).get(subject, []), value=None)
     
     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])
 
 demo.launch()