Update app.py

app.py

@@ -1,103 +1,165 @@
 import os
 import gradio as gr
+import datetime
+from tempfile import NamedTemporaryFile
+from docx import Document
 from groq import Groq
 
-# ✅ Load Groq API Key
+# ✅ Initialize Groq client
 client = Groq(api_key=os.getenv("GROQ_API_KEY"))
 
-# ✅ Define models
 URDU_MODEL = "openai/gpt-oss-120b"
 ENGLISH_MODEL = "llama-3.1-8b-instant"
 
 
-# ✅ Core Function
-def generate_content(language, plan_type, feature, grade, subject, topic, duration, student_level):
+# ✅ Function: Generate SMART Lesson Plan
+def generate_lesson_plan(language, grade, subject, topic, duration, student_level):
     if language == "Urdu":
         model = URDU_MODEL
         prompt = f"""
-        آپ ایک ماہر استاد ہیں۔ براہ کرم نیچے دی گئی معلومات کی بنیاد پر SMART طرز کا {plan_type} تیار کریں۔
+        آپ ایک ماہر استاد ہیں۔ دی گئی معلومات کی بنیاد پر ایک SMART طرز کا تدریسی منصوبہ تیار کریں۔
         درج ذیل حصے شامل ہوں:
         1. معلوماتی جدول (Input Table)
-        2. SMART سیکھنے کے نتائج (SLOs)
-        3. "پڑھایا جانے والا مواد" (Content to be delivered) — تعریفیں، کلیدی تصورات اور بنیادی اصطلاحات کے ساتھ۔
-        4. گھریلو کام (Home Assignment) — SLOs کے مطابق۔
-        5. سرپرست یا پرنسپل کے تاثرات کے لیے جگہ (Feedback by Mentor/Principal)
+        2. SMART تعلیمی مقاصد (SLOs)
+        3. "پڑھایا جانے والا مواد" (Content to be delivered): تعریفیں، کلیدی اصطلاحات اور تصورات۔
+        4. گھریلو کام (Home Assignment) — جو SLOs سے مطابقت رکھتا ہو۔
+        5. سرپرست یا پرنسپل کی رائے کے لیے خالی جگہ (Feedback by Mentor/Principal)
 
         جماعت: {grade}
         مضمون: {subject}
         موضوع: {topic}
         دورانیہ: {duration} منٹ
-        طلبہ کی سطح: {student_level}
-        اگر {feature} منتخب کیا گیا ہے تو اس کے مطابق مواد بھی تیار کریں۔
+        طلباء کی سطح: {student_level}
         """
     else:
         model = ENGLISH_MODEL
         prompt = f"""
-        You are an experienced Pakistani school teacher. 
-        Please create a SMART-style {plan_type} lesson plan based on the given details.
+        You are an expert Pakistani school teacher. Create a SMART-style lesson plan with the following structure:
 
-        Include only the following sections:
-        1. Input Table (with basic information)
+        1. Input Table
         2. SMART Learning Outcomes (SLOs)
-        3. "Content to be Delivered" — include definitions, key terms, and core concepts.
-        4. Home Assignment aligned with SLOs.
-        5. Feedback by Mentor/Principal (leave blank space for handwritten feedback).
+        3. Content to be Delivered — include definitions, key terms, and key concepts related to the topic.
+        4. Home Assignment — must directly align with the SLOs.
+        5. Feedback by Mentor/Principal (leave blank space for handwritten notes)
 
         Grade: {grade}
         Subject: {subject}
         Topic: {topic}
         Duration: {duration} minutes
         Student Level: {student_level}
-        Also, if {feature} is selected, generate relevant {feature} content aligned with SLOs.
         """
 
-    try:
-        response = client.chat.completions.create(
-            model=model,
-            messages=[{"role": "user", "content": prompt}],
-            temperature=0.6,
-            max_tokens=1800
-        )
-        return response.choices[0].message.content.strip()
-    except Exception as e:
-        return f"❌ Error: {str(e)}"
+    response = client.chat.completions.create(
+        model=model,
+        messages=[{"role": "user", "content": prompt}],
+        temperature=0.7,
+        max_tokens=1800
+    )
+    content = response.choices[0].message.content.strip()
 
+    # ✅ Create downloadable file
+    if language == "English":
+        doc = Document()
+        doc.add_heading("SMART LESSON PLAN", level=1)
 
-# ✅ Build Gradio Interface
-with gr.Blocks(theme=gr.themes.Soft(primary_hue="cyan")) as demo:
-    gr.Markdown("## 🧠 SMART LESSON PLAN GENERATOR")
-    gr.Markdown("**Developed by: Najaf Ali**")
+        table = doc.add_table(rows=5, cols=2)
+        headers = ["Grade", "Subject", "Topic", "Duration", "Student Level"]
+        values = [grade, subject, topic, f"{duration} minutes", student_level]
+        for i in range(5):
+            table.cell(i, 0).text = headers[i]
+            table.cell(i, 1).text = str(values[i])
 
-    with gr.Row():
-        language = gr.Dropdown(
-            ["English", "Urdu"],
-            label="Select Language",
-            value="English"
-        )
+        doc.add_paragraph("\n" + content)
+        doc.add_paragraph("\nDeveloped by: Najaf Ali")
 
-        plan_type = gr.Dropdown(
-            ["SMART Lesson Plan", "Daily Lesson Plan", "Weekly Lesson Plan", "Unit Plan"],
-            label="Select Lesson Plan Type",
-            value="SMART Lesson Plan"
-        )
+        temp = NamedTemporaryFile(delete=False, suffix=".docx")
+        doc.save(temp.name)
+        file_path = temp.name
+        message = f"✅ English SMART Lesson Plan generated successfully for {subject} ({grade})."
+
+    else:
+        md_text = f"""
+<div dir="rtl" align="right">
+
+# 📘 SMART LESSON PLAN
+
+| کلیدی معلومات | تفصیل |
+|----------------|--------|
+| جماعت | {grade} |
+| مضمون | {subject} |
+| عنوان | {topic} |
+| دورانیہ | {duration} منٹ |
+| طلبہ کی سطح | {student_level} |
+
+---
+
+{content}
+
+---
+
+**مرتب کردہ: نجف علی**
+</div>
+"""
+        temp = NamedTemporaryFile(delete=False, suffix=".md", mode="w", encoding="utf-8")
+        temp.write(md_text)
+        temp.close()
+        file_path = temp.name
+        message = f"✅ اردو تدریسی منصوبہ ({subject}) کامیابی سے تیار ہو گیا۔"
+
+    return message, file_path, content
+
+
+# ✅ Function: Generate Flashcards
+def generate_flashcards(language, topic, slos_text):
+    model = URDU_MODEL if language == "Urdu" else ENGLISH_MODEL
+
+    prompt = f"""
+    Based on the topic "{topic}" and the following SLOs:
+    {slos_text}
+    Generate 5 short and clear **Flashcards** (Question/Term on one side and Answer/Definition on the other side).
+    Ensure alignment with the SLOs and curriculum context of Pakistan.
+    """
+    response = client.chat.completions.create(
+        model=model,
+        messages=[{"role": "user", "content": prompt}],
+        temperature=0.6,
+        max_tokens=800
+    )
+    return response.choices[0].message.content.strip()
+
+
+# ✅ Function: Generate Quiz
+def generate_quiz(language, topic, slos_text):
+    model = URDU_MODEL if language == "Urdu" else ENGLISH_MODEL
+
+    prompt = f"""
+    Based on the topic "{topic}" and these SLOs:
+    {slos_text}
+    Create a short quiz including:
+    - 5 Multiple Choice Questions (MCQs)
+    - 2 Short Answer Questions
+    - 2 Fill-in-the-Blank Questions
+    Ensure all are aligned with the SLOs and Pakistani curriculum.
+    """
+    response = client.chat.completions.create(
+        model=model,
+        messages=[{"role": "user", "content": prompt}],
+        temperature=0.7,
+        max_tokens=800
+    )
+    return response.choices[0].message.content.strip()
 
-        feature = gr.Dropdown(
-            ["None", "Flashcards", "Quiz"],
-            label="Select Additional Feature",
-            value="None"
-        )
+
+# ✅ Build Gradio Interface
+with gr.Blocks(theme=gr.themes.Soft(primary_hue="teal")) as demo:
+    gr.Markdown("## 🧠 SMART LESSON PLAN GENERATOR (Bilingual)")
+    gr.Markdown("**Developed by: Najaf Ali**")
 
     with gr.Row():
+        language = gr.Dropdown(["English", "Urdu"], label="Select Language", value="English")
         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 Class/Grade",
-            value="Grade 8"
+            [f"Grade {i}" for i in range(1, 13)], label="Select Grade", value="Grade 9"
         )
-
         subject = gr.Dropdown(
             [
                 "English", "Urdu", "Mathematics", "Science", "Computer",
@@ -109,22 +171,33 @@ with gr.Blocks(theme=gr.themes.Soft(primary_hue="cyan")) as demo:
         )
 
     with gr.Row():
-        topic = gr.Textbox(label="Enter Topic (e.g., Photosynthesis / Noun Types)")
-        duration = gr.Number(label="Lesson Duration (minutes)", value=40)
-        student_level = gr.Dropdown(
-            ["Beginner", "Average", "Advanced"],
-            label="Student Level",
-            value="Average"
-        )
+        topic = gr.Textbox(label="Enter Topic (e.g., Photosynthesis, Fractions, Noun Types)")
+        duration = gr.Number(label="Duration (minutes)", value=40)
+        student_level = gr.Dropdown(["Beginner", "Average", "Advanced"], label="Student Level", value="Average")
 
     generate_btn = gr.Button("🚀 Generate SMART Lesson Plan")
-    output = gr.Textbox(label="Generated Output", lines=25)
+    output_message = gr.HTML(label="Status")
+    file_output = gr.File(label="⬇️ Download Lesson Plan")
+    lesson_textbox = gr.Textbox(label="Generated Lesson Plan Text", lines=20, visible=False)
+
+    # ✅ Flashcards and Quiz buttons
+    gr.Markdown("### 🎯 Generate Additional Materials (Aligned with Lesson Plan)")
+    with gr.Row():
+        flash_btn = gr.Button("🃏 Generate Flashcards")
+        quiz_btn = gr.Button("📝 Generate Quiz")
 
+    flash_output = gr.Textbox(label="Generated Flashcards", lines=10)
+    quiz_output = gr.Textbox(label="Generated Quiz", lines=12)
+
+    # ✅ Button connections
     generate_btn.click(
-        generate_content,
-        inputs=[language, plan_type, feature, grade, subject, topic, duration, student_level],
-        outputs=output
+        generate_lesson_plan,
+        inputs=[language, grade, subject, topic, duration, student_level],
+        outputs=[output_message, file_output, lesson_textbox]
     )
 
+    flash_btn.click(generate_flashcards, inputs=[language, topic, lesson_textbox], outputs=flash_output)
+    quiz_btn.click(generate_quiz, inputs=[language, topic, lesson_textbox], outputs=quiz_output)
+
 # ✅ Launch App
 demo.launch()