Update app.py

app.py

@@ -6,16 +6,18 @@ from docx import Document
 from docx.shared import Pt
 import gradio as gr
 
-# Load GROQ API key from Hugging Face secret environment
+# 🔐 Load Groq API Key from Hugging Face Secrets
 client = Groq(api_key=os.environ["GROQ_API_KEY"])
 
-# Memory to store TOC and chapters
+# Memory to store TOC and Chapters
 toc_memory = {"toc": ""}
 chapters_memory = {}
 
-# 🔧 Clean markdown, emojis, bullets
+# ✅ Clean Markdown, Symbols, Emojis
 def clean_formatting(text):
-    text = re.sub(r"[*#•●→✅🔹🔸📘📖📑📤📥⬇️🎉>]+", "", text)
+    text = re.sub(r"\*\*(.*?)\*\*", r"\1", text)  # Remove **bold**
+    text = re.sub(r"\*(.*?)\*", r"\1", text)      # Remove *italic*
+    text = re.sub(r"[#•●→✅🔹🔸📘📖📑📤📥⬇️🎉>]", "", text)
     text = re.sub(r"\s{2,}", " ", text)
     return text.strip()
 
@@ -39,12 +41,8 @@ Do NOT write the chapter content. Only return the Table of Contents clearly orga
     toc_memory["toc"] = toc
     return toc
 
-# 📖 Generate Chapter
+# 📘 Generate One Chapter
 def generate_chapter(topic, chapter_number):
-    toc = toc_memory["toc"]
-    if toc == "":
-        return "⚠️ Please generate the Table of Contents first."
-
     prompt = f"""
 Using the following Table of Contents for a beginner-level textbook on **{topic}**, generate the complete content of **Chapter {chapter_number}**.
 
@@ -65,15 +63,27 @@ Do NOT include system messages like “Here is the content of...” Just return
         messages=[{"role": "user", "content": prompt}],
         temperature=0.7
     )
-    chapter = response.choices[0].message.content.strip()
-    chapters_memory[chapter_number] = chapter
-    return chapter
+    content = response.choices[0].message.content.strip()
+    chapters_memory[chapter_number] = content
+    return content
 
-# 📄 Generate Word document
+# 🔁 Generate All Chapters (1 to 7)
+def generate_all_chapters(topic):
+    toc = toc_memory["toc"]
+    if toc == "":
+        return "⚠️ Please generate the Table of Contents first."
+    
+    summary = ""
+    for i in range(1, 8):
+        content = generate_chapter(topic, str(i))
+        summary += f"\n\n--- Chapter {i} Generated ---\n\n"
+    return summary + "\n\n✅ All chapters generated. Now download the book."
+
+# 📄 Download Final Book
 def download_textbook(topic, grade):
     doc = Document()
 
-    # === Set Default Font ===
+    # Font settings
     style = doc.styles['Normal']
     font = style.font
     font.name = 'Times New Roman'
@@ -93,19 +103,19 @@ def download_textbook(topic, grade):
         size_map = {1: 16, 2: 14}
         return add_paragraph(text.strip(), size=size_map.get(level, 12), bold=True)
 
-    # === Cover Page ===
+    # 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()
 
-    # === Table of Contents ===
+    # Table of Contents
     add_heading("Table of Contents", level=1)
     add_paragraph(toc_memory["toc"])
     doc.add_page_break()
 
-    # === Chapters ===
+    # Chapters
     for num in sorted(chapters_memory.keys()):
         add_heading(f"Chapter {num}", level=1)
         cleaned = clean_formatting(chapters_memory[num])
@@ -118,50 +128,39 @@ def download_textbook(topic, grade):
                 add_paragraph(line.strip())
         doc.add_page_break()
 
-    # Save file
     file_path = "/tmp/Generated_Textbook.docx"
     doc.save(file_path)
     return file_path
-
-# 🌐 Gradio Interface
 with gr.Blocks() as demo:
     gr.Markdown("## 📘 AI Textbook Generator App")
     gr.Markdown("""
-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 that chapter
-4. After generating all chapters, click **Download Word File**
+Developed by **Najaf Ali Sharqi**  
+Generate a complete beginner-level textbook with SLOs, examples, MCQs, glossary, and export it in MS Word.
+
+### 🛠️ Steps:
+1. Enter your textbook **topic** and **grade level**
+2. Click **Generate TOC**
+3. Then click **Generate All Chapters**
+4. Finally, click **Download Word File**
 """)
 
     with gr.Row():
-        topic_input = gr.Textbox(label="Textbook Topic", placeholder="e.g., Basics of AI")
-        grade_input = gr.Textbox(label="Grade Level", placeholder="e.g., 6")
-
-    toc_btn = gr.Button("🔹 Generate Table of Contents")
-    toc_output = gr.Textbox(label="📑 Table of Contents", lines=10)
+        topic_input = gr.Textbox(label="📚 Textbook Topic", placeholder="e.g., Basics of AI")
+        grade_input = gr.Textbox(label="🎓 Grade Level", placeholder="e.g., Grade 6")
 
+    toc_btn = gr.Button("📑 Generate Table of Contents")
+    toc_output = gr.Textbox(label="📘 Table of Contents", lines=10)
     toc_btn.click(fn=generate_toc, inputs=topic_input, outputs=toc_output)
 
-    gr.Markdown("### 📥 Generate Chapter Content")
+    generate_all_btn = gr.Button("🧠 Generate All Chapters (1–7)")
+    status = gr.Textbox(label="📖 Generation Log", lines=10)
+    generate_all_btn.click(fn=generate_all_chapters, inputs=topic_input, outputs=status)
 
-    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=chapter_display)
-
-    gr.Markdown("### 📤 Export Full Textbook")
-
-    download_btn = gr.Button("⬇️ Download Word File")
+    download_btn = gr.Button("⬇️ Download Textbook (.docx)")
     download_file = gr.File()
-
     download_btn.click(fn=download_textbook, inputs=[topic_input, grade_input], outputs=download_file)
 
     gr.Markdown("---")
-    gr.Markdown("App developed by **Najaf Ali Sharqi** | Powered by **Groq + Gradio**")
+    gr.Markdown("Made with ❤️ by **Najaf Ali Sharqi** | Powered by **Groq + Gradio**")
 
 demo.launch()