Update app.py

app.py

@@ -1,203 +1,203 @@
-import os
 import gradio as gr
-import datetime
-from tempfile import NamedTemporaryFile
-from docx import Document
 from groq import Groq
+from docx import Document
+import datetime
+import os
 
-# ✅ Initialize Groq client
-client = Groq(api_key=os.getenv("GROQ_API_KEY"))
+# ✅ Initialize Groq Client
+client = Groq(api_key="YOUR_GROQ_API_KEY")
 
-URDU_MODEL = "openai/gpt-oss-120b"
-ENGLISH_MODEL = "llama-3.1-8b-instant"
+# ================== Helper Functions ==================
 
+def generate_text(prompt, model_name):
+    response = client.chat.completions.create(
+        messages=[{"role": "user", "content": prompt}],
+        model=model_name,
+    )
+    return response.choices[0].message.content.strip()
 
-# ✅ 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 طرز کا تدریسی منصوبہ تیار کریں۔
-        درج ذیل حصے شامل ہوں:
-        1. معلوماتی جدول (Input Table)
-        2. SMART تعلیمی مقاصد (SLOs)
-        3. "پڑھایا جانے والا مواد" (Content to be delivered): تعریفیں، کلیدی اصطلاحات اور تصورات۔
-        4. گھریلو کام (Home Assignment) — جو SLOs سے مطابقت رکھتا ہو۔
-        5. سرپرست یا پرنسپل کی رائے کے لیے خالی جگہ (Feedback by Mentor/Principal)
+def save_docx(content, filename):
+    doc = Document()
+    doc.add_paragraph(content)
+    filepath = f"/tmp/{filename}.docx"
+    doc.save(filepath)
+    return filepath
 
-        جماعت: {grade}
-        مضمون: {subject}
-        موضوع: {topic}
-        دورانیہ: {duration} منٹ
-        طلباء کی سطح: {student_level}
-        """
-    else:
-        model = ENGLISH_MODEL
-        prompt = f"""
-        You are an expert Pakistani school teacher. Create a SMART-style lesson plan with the following structure:
+def save_md(content, filename):
+    content = content.replace("✨", "").replace("🌟", "").replace("✅", "")
+    filepath = f"/tmp/{filename}.md"
+    with open(filepath, "w", encoding="utf-8") as f:
+        f.write(content)
+    return filepath
 
-        1. Input Table
-        2. SMART Learning Outcomes (SLOs)
-        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)
+# ================== Core Generation Functions ==================
 
+def generate_lesson_plan(language, grade, subject, num_students, date, slo, content):
+    if language == "English":
+        model = "llama-3.1-8b-instant"
+        prompt = f"""
+        Create a SMART Lesson Plan in English for:
         Grade: {grade}
         Subject: {subject}
-        Topic: {topic}
-        Duration: {duration} minutes
-        Student Level: {student_level}
+        Number of Students: {num_students}
+        Date: {date}
+
+        Include the following sections:
+        1. Input Table
+        2. SLOs
+        3. Content to be Delivered
+        4. Home Assignment (aligned with SLOs)
+        5. Feedback by Mentor/Principal
+
+        Use this context:
+        - SLOs: {slo}
+        - Content: {content}
         """
+    else:
+        model = "openai/gpt-oss-120b"
+        prompt = f"""
+        درج ذیل معلومات کی بنیاد پر ایک SMART سبق کا منصوبہ اردو میں تیار کریں۔
+        جماعت: {grade}
+        مضمون: {subject}
+        طلبہ کی تعداد: {num_students}
+        تاریخ: {date}
 
-    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()
+        درج ذیل حصے شامل ہوں:
+        1. ان پٹ جدول  
+        2. سیکھنے کے مقاصد (SLOs)  
+        3. پڑھائی جانے والا مواد  
+        4. گھریلو مشق (جو SLOs سے ہم آہنگ ہو)  
+        5. سرپرست / پرنسپل کی رائے  
+
+        سیاق و سباق:
+        - سیکھنے کے مقاصد: {slo}
+        - مواد: {content}
+        """
 
-    # ✅ Create downloadable file
+    output = generate_text(prompt, model)
+    filename = f"SMART_Lesson_Plan_{language}_{grade}_{subject}"
     if language == "English":
-        doc = Document()
-        doc.add_heading("SMART LESSON PLAN", level=1)
-
-        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])
-
-        doc.add_paragraph("\n" + content)
-        doc.add_paragraph("\nDeveloped by: Najaf Ali")
-
-        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})."
-
+        filepath = save_docx(output, filename)
     else:
-        md_text = f"""
-<div dir="rtl" align="right">
+        filepath = save_md(output, filename)
+    return output, filepath
 
-# 📘 SMART LESSON PLAN
 
-| کلیدی معلومات | تفصیل |
-|----------------|--------|
-| جماعت | {grade} |
-| مضمون | {subject} |
-| عنوان | {topic} |
-| دورانیہ | {duration} منٹ |
-| طلبہ کی سطح | {student_level} |
+def generate_flashcards(language, grade, subject, slo, content):
+    if language == "English":
+        model = "llama-3.1-8b-instant"
+        prompt = f"""
+        Create concise classroom flashcards in English for Grade {grade}, Subject {subject}.
+        Align flashcards with the following SLOs and lesson content:
+        SLOs: {slo}
+        Content: {content}
+        Format as Q&A flashcards.
+        """
+    else:
+        model = "openai/gpt-oss-120b"
+        prompt = f"""
+        درج ذیل سیکھنے کے مقاصد اور مواد کی بنیاد پر جماعت {grade} کے لیے فلیش کارڈز تیار کریں۔
+        مضمون: {subject}
+        ہر کارڈ میں سوال اور مختصر جواب شامل ہو۔
+        سیکھنے کے مقاصد: {slo}
+        مواد: {content}
+        """
 
----
+    output = generate_text(prompt, model)
+    filename = f"Flashcards_{language}_{grade}_{subject}"
+    if language == "English":
+        filepath = save_docx(output, filename)
+    else:
+        filepath = save_md(output, filename)
+    return output, filepath
 
-{content}
 
----
+def generate_quiz(language, grade, subject, slo, content):
+    if language == "English":
+        model = "llama-3.1-8b-instant"
+        prompt = f"""
+        Create a short quiz in English for Grade {grade}, Subject {subject}.
+        Align questions with the following SLOs and lesson content.
+        SLOs: {slo}
+        Content: {content}
+        Include 5 multiple-choice questions with answers.
+        """
+    else:
+        model = "openai/gpt-oss-120b"
+        prompt = f"""
+        جماعت {grade}، مضمون {subject} کے لیے ایک مختصر کوئز اردو میں تیار کریں۔
+        کوئز کے سوالات درج ذیل سیکھنے کے مقاصد اور مواد سے ہم آہنگ ہوں۔
+        سیکھنے کے مقاصد: {slo}
+        مواد: {content}
+        پانچ کثیرالاختیاری سوالات (MCQs) شامل کریں اور جوابات بھی دیں۔
+        """
 
-**مرتب کردہ: نجف علی**
-</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}) کامیابی سے تیار ہو گیا۔"
+    output = generate_text(prompt, model)
+    filename = f"Quiz_{language}_{grade}_{subject}"
+    if language == "English":
+        filepath = save_docx(output, filename)
+    else:
+        filepath = save_md(output, filename)
+    return output, filepath
 
-    return message, file_path, content
 
+# ================== Gradio App ==================
 
-# ✅ Function: Generate Flashcards
-def generate_flashcards(language, topic, slos_text):
-    model = URDU_MODEL if language == "Urdu" else ENGLISH_MODEL
+def main_app(language, feature, grade, subject, num_students, date, slo, content):
+    if feature == "Lesson Plan":
+        result, file = generate_lesson_plan(language, grade, subject, num_students, date, slo, content)
+        return result, file, None, None
 
-    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()
+    elif feature == "Flashcards":
+        result, file = generate_flashcards(language, grade, subject, slo, content)
+        return result, None, file, None
 
+    elif feature == "Quiz":
+        result, file = generate_quiz(language, grade, subject, slo, content)
+        return result, None, None, file
 
-# ✅ Function: Generate Quiz
-def generate_quiz(language, topic, slos_text):
-    model = URDU_MODEL if language == "Urdu" else ENGLISH_MODEL
+    else:
+        return "Please select a valid feature.", None, None, None
 
-    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()
 
+# ================== UI Layout ==================
 
-# ✅ 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.Blocks(title="SMART Lesson Plan Generator") as app:
+    gr.Markdown("<h1 style='text-align:center;'>📘 SMART LESSON PLAN GENERATOR</h1>")
 
     with gr.Row():
         language = gr.Dropdown(["English", "Urdu"], label="Select Language", value="English")
+        feature = gr.Dropdown(["Lesson Plan", "Flashcards", "Quiz"], label="Select Feature", value="Lesson Plan")
+
+    with gr.Row():
         grade = gr.Dropdown(
-            [f"Grade {i}" for i in range(1, 13)], label="Select Grade", value="Grade 9"
+            ["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"
         )
         subject = gr.Dropdown(
-            [
-                "English", "Urdu", "Mathematics", "Science", "Computer",
-                "Islamiyat", "Pakistan Studies", "Physics", "Chemistry",
-                "Biology", "Civics", "History", "Geography"
-            ],
-            label="Select Subject",
-            value="Science"
+            ["English", "Urdu", "Mathematics", "Science", "Islamiyat", "Social Studies", "Computer Science"],
+            label="Select Subject"
         )
 
     with gr.Row():
-        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")
+        num_students = gr.Number(label="Number of Students", value=30)
+        date = gr.Textbox(label="Date (YYYY-MM-DD)", value=str(datetime.date.today()))
 
-    generate_btn = gr.Button("🚀 Generate SMART Lesson Plan")
-    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)
+    slo = gr.Textbox(label="Enter SLOs (Learning Outcomes)", lines=3)
+    content = gr.Textbox(label="Enter Lesson Content Summary", lines=3)
 
-    # ✅ 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")
+    generate_btn = gr.Button("🚀 Generate")
+    output_text = gr.Textbox(label="Generated Output", lines=20)
 
-    flash_output = gr.Textbox(label="Generated Flashcards", lines=10)
-    quiz_output = gr.Textbox(label="Generated Quiz", lines=12)
+    lesson_plan_file = gr.File(label="📄 Download Lesson Plan")
+    flashcard_file = gr.File(label="📘 Download Flashcards")
+    quiz_file = gr.File(label="🧾 Download Quiz")
 
-    # ✅ Button connections
     generate_btn.click(
-        generate_lesson_plan,
-        inputs=[language, grade, subject, topic, duration, student_level],
-        outputs=[output_message, file_output, lesson_textbox]
+        main_app,
+        inputs=[language, feature, grade, subject, num_students, date, slo, content],
+        outputs=[output_text, lesson_plan_file, flashcard_file, quiz_file]
     )
 
-    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()
+app.launch()