Update app.py

app.py

@@ -3,6 +3,10 @@ import os
 from datetime import datetime
 from google import genai
 from google.genai import types
+from docx import Document
+from docx.shared import Pt
+from docx.enum.text import WD_ALIGN_PARAGRAPH
+import re
 
 # ✅ Initialize Gemini client
 client = genai.Client(api_key=os.getenv("GOOGLE_API_KEY"))
@@ -51,10 +55,15 @@ Inputs (to be considered while generating the plan):
 Generate the final lesson plan in the requested language, following the exact format above.
 """
 
-# ✅ Function to generate SMART Lesson Plan using Gemini 2.5 Flash
+# ✅ Function: clean text (remove unwanted symbols)
+def clean_text(text):
+    text = re.sub(r"[*_•\-–—]", "", text)
+    text = text.replace("❌", "").replace("✅", "").strip()
+    return text
+
+# ✅ Generate SMART Lesson Plan using Gemini
 def generate_smart_lesson(language, grade, subject, topic, num_students, duration, teacher_name, date):
     try:
-        # Combine user inputs into a structured query
         user_input = f"""
 Language: {language}
 Grade Level: {grade}
@@ -68,7 +77,6 @@ Date: {date}
 
         full_prompt = BASE_PROMPT + "\n\n" + user_input
 
-        # Generate content using Gemini model
         response = client.models.generate_content(
             model="gemini-2.5-flash",
             contents=full_prompt,
@@ -78,15 +86,44 @@ Date: {date}
             )
         )
 
-        return response.text.strip()
+        lesson_text = clean_text(response.text.strip())
+        word_file = create_word_doc(language, lesson_text, topic)
+        return lesson_text, word_file
 
     except Exception as e:
-        return f"❌ Error generating lesson plan: {e}"
+        return f"❌ Error generating lesson plan: {e}", None
+
+
+# ✅ Create Word / Markdown file
+def create_word_doc(language, lesson_text, topic):
+    file_path = f"SMART_Lesson_{topic.replace(' ', '_')}.docx" if language == "English" else f"SMART_Lesson_{topic.replace(' ', '_')}.md"
+
+    # For English version: Create Word document
+    if language == "English":
+        doc = Document()
+        style = doc.styles['Normal']
+        style.font.name = 'Times New Roman'
+        style.font.size = Pt(12)
+
+        for line in lesson_text.split("\n"):
+            p = doc.add_paragraph(line.strip())
+            p.alignment = WD_ALIGN_PARAGRAPH.LEFT
+
+        doc.save(file_path)
+        return file_path
+
+    # For Urdu version: Markdown file (Right-to-Left)
+    else:
+        urdu_text = f"<div dir='rtl' style='font-family:Jameel Noori Nastaleeq; text-align:right;'>\n{lesson_text}\n</div>"
+        with open(file_path, "w", encoding="utf-8") as f:
+            f.write(urdu_text)
+        return file_path
+
 
-# ✅ Gradio UI
+# ✅ Gradio Interface
 with gr.Blocks(title="SMART Lesson Plan Generator (Gemini 2.5 Flash)") as app:
     gr.Markdown("## 🧠 SMART LESSON PLAN GENERATOR (Gemini 2.5 Flash)")
-    gr.Markdown("Generate complete Urdu or English lesson plans aligned with Pakistan’s National Curriculum.")
+    gr.Markdown("Generate complete Urdu or English lesson plans aligned with Pakistan’s National Curriculum. Download them as Word or Markdown files.")
 
     with gr.Row():
         language = gr.Radio(["English", "Urdu"], label="Select Language", value="English")
@@ -101,14 +138,14 @@ with gr.Blocks(title="SMART Lesson Plan Generator (Gemini 2.5 Flash)") as app:
         date = gr.Textbox(label="Date", value=datetime.now().strftime("%Y-%m-%d"))
 
     generate_btn = gr.Button("🎓 Generate SMART Lesson Plan")
-    output_box = gr.Textbox(label="Generated Lesson Plan", lines=30)
+    output_text = gr.Textbox(label="Generated Lesson Plan", lines=25)
+    download_file = gr.File(label="Download Lesson Plan")
 
     generate_btn.click(
         fn=generate_smart_lesson,
         inputs=[language, grade, subject, topic, num_students, duration, teacher_name, date],
-        outputs=output_box
+        outputs=[output_text, download_file]
     )
 
-# ✅ Launch app
 if __name__ == "__main__":
     app.launch()