Update app.py

app.py

@@ -1,173 +1,166 @@
-# app.py
 import os
 import gradio as gr
 from docx import Document
-from docx.oxml.ns import qn
+from docx.shared import Pt, Inches
+from docx.enum.text import WD_ALIGN_PARAGRAPH
 from docx.oxml import OxmlElement
-from docx.shared import Pt
+from docx.oxml.ns import qn
 import datetime
 from groq import Groq
 
-# ✅ Load API key from Hugging Face Secrets
+# ✅ Get Groq API Key
 GROQ_API_KEY = os.getenv("GROQ_API_KEY")
-
-# Initialize Groq client
 client = Groq(api_key=GROQ_API_KEY)
 
 
-# ✅ Function to apply Urdu font and RTL formatting
-def apply_urdu_format(paragraph):
-    """Apply right-to-left Nastaleeq font to a paragraph"""
-    run = paragraph.runs[0] if paragraph.runs else paragraph.add_run()
+# ✅ Urdu Nastaliq Font Style (Jameel or similar)
+def set_urdu_font(run):
     run.font.name = "Jameel Noori Nastaleeq"
-    r = run._element.rPr
-    r.rFonts.set(qn("w:eastAsia"), "Jameel Noori Nastaleeq")
-    r.rFonts.set(qn("w:cs"), "Jameel Noori Nastaleeq")
-    paragraph.alignment = 2  # right alignment
-    rtl = OxmlElement("w:rtl")
-    rtl.set(qn("w:val"), "1")
-    paragraph._element.insert(0, rtl)
+    r = run._element
+    r.rPr.rFonts.set(qn('w:eastAsia'), "Jameel Noori Nastaleeq")
 
 
-# ✅ Text Generation Function
+# ✅ Generate Lesson Plan Function
 def generate_lesson_plan(language, class_, subject, topic, students, duration, date, teacher):
