Update app.py

app.py

@@ -1,122 +1,165 @@
 import gradio as gr
 from google import genai
 from docx import Document
+from docx.oxml import parse_xml
+from docx.oxml.ns import qn
+from docx.enum.text import WD_ALIGN_PARAGRAPH
 from datetime import date
 
 # Initialize Gemini client
 client = genai.Client()
 
 def generate_lesson_plan(language, grade, subject, topic, students, duration, teacher_name):
-    # Build the structured prompt
+    # ===== Prompt =====
     prompt = f"""
 You are a qualified and experienced school teacher in Pakistan, well-versed in the National Curriculum of Pakistan. 
 Your task is to create a complete and structured "SMART LESSON PLAN" in the language requested by the user (either Urdu or English).
 
-Follow the exact format and structure below without using any symbols or special characters.
-Use clear, teacher-friendly, classroom-appropriate language.
-If the selected language is Urdu, write in right-to-left direction using Jameel Noori Nastaleeq font style and write English terms in Urdu script (for example, "Science" as "سائنس", "Homework" as "ہوم ورک").
-Ensure the lesson content aligns with the given topic and is relevant to Pakistani classrooms.
-
-Generate the lesson plan as follows:
+If Urdu is selected, write right-to-left in Jameel Noori Nastaleeq style, with English words in Urdu script (for example, "Science" as "سائنس"). 
+Avoid all symbols, bullets, or special characters. 
+Generate only the following sections once, clearly separated and formatted:
 
 SMART LESSON PLAN
 
 A) Students Learning Outcomes:
 By the end of this lesson, students will be able to:
-1. Write the first clear and measurable SLO aligned with the topic "{topic}".
-2. Write the second clear and measurable SLO aligned with the topic "{topic}".
-3. Write the third clear and measurable SLO aligned with the topic "{topic}".
+1. Write the first SLO related to "{topic}".
+2. Write the second SLO related to "{topic}".
+3. Write the third SLO related to "{topic}".
 
 B) Content to be delivered:
-Write detailed and meaningful content based on "{topic}" and the SLOs, including clear definitions, short explanations, and at least one example that can be taught in the classroom. 
-Ensure the content is conceptually sound and suitable for grade {grade} students.
+Write detailed and clear content (definitions, explanations, and one example) about "{topic}" aligned with the SLOs. 
+Ensure it fits Grade {grade} Pakistani context.
 
 C) Home Assignment:
-Create a simple, practical, and creative home assignment aligned with the above SLOs. 
-The assignment should be doable within 24 hours and promote understanding and application of the topic. 
-Write actual directions that a teacher would give to students.
+Create a simple and creative home assignment aligned with SLOs that can be completed within 24 hours.
 
 D) Feedback by the Mentor/ Principal:
-(Leave four continuous blank lines for handwritten feedback with approximately 2.5 line spacing.)
-    """
+(Leave four continuous blank lines.)
+"""
 
-    # Generate lesson content using Gemini
+    # ===== Generate Response =====
     response = client.models.generate_content(
         model="gemini-2.5-flash",
         contents=prompt,
     )
-    lesson_text = response.text
-
-    # Create Word Document
-    doc = Document()
-    doc.add_heading("SMART LESSON PLAN", level=1)
-
-    # Input Table
-    table = doc.add_table(rows=8, cols=2)
-    table.style = 'Table Grid'
-    fields = [
-        ("Language", language),
-        ("Grade Level", grade),
-        ("Subject", subject),
-        ("Topic", topic),
-        ("Number of Students", students),
-        ("Duration", f"{duration} minutes"),
-        ("Teacher Name", teacher_name),
-        ("Date", str(date.today()))
-    ]
-    for i, (label, value) in enumerate(fields):
-        table.cell(i, 0).text = label
-        table.cell(i, 1).text = value
-
-    # Add lesson sections
-    doc.add_paragraph()
-    doc.add_paragraph("A) Students Learning Outcomes:", style='Heading2')
-    doc.add_paragraph("By the end of this lesson, students will be able to:")
-    doc.add_paragraph(lesson_text.split("B)")[0].strip())
-
-    doc.add_paragraph()
-    doc.add_paragraph("B) Content to be delivered:", style='Heading2')
-    content_section = lesson_text.split("B)")[1].split("C)")[0].strip()
-    doc.add_paragraph(content_section)
-
-    doc.add_paragraph()
-    doc.add_paragraph("C) Home Assignment:", style='Heading2')
-    assignment_section = lesson_text.split("C)")[1].split("D)")[0].strip()
-    doc.add_paragraph(assignment_section)
-
-    doc.add_paragraph()
-    doc.add_paragraph("D) Feedback by the Mentor/ Principal:", style='Heading2')
-    for _ in range(4):
-        doc.add_paragraph(" ")
-
-    # Save document
-    file_name = f"SMART_Lesson_Plan_{topic.replace(' ', '_')}.docx"
-    doc.save(file_name)
-
-    return lesson_text, file_name
-
-
-# Gradio UI
+    lesson_text = response.text.strip()
+
+    # ===== File Generation Based on Language =====
+    if language == "English":
+        # Create Word Document
+        doc = Document()
+
+        # Header
+        section = doc.sections[0]
+        header = section.header
+        header_para = header.paragraphs[0]
+        header_para.text = "School Education Department, Gilgit-Baltistan"
+        header_para.alignment = WD_ALIGN_PARAGRAPH.CENTER
+
+        # Footer with page numbers
+        footer = section.footer
+        footer_para = footer.paragraphs[0]
+        footer_para.alignment = WD_ALIGN_PARAGRAPH.CENTER
+        footer_para.add_run("Page ").add_field('PAGE')
+        footer_para.add_run(" of ").add_field('NUMPAGES')
+
+        # Title
+        doc.add_heading("SMART LESSON PLAN", level=1).alignment = WD_ALIGN_PARAGRAPH.CENTER
+
+        # Input Table
+        table = doc.add_table(rows=8, cols=2)
+        table.style = 'Table Grid'
+        fields = [
+            ("Language", language),
+            ("Grade Level", grade),
+            ("Subject", subject),
+            ("Topic", topic),
+            ("Number of Students", students),
+            ("Duration", f"{duration} minutes"),
+            ("Teacher Name", teacher_name),
+            ("Date", str(date.today()))
+        ]
+        for i, (label, value) in enumerate(fields):
+            table.cell(i, 0).text = label
+            table.cell(i, 1).text = value
+
+        # Add Lesson Content
+        doc.add_paragraph()
+        for line in lesson_text.splitlines():
+            doc.add_paragraph(line)
+
+        # Save DOCX
+        file_name = f"SMART_Lesson_Plan_{topic.replace(' ', '_')}.docx"
+        doc.save(file_name)
+        return lesson_text, file_name
+
+    else:
+        # Urdu version: Markdown format (Right-to-Left)
+        md_content = f"""
+SMART LESSON PLAN
+
+زبان: اردو  
+جماعت: {grade}  
+مضمون: {subject}  
+موضوع: {topic}  
+طلباء کی تعداد: {students}  
+دورانیہ: {duration} منٹ  
+استاد کا نام: {teacher_name}  
+تاریخ: {date.today()}  
+
+---
+
+{lesson_text}
+
+---
+
+ڈیزائن کیا گیا از: نجف علی شرقی  
+School Education Department, Gilgit-Baltistan
+"""
+        # Save as Markdown
+        file_name = f"SMART_Lesson_Plan_{topic.replace(' ', '_')}.md"
+        with open(file_name, "w", encoding="utf-8") as f:
+            f.write(md_content)
+
+        return lesson_text, file_name
+
+
+# ===== Gradio UI =====
 with gr.Blocks(title="SMART Lesson Plan Generator") as demo:
     gr.Markdown("<h2 style='text-align:center;'>SMART Lesson Plan Generator</h2>")
-    gr.Markdown("<p style='text-align:center;'>Generate complete, structured lesson plans in Urdu or English for Pakistani classrooms.</p>")
+    gr.Markdown("<p style='text-align:center;'>Create structured lesson plans in Urdu or English for Pakistani classrooms.</p>")
 
     with gr.Row():
         language = gr.Dropdown(["English", "Urdu"], label="Select Language")
-        grade = gr.Textbox(label="Grade Level (e.g., Grade 8)")
-        subject = gr.Textbox(label="Subject (e.g., Science)")
+        grade = gr.Dropdown(
+            ["Grade 1", "Grade 2", "Grade 3", "Grade 4", "Grade 5", "Grade 6", "Grade 7", "Grade 8", "Grade 9", "Grade 10", "Grade 11", "Grade 12"],
+            label="Select Grade Level"
+        )
+
+    with gr.Row():
+        subject = gr.Dropdown(
+            ["Science", "Mathematics", "English", "Urdu", "Islamiyat", "Computer", "Physics", "Chemistry", "Biology", "Social Studies"],
+            label="Select Subject"
+        )
+        topic = gr.Textbox(label="Enter Topic (e.g., Chemical Bonding)")
 
     with gr.Row():
-        topic = gr.Textbox(label="Topic (e.g., Chemical Bonding)")
-        students = gr.Textbox(label="Number of Students")
-        duration = gr.Textbox(label="Duration (in minutes)")
+        students = gr.Dropdown(
+            ["20", "25", "30", "35", "40", "45", "50+"],
+            label="Number of Students"
+        )
+        duration = gr.Dropdown(
+            ["30", "35", "40", "45", "50", "60"],
+            label="Duration (in minutes)"
+        )
 
     with gr.Row():
         teacher_name = gr.Textbox(label="Teacher Name")
 
     generate_btn = gr.Button("Generate Lesson Plan")
     output_text = gr.Textbox(label="Generated Lesson Plan", lines=25)
-    download_file = gr.File(label="Download MS Word File")
+    download_file = gr.File(label="Download File")
 
     generate_btn.click(
         generate_lesson_plan,
@@ -124,7 +167,7 @@ with gr.Blocks(title="SMART Lesson Plan Generator") as demo:
         outputs=[output_text, download_file],
     )
 
-    gr.Markdown("<hr><p style='text-align:center;'>Developed by <b>Najaf Ali Sharqi</b></p>")
+    gr.Markdown("<hr><p style='text-align:center;'>Developed by <b>Najaf Ali Sharqi</b> | School Education Department, Gilgit-Baltistan</p>")
 
 demo.launch()