Update app.py

app.py

@@ -2,13 +2,15 @@ import os
 import gradio as gr
 import datetime
 from groq import Groq
+from docx import Document
+import tempfile
 
 # ✅ Initialize Groq API
 GROQ_API_KEY = os.getenv("GROQ_API_KEY")
 client = Groq(api_key=GROQ_API_KEY)
 
 # ✅ Function to generate lesson plan
-def generate_lesson_plan(language, grade, subject, topic, students, duration, date, teacher):
+def generate_lesson_plan(language, grade, subject, topic, students, duration, date, teacher, theme):
     if language == "Urdu":
         model = "openai/gpt-oss-120b"
         prompt = f"""
@@ -41,7 +43,7 @@ def generate_lesson_plan(language, grade, subject, topic, students, duration, da
     else:
         model = "llama-3.1-8b-instant"
         prompt = f"""
-You are a professional teacher. Based on the following details, generate a complete **Lesson Plan in English** using the **BOPPPS Model**.
+You are a professional teacher. Based on the following details, generate a complete Lesson Plan in English using the BOPPPS Model.
 
 Class: {grade}
 Subject: {subject}
@@ -61,6 +63,7 @@ Lesson Plan Structure:
 7. Home Assignment aligned with SLOs (reasonable length, logical order, creative, interesting, suitable for students’ level, and completable in 1–2 hours)
 """
 
+    # ✅ Generate via Groq
     completion = client.chat.completions.create(
         model=model,
         messages=[
@@ -70,12 +73,35 @@ Lesson Plan Structure:
     )
 
     response = completion.choices[0].message.content
-
-    # Add footer
     footer = "\n\n---\n**Developed by Najaf Ali Sharqi**"
-    return response + footer
+    final_text = response + footer
+
+    # ✅ Save as Markdown for Urdu / DOCX for English
+    if language == "Urdu":
+        file_ext = ".md"
+        temp_file = tempfile.NamedTemporaryFile(delete=False, suffix=file_ext)
+        with open(temp_file.name, "w", encoding="utf-8") as f:
+            f.write(final_text)
+    else:
+        file_ext = ".docx"
+        temp_file = tempfile.NamedTemporaryFile(delete=False, suffix=file_ext)
+        doc = Document()
+        doc.add_heading("Lesson Plan", level=1)
+        doc.add_paragraph(final_text)
+        doc.add_paragraph("\nDeveloped by Najaf Ali Sharqi")
+        doc.save(temp_file.name)
+
+    # ✅ Adjust background color theme
+    if theme == "Dark":
+        theme_css = "background-color:#111;color:#fff;padding:10px;border-radius:10px;"
+    else:
+        theme_css = "background-color:#fdfdfd;color:#000;padding:10px;border-radius:10px;"
+
+    display_html = f"<div style='{theme_css}'>{final_text.replace('\n', '<br>')}</div>"
 
-# ✅ Gradio Interface
+    return display_html, temp_file.name
+
+# ✅ Dropdown options
 grades = [f"Grade {i}" for i in range(1, 17)]
 subjects = [
     "Urdu", "English", "Arabic", "Science", "Mathematics",
@@ -83,33 +109,36 @@ subjects = [
     "Civics", "Islamiat"
 ]
 
-with gr.Blocks(title="Lesson Plan Generator") as app:
-    gr.Markdown("## 📘 AI Lesson Plan Generator (English / Urdu)\nDeveloped by **Najaf Ali Sharqi**")
+# ✅ Build the Gradio app
+with gr.Blocks(title="AI Lesson Plan Generator") as app:
+    gr.Markdown("## 🧠 **AI Lesson Plan Generator (English / Urdu)**\nDeveloped by **Najaf Ali Sharqi**")
 
     with gr.Row():
         language = gr.Dropdown(["English", "Urdu"], label="Language / زبان", value="English")
         grade = gr.Dropdown(grades, label="Class / جماعت")
         subject = gr.Dropdown(subjects, label="Subject / مضمون")
+        theme = gr.Radio(["Light", "Dark"], label="Screen Theme / سکرین موڈ", value="Light")
 
     with gr.Row():
         topic = gr.Textbox(label="Topic / عنوان", placeholder="Enter topic")
         students = gr.Slider(1, 200, step=1, label="No. of Students / طلبہ کی تعداد")
         duration = gr.Slider(30, 60, step=5, label="Time Duration (minutes) / دورانیہ (منٹ)")
 
-    # ✅ Replaced gr.Calendar() with gr.Textbox (date input)
     with gr.Row():
         date = gr.Textbox(label="Date / تاریخ", placeholder="YYYY-MM-DD", value=str(datetime.date.today()))
         teacher = gr.Textbox(label="Teacher's Name / استاد کا نام")
 
-    generate_btn = gr.Button("Generate Lesson Plan / سبق کا منصوبہ تیار کریں")
+    generate_btn = gr.Button("🚀 Generate Lesson Plan / سبق کا منصوبہ تیار کریں")
 
-    output = gr.Markdown(label="Generated Lesson Plan")
+    output_display = gr.HTML(label="Generated Lesson Plan")
+    file_output = gr.File(label="Download Lesson Plan File")
 
     generate_btn.click(
         fn=generate_lesson_plan,
-        inputs=[language, grade, subject, topic, students, duration, date, teacher],
-        outputs=output,
+        inputs=[language, grade, subject, topic, students, duration, date, teacher, theme],
+        outputs=[output_display, file_output],
     )
 
-# ✅ Launch
+# ✅ Launch the app
 app.launch()
+