-    try:
-        # English mode → Llama 3.1 model
-        if language == "English":
-            model_name = "llama-3.1-8b-instant"
-            prompt = f"""
-You are an expert lesson plan designer.
-Create a comprehensive English lesson plan using the BOPPPS model for the topic: "{topic}".
-Do NOT repeat the basic info (class, subject, teacher, etc.) in headings.
-Include only these sections:
-1. Bridge-In: short, engaging introduction.
-2. 3 Clear Learning Objectives.
-3. 2 Pre-Assessment Questions.
-4. 1 Detailed Participatory Learning Activity.
-5. 2–3 Post-Assessment Questions.
-6. Summary paragraph by the teacher.
-            """
-
-        # Urdu mode → OpenAI GPT-OSS-120B model
-        else:
-            model_name = "openai/gpt-oss-120b"
-            prompt = f"""
-آپ ایک ماہر اسباقی منصوبہ ساز ہیں۔
-موضوع "{topic}" پر BOPPPS ماڈل کے مطابق ایک مفصل اردو سبق منصوبہ تیار کریں۔
-براہ کرم درج ذیل حصے لازمی شامل کریں:
-1. تعارفی سرگرمی (Bridge-In)
-2. 3 واضح تعلیمی مقاصد
-3. 2 ابتدائی سوالات (Pre-assessment)
-4. ایک تفصیلی سرگرمی (Participatory Learning Activity)
-5. 2 تا 3 بعد از تدریس سوالات (Post-assessment)
-6. سبق کا خلاصہ (Summary)
-            """
-
-        # ✅ Generate text
-        response = client.chat.completions.create(
-            model=model_name,
-            messages=[{"role": "user", "content": prompt}],
-            temperature=0.8,
-            max_completion_tokens=2048,
-            top_p=1,
-            stream=False
-        )
-        lesson_text = response.choices[0].message.content.strip()
-
-        # ✅ Create Word document
-        doc = Document()
-        doc.add_heading("BOPPPS Lesson Plan", 0)
-
-        # Create info table
-        table = doc.add_table(rows=0, cols=2)
-
-        # Urdu field labels if language == Urdu
+    # ✅ Define model based on language
+    if language == "Urdu":
+        model_name = "openai/gpt-oss-120b"
+    else:
+        model_name = "llama-3.1-8b-instant"
+
+    # ✅ Create prompt
+    if language == "English":
+        prompt = f"""
+        Generate a comprehensive daily lesson plan in English.
+        Class: {class_}
+        Subject: {subject}
+        Topic: {topic}
+        Students: {students}
+        Duration: {duration}
+        Date: {date}
+        Teacher: {teacher}
+
+        The plan should include:
+        1. Learning Outcomes (SLOs)
+        2. Introduction
+        3. Teaching Methodology
+        4. Activities
+        5. Assessment
+        6. Reflection
+        7. Home Assignment aligned with SLOs.
+        """
+    else:
+        prompt = f"""
+        درج ذیل تفصیلات کی بنیاد پر ایک جامع، خوبصورت اور بامعنی روزانہ اسباق کا منصوبہ اردو زبان میں تیار کریں۔
+        جماعت: {class_}
+        مضمون: {subject}
+        عنوان: {topic}
+        طلباء کی تعداد: {students}
+        دورانیہ: {duration}
+        تاریخ: {date}
+        معلم: {teacher}
+
+        منصوبے میں درج ذیل اجزاء لازمی شامل ہوں:
+        ۱۔ تعلیمی مقاصد (SLOs)
+        ۲۔ تعارف سبق
+        ۳۔ تدریسی طریقہ کار
+        ۴۔ تعلیمی سرگرمیاں
+        ۵۔ تشخیص
+        ۶۔ عکاسی (Reflection)
+        ۷۔ گھریلو کام (Home Assignment) جو درج ذیل خوبیوں کا حامل ہو:
+            • مناسب طوالت رکھتا ہو۔
+            • منطقی ترتیب رکھتا ہو۔
+            • طلباء میں تخلیقی صلاحیت پیدا کرے۔
+            • دلچسپ اور پرکشش ہو۔
+            • طلباء کی ذہنی سطح اور جماعت کے مطابق ہو۔
+            • ایک سے دو گھنٹوں میں مکمل کیا جا سکے۔
+        """
+
+    # ✅ Get response from Groq API
+    response = client.chat.completions.create(
+        model=model_name,
+        messages=[{"role": "user", "content": prompt}],
+    )
+
+    lesson_text = response.choices[0].message.content.strip()
+
+    # ✅ Create Word Document
+    doc = Document()
+
+    if language == "English":
+        doc.add_heading("Daily Lesson Plan", level=1)
+    else:
+        p = doc.add_paragraph("یومیہ تدریسی منصوبہ")
+        p.alignment = WD_ALIGN_PARAGRAPH.RIGHT
+        run = p.runs[0]
+        set_urdu_font(run)
+        run.bold = True
+
+    # ✅ Table (Top Info)
+    table = doc.add_table(rows=6, cols=2)
+    table.autofit = True
+
+    if language == "English":
+        info = [
+            ("Class:", class_),
+            ("Subject:", subject),
+            ("Topic:", topic),
+            ("Number of Students:", students),
+            ("Duration:", duration),
+            ("Teacher:", teacher),
+        ]
+    else:
+        info = [
+            ("جماعت:", class_),
+            ("مضمون:", subject),
+            ("عنوان:", topic),
+            ("طلباء کی تعداد:", students),
+            ("دورانیہ:", duration),
+            ("معلم:", teacher),
+        ]
+
+    for i, (key, value) in enumerate(info):
+        cells = table.rows[i].cells
         if language == "Urdu":
-            data = {
-                "کلاس": class_,
-                "مضمون": subject,
-                "عنوان": topic,
-                "طلباء کی تعداد": students,
-                "دورانیہ (منٹوں میں)": duration,
-                "تاریخ": date,
-                "استاد کا نام": teacher
-            }
+            cells[0].text = value
+            cells[1].text = key
         else:
-            data = {
-                "Class": class_,
-                "Subject": subject,
-                "Topic": topic,
-                "No. of Students": students,
-                "Duration": duration,
-                "Date": date,
-                "Teacher Name": teacher
-            }
-
-        for key, value in data.items():
-            row_cells = table.add_row().cells
-            row_cells[0].text = key
-            row_cells[1].text = str(value)
-
-            # Apply Urdu formatting to both cells if Urdu mode
-            if language == "Urdu":
-                for cell in row_cells:
-                    for paragraph in cell.paragraphs:
-                        apply_urdu_format(paragraph)
-
-        doc.add_paragraph("\n")
-
-        # Add generated text paragraph
-        paragraph = doc.add_paragraph(lesson_text)
-
-        # Apply Urdu formatting
-        if language == "Urdu":
-            apply_urdu_format(paragraph)
-
-        doc.add_paragraph("\nDeveloped by Najaf Ali Sharqi")
+            cells[0].text = key
+            cells[1].text = value
 
-        filename = f"lesson_plan_{datetime.datetime.now().strftime('%Y%m%d%H%M%S')}.docx"
-        filepath = f"/tmp/{filename}"
-        doc.save(filepath)
+    # ✅ Add Lesson Plan Text
+    doc.add_paragraph("")
+    if language == "English":
+        p = doc.add_paragraph(lesson_text)
+    else:
+        p = doc.add_paragraph(lesson_text)
+        p.alignment = WD_ALIGN_PARAGRAPH.RIGHT
+        for run in p.runs:
+            set_urdu_font(run)
 
-        return filepath
+    # ✅ Save File
+    file_name = f"Lesson_Plan_{language}_{datetime.datetime.now().strftime('%Y%m%d_%H%M%S')}.docx"
+    doc.save(file_name)
 
-    except Exception as e:
-        error_doc = Document()
-        error_doc.add_heading("Error in Lesson Plan Generator", 0)
-        error_doc.add_paragraph(str(e))
-        filename = f"error_{datetime.datetime.now().strftime('%Y%m%d%H%M%S')}.docx"
-        filepath = f"/tmp/{filename}"
-        error_doc.save(filepath)
-        return filepath
+    return file_name
 
 
 # ✅ Gradio UI
-with gr.Blocks() as demo:
-    gr.Markdown("## 🌍 Bilingual BOPPPS Lesson Plan Generator (English + Urdu)")
-
-    with gr.Group():
-        language = gr.Radio(["English", "Urdu"], label="Select Language", value="English")
-        with gr.Row():
-            class_ = gr.Textbox(label="Class / کلاس", placeholder="e.g., 9th / نہم")
-            subject = gr.Textbox(label="Subject / مضمون", placeholder="e.g., Physics / فزکس")
-        with gr.Row():
-            topic = gr.Textbox(label="Topic / عنوان", placeholder="e.g., Newton’s First Law / حرکیات کا پہلا قانون")
-            students = gr.Textbox(label="Number of Students / طلباء کی تعداد", placeholder="e.g., 30")
-        with gr.Row():
-            duration = gr.Textbox(label="Duration (minutes) / دورانیہ", placeholder="e.g., 40")
-            date = gr.Textbox(label="Date / تاریخ", placeholder="e.g., 2025-10-07")
-        teacher = gr.Textbox(label="Teacher Name / استاد کا نام", placeholder="e.g., Najaf Ali Sharqi / نجف علی شرقی")
-
-        generate_button = gr.Button("Generate Lesson Plan / سبق منصوبہ تیار کریں")
-        output_file = gr.File(label="📄 Download Word File / فائل ڈاؤن لوڈ کریں")
-
-    generate_button.click(
-        fn=generate_lesson_plan,
-        inputs=[language, class_, subject, topic, students, duration, date, teacher],
-        outputs=output_file
-    )
-
-    gr.Markdown("---")
-    gr.Markdown("🔹 Developed by **Najaf Ali Sharqi**")
-
-demo.launch()
+interface = gr.Interface(
+    fn=generate_lesson_plan,
+    inputs=[
+        gr.Dropdown(["English", "Urdu"], label="Select Language"),
+        gr.Textbox(label="Class / جماعت"),
+        gr.Textbox(label="Subject / مضمون"),
+        gr.Textbox(label="Topic / عنوان"),
+        gr.Textbox(label="Number of Students / طلباء کی تعداد"),
+        gr.Textbox(label="Duration / دورانیہ"),
+        gr.Textbox(label="Date / تاریخ", value=str(datetime.date.today())),
+        gr.Textbox(label="Teacher Name / معلم کا نام"),
+    ],
+    outputs=gr.File(label="Download Lesson Plan"),
+    title="AI Lesson Plan Generator",
+    description="Generate bilingual (English / Urdu) lesson plans powered by Groq API and open models.",
+)
+
+interface.launch()