Update app.py

app.py

@@ -1,18 +1,25 @@
 import os
+import re
 from datetime import datetime
 from groq import Groq
 from docx import Document
 from docx.shared import Pt
 import gradio as gr
 
-# Load GROQ API Key from environment
+# Load GROQ API key from Hugging Face secret environment
 client = Groq(api_key=os.environ["GROQ_API_KEY"])
 
 # Memory to store TOC and chapters
 toc_memory = {"toc": ""}
 chapters_memory = {}
 
-# Generate TOC
+# 🔧 Clean markdown, emojis, bullets
+def clean_formatting(text):
+    text = re.sub(r"[*#•●→✅🔹🔸📘📖📑📤📥⬇️🎉>]+", "", text)
+    text = re.sub(r"\s{2,}", " ", text)
+    return text.strip()
+
+# 📑 Generate Table of Contents
 def generate_toc(topic):
     prompt = f"""
 You are an expert curriculum developer. Create a detailed Table of Contents for an absolute beginner-level textbook on the topic: "{topic}".
@@ -32,7 +39,7 @@ Do NOT write the chapter content. Only return the Table of Contents clearly orga
     toc_memory["toc"] = toc
     return toc
 
-# Generate Chapter
+# 📖 Generate Chapter
 def generate_chapter(topic, chapter_number):
     toc = toc_memory["toc"]
     if toc == "":
@@ -62,29 +69,61 @@ Do NOT include system messages like “Here is the content of...” Just return
     chapters_memory[chapter_number] = chapter
     return chapter
 
-# Generate DOCX
+# 📄 Generate Word document
 def download_textbook(topic, grade):
     doc = Document()
-    doc.add_heading(topic, 0)
-    doc.add_paragraph(f"Author: Najaf Ali Sharqi")
-    doc.add_paragraph(f"Grade: {grade}")
-    doc.add_paragraph(f"Date of Publication: {datetime.now().strftime('%B %d, %Y')}")
+
+    # === Set Default Font ===
+    style = doc.styles['Normal']
+    font = style.font
+    font.name = 'Times New Roman'
+    font.size = Pt(12)
+
+    def add_paragraph(text, size=12, bold=False, space_after=12):
+        p = doc.add_paragraph()
+        run = p.add_run(text)
+        run.font.name = 'Times New Roman'
+        run.font.size = Pt(size)
+        run.bold = bold
+        p.paragraph_format.line_spacing = 1.5
+        p.paragraph_format.space_after = Pt(space_after)
+        return p
+
+    def add_heading(text, level=1):
+        size_map = {1: 16, 2: 14}
+        return add_paragraph(text.strip(), size=size_map.get(level, 12), bold=True)
+
+    # === Cover Page ===
+    add_heading(topic.upper(), level=1)
+    add_paragraph(f"Author: Najaf Ali Sharqi")
+    add_paragraph(f"Grade: {grade}")
+    add_paragraph(f"Date of Publication: {datetime.now().strftime('%B %d, %Y')}")
     doc.add_page_break()
 
-    doc.add_heading("Table of Contents", level=1)
-    doc.add_paragraph(toc_memory["toc"])
+    # === Table of Contents ===
+    add_heading("Table of Contents", level=1)
+    add_paragraph(toc_memory["toc"])
     doc.add_page_break()
 
+    # === Chapters ===
     for num in sorted(chapters_memory.keys()):
-        doc.add_heading(f"Chapter {num}", level=1)
-        doc.add_paragraph(chapters_memory[num])
+        add_heading(f"Chapter {num}", level=1)
+        cleaned = clean_formatting(chapters_memory[num])
+        for line in cleaned.split('\n'):
+            if line.strip() == "":
+                continue
+            if any(keyword in line.lower() for keyword in ["introduction", "unit slos", "examples", "glossary", "answer key", "mcqs"]):
+                add_heading(line.strip(), level=2)
+            else:
+                add_paragraph(line.strip())
         doc.add_page_break()
 
+    # Save file
     file_path = "/tmp/Generated_Textbook.docx"
     doc.save(file_path)
     return file_path
 
-# Gradio UI
+# 🌐 Gradio Interface
 with gr.Blocks() as demo:
     gr.Markdown("## 📘 AI Textbook Generator App")
     gr.Markdown("""
@@ -93,7 +132,7 @@ Welcome to the AI Textbook Generator developed by **Najaf Ali Sharqi**.
 ### 👉 Instructions:
 1. Enter the **textbook topic** and **grade level**
 2. Click **Generate Table of Contents**
-3. Tap a chapter button to generate its content
+3. Tap a chapter button to generate that chapter
 4. After generating all chapters, click **Download Word File**
 """)
 
@@ -108,12 +147,15 @@ Welcome to the AI Textbook Generator developed by **Najaf Ali Sharqi**.
 
     gr.Markdown("### 📥 Generate Chapter Content")
 
+    chapter_display = gr.Textbox(label="📖 Chapter Content", lines=30)
+
     with gr.Row():
         for i in range(1, 8):
             btn = gr.Button(f"Chapter {i}")
-            btn.click(fn=generate_chapter, inputs=[topic_input, gr.Textbox(value=str(i), visible=False)], outputs=gr.Textbox(label="📖 Chapter Content", lines=30))
+            btn.click(fn=generate_chapter, inputs=[topic_input, gr.Textbox(value=str(i), visible=False)], outputs=chapter_display)
 
     gr.Markdown("### 📤 Export Full Textbook")
+
     download_btn = gr.Button("⬇️ Download Word File")
     download_file = gr.File()