Update app.py

app.py

@@ -1,53 +1,54 @@
+import os
 import gradio as gr
-import datetime
 from groq import Groq
 
-# Initialize Groq Client
-client = Groq(api_key="YOUR_GROQ_API_KEY")
+# ✅ Load Groq API Key
+client = Groq(api_key=os.getenv("GROQ_API_KEY"))
 
-# Models
+# ✅ Define models
 URDU_MODEL = "openai/gpt-oss-120b"
 ENGLISH_MODEL = "llama-3.1-8b-instant"
 
-# Generation function
+
+# ✅ Core Function
 def generate_content(language, plan_type, feature, grade, subject, topic, duration, student_level):
     if language == "Urdu":
         model = URDU_MODEL
         prompt = f"""
-        آپ ایک ماہر استاد ہیں۔ براہ کرم نیچے دی گئی معلومات کی بنیاد پر ایک تفصیلی SMART طرز کا {plan_type} پلان تیار کریں۔
-        اگر {feature} منتخب کیا گیا ہے تو اس کے مطابق مواد بھی شامل کریں۔
-
-        درجہ: {grade}
+        آپ ایک ماہر استاد ہیں۔ براہ کرم نیچے دی گئی معلومات کی بنیاد پر SMART طرز کا {plan_type} تیار کریں۔
+        درج ذیل حصے شامل ہوں:
+        1. معلوماتی جدول (Input Table)
+        2. SMART سیکھنے کے نتائج (SLOs)
+        3. "پڑھایا جانے والا مواد" (Content to be delivered) — تعریفیں، کلیدی تصورات اور بنیادی اصطلاحات کے ساتھ۔
+        4. گھریلو کام (Home Assignment) — SLOs کے مطابق۔
+        5. سرپرست یا پرنسپل کے تاثرات کے لیے جگہ (Feedback by Mentor/Principal)
+
+        جماعت: {grade}
         مضمون: {subject}
         موضوع: {topic}
         دورانیہ: {duration} منٹ
         طلبہ کی سطح: {student_level}
-
-        مندرجہ ذیل چیزیں شامل کریں:
-        1. مقاصدِ سبق (SMART طرز میں)
-        2. سرگرمیاں
-        3. تدریسی حکمتِ عملی
-        4. جائزہ/Assessment
-        5. اگر منتخب کیا گیا ہو تو {feature} کے نمونے
+        اگر {feature} منتخب کیا گیا ہے تو اس کے مطابق مواد بھی تیار کریں۔
         """
     else:
         model = ENGLISH_MODEL
         prompt = f"""
-        You are an expert teacher. Create a detailed SMART-style {plan_type} plan.
-        If {feature} is selected, include relevant examples or content.
+        You are an experienced Pakistani school teacher. 
+        Please create a SMART-style {plan_type} lesson plan based on the given details.
+
+        Include only the following sections:
+        1. Input Table (with basic information)
+        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).
 
         Grade: {grade}
         Subject: {subject}
         Topic: {topic}
         Duration: {duration} minutes
         Student Level: {student_level}
-
-        Include the following sections:
-        1. SMART Objectives
-        2. Learning Activities
-        3. Teaching Strategies
-        4. Assessment
-        5. Include {feature} if applicable
+        Also, if {feature} is selected, generate relevant {feature} content aligned with SLOs.
         """
 
     try:
@@ -62,18 +63,24 @@ def generate_content(language, plan_type, feature, grade, subject, topic, durati
         return f"❌ Error: {str(e)}"
 
 
-# UI Layout
-with gr.Blocks(theme=gr.themes.Soft(primary_hue="teal")) as demo:
+# ✅ 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**")
 
     with gr.Row():
-        language = gr.Dropdown(["English", "Urdu"], label="Select Language", value="English")
+        language = gr.Dropdown(
+            ["English", "Urdu"],
+            label="Select Language",
+            value="English"
+        )
+
         plan_type = gr.Dropdown(
-            ["Daily Lesson Plan", "Weekly Lesson Plan", "Unit Plan", "SMART Lesson Plan"],
+            ["SMART Lesson Plan", "Daily Lesson Plan", "Weekly Lesson Plan", "Unit Plan"],
             label="Select Lesson Plan Type",
             value="SMART Lesson Plan"
         )
+
         feature = gr.Dropdown(
             ["None", "Flashcards", "Quiz"],
             label="Select Additional Feature",
@@ -81,15 +88,36 @@ with gr.Blocks(theme=gr.themes.Soft(primary_hue="teal")) as demo:
         )
 
     with gr.Row():
-        grade = gr.Textbox(label="Grade / Class (e.g., Grade 8)")
-        subject = gr.Textbox(label="Subject (e.g., Science, Urdu)")
+        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"
+        )
+
+        subject = gr.Dropdown(
+            [
+                "English", "Urdu", "Mathematics", "Science", "Computer",
+                "Islamiyat", "Pakistan Studies", "Physics", "Chemistry",
+                "Biology", "Civics", "History", "Geography"
+            ],
+            label="Select Subject",
+            value="Science"
+        )
 
     with gr.Row():
-        topic = gr.Textbox(label="Topic (e.g., Photosynthesis)")
-        duration = gr.Number(label="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 / Noun Types)")
+        duration = gr.Number(label="Lesson Duration (minutes)", value=40)
+        student_level = gr.Dropdown(
+            ["Beginner", "Average", "Advanced"],
+            label="Student Level",
+            value="Average"
+        )
 
-    generate_btn = gr.Button("🚀 Generate Now")
+    generate_btn = gr.Button("🚀 Generate SMART Lesson Plan")
     output = gr.Textbox(label="Generated Output", lines=25)
 
     generate_btn.click(
@@ -98,5 +126,5 @@ with gr.Blocks(theme=gr.themes.Soft(primary_hue="teal")) as demo:
         outputs=output
     )
 
-# Launch App
+# ✅ Launch App
 demo.launch()