Update app.py

app.py

@@ -16,167 +16,174 @@ book_data = {
     "slos": [],
     "toc": [],
     "chapters": {},
+    "preface": ""
 }
 
-# ✂️ Clean text for Word
+# Clean formatting
+
 def clean_formatting(text):
     text = re.sub(r"\*\*(.*?)\*\*", r"\1", text)
     text = re.sub(r"\*(.*?)\*", r"\1", text)
-    text = re.sub(r"[#•●→✅🔹🔸📘📖📑📤📥⬇️🎉>]", "", text)
+    text = re.sub(r"[#•●→✅🔹🔸📘📖📑📤📥⬇️🎉>"]", "", text)
     text = re.sub(r"\s{2,}", " ", text)
     return text.strip()
 
-# 📄 Generate Title Page + Preface + TOC + SLOs
-def generate_intro(title, grade, slos_input):
+# Generate TOC and Preface
+
+def generate_book_intro(title, grade, slos_input):
     book_data["title"] = title.strip()
     book_data["grade"] = grade.strip()
-    slos = [line.strip() for line in slos_input.split('\n') if line.strip()]
-    book_data["slos"] = slos
-
-    toc_prompt = f"""You are a textbook writer. Given the following title: "{title}" and list of SLOs, generate:
-1. A brief Preface (purpose, audience, usage)
-2. A suggested Table of Contents with 5–7 chapters titled according to the SLOs
-3. Return output in this order only: Preface -> TOC
-SLOs:\n{slos_input}
-"""
+    book_data["slos"] = [line.strip() for line in slos_input.split('\n') if line.strip()]
+
+    prompt = f"""
+    You are a professional educational author. 
+    Write the following sections for a textbook titled: "{title}" for Grade {grade}:
+
+    1. A formal preface aligned with APA 7th Edition style.
+    2. A Table of Contents with 6 chapters. Title each chapter clearly according to common educational structure or inferred from the following SLOs:
+
+    SLOs:
+    {slos_input}
+
+    Do NOT write any chapter content. Only return:
+    - Preface
+    - Table of Contents
+    """
+
     response = client.chat.completions.create(
         model="llama3-70b-8192",
-        messages=[{"role": "user", "content": toc_prompt}],
+        messages=[{"role": "user", "content": prompt}],
         temperature=0.7
     )
-    result = response.choices[0].message.content.strip()
-    sections = result.split("Table of Contents")
-    preface = sections[0].strip()
-    toc = sections[1].strip() if len(sections) > 1 else ""
-    book_data["toc"] = toc.split('\n')
-    return f"📘 Preface:\n\n{preface}\n\n📑 Table of Contents:\n\n{toc}\n\n📋 SLOs:\n" + "\n".join([f"{i+1}. {s}" for i, s in enumerate(slos)])
-
-# 📚 Generate Chapter
+
+    output = response.choices[0].message.content.strip()
+    preface, toc = output.split("Table of Contents", 1)
+    book_data["preface"] = clean_formatting(preface)
+    book_data["toc"] = [clean_formatting(line) for line in toc.strip().split('\n') if line.strip()]
+
+    return f"Preface:\n\n{book_data['preface']}\n\nTable of Contents:\n\n" + "\n".join(book_data["toc"])
+
+# Generate chapter by number
+
 def generate_chapter(ch_num):
     if ch_num in book_data["chapters"]:
-        return f"Chapter {ch_num} already generated."
+        return book_data["chapters"][ch_num]
+
+    ch_title = next((line for line in book_data["toc"] if line.startswith(str(ch_num))), f"Chapter {ch_num}")
+    slos = book_data["slos"][(ch_num - 1)*5: ch_num*5]  # 5 SLOs per chapter
+    slo_text = "\n".join(f"- {slo}" for slo in slos)
 
-    ch_title_line = next((line for line in book_data["toc"] if line.strip().lower().startswith(f"{ch_num}")), f"Chapter {ch_num}")
-    slo = book_data["slos"][ch_num - 1] if ch_num - 1 < len(book_data["slos"]) else "No SLO assigned."
+    prompt = f"""
+    You are an expert textbook writer. Write Chapter {ch_num} for the book titled "{book_data['title']}" for Grade {book_data['grade']}. 
 
-    prompt = f"""You are an educational textbook writer.
+    Chapter Title: {ch_title}
 
-Write Chapter {ch_num} titled: "{ch_title_line}" for the subject "{book_data['title']}" for grade {book_data['grade']}, based on the SLO: "{slo}".
+    Include the following sections:
+    - Introduction
+    - 5 to 7 Student Learning Outcomes (SLOs):\n{slo_text}
+    - Detailed content aligned with each SLO
+    - Examples / Case Studies
+    - Chapter Summary
+    - 10 Review Questions or Activities (no answers)
+
+    Follow APA 7th edition tone. Do NOT use any asterisks, markdown, emojis, or non-printable characters.
+    """
 
-Structure the chapter with:
-1. Introduction
-2. Main Concepts and Explanations
-3. Examples or Case Studies
-4. Chapter Summary
-5. Review Questions / Activities (10)
-Maintain academic tone, no markdown, no asterisks. Format clearly as plain text. No emojis.
-"""
     response = client.chat.completions.create(
         model="llama3-70b-8192",
         messages=[{"role": "user", "content": prompt}],
         temperature=0.7
     )
+
     chapter = clean_formatting(response.choices[0].message.content.strip())
     book_data["chapters"][ch_num] = chapter
     return chapter
 
-# 📤 Export Word document
+# Export full textbook to Word
+
 def export_book():
     doc = Document()
     style = doc.styles['Normal']
     style.font.name = 'Times New Roman'
     style.font.size = Pt(12)
 
-    def add_paragraph(text, size=12, bold=False, space_after=12):
+    def add_paragraph(text, size=12, bold=False):
         p = doc.add_paragraph()
         run = p.add_run(text.strip())
         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 = 16 if level == 1 else 14
         add_paragraph(text.strip(), size=size, bold=True)
 
     # Title Page
-    add_heading(book_data['title'].upper(), level=1)
+    add_heading(book_data['title'], level=1)
     add_paragraph(f"Subject: {book_data['title']}")
     add_paragraph(f"Grade: {book_data['grade']}")
     add_paragraph("Author: AI-Generated by Groq-powered App")
-    add_paragraph(f"Date of Publication: {datetime.now().strftime('%B %d, %Y')}")
+    add_paragraph(f"Date: {datetime.now().strftime('%B %d, %Y')}")
     doc.add_page_break()
 
-    # Preface + TOC
+    # Preface
     add_heading("Preface", level=1)
-    add_paragraph("This AI-generated textbook is designed for students and teachers. It aims to provide clear, structured, and curriculum-aligned content based on defined SLOs.")
+    add_paragraph(book_data['preface'])
     doc.add_page_break()
 
+    # TOC
     add_heading("Table of Contents", level=1)
-    for line in book_data["toc"]:
-        add_paragraph(clean_formatting(line))
+    for line in book_data['toc']:
+        add_paragraph(line)
     doc.add_page_break()
 
     # SLOs
-    add_heading("List of Student Learning Outcomes (SLOs)", level=1)
-    for i, slo in enumerate(book_data["slos"], start=1):
+    add_heading("Student Learning Outcomes (SLOs)", level=1)
+    for i, slo in enumerate(book_data['slos'], 1):
         add_paragraph(f"{i}. {slo}")
     doc.add_page_break()
 
     # Chapters
-    for ch_num in sorted(book_data["chapters"].keys()):
-        add_heading(f"Chapter {ch_num}", level=1)
-        for line in book_data["chapters"][ch_num].split('\n'):
-            if any(x in line.lower() for x in ["introduction", "summary", "example", "review"]):
-                add_heading(line.strip(), level=2)
+    for num in sorted(book_data['chapters'].keys()):
+        add_heading(f"{book_data['toc'][num-1]}", level=1)
+        for line in book_data['chapters'][num].split('\n'):
+            if any(keyword in line.lower() for keyword in ["introduction", "summary", "examples", "review", "student learning outcomes"]):
+                add_heading(line, level=2)
             else:
                 add_paragraph(line)
         doc.add_page_break()
 
-    file_path = "/tmp/AI_Generated_Textbook.docx"
+    file_path = "/tmp/Textbook_AI_Generated.docx"
     doc.save(file_path)
     return file_path
 
-# 🎛️ Gradio UI
+# Gradio Interface
 with gr.Blocks() as demo:
-    gr.Markdown("## 📘 AI Textbook Generator – Powered by Groq")
-    gr.Markdown("""
-Welcome! This app generates a full textbook from your provided **Subject Title** and **SLOs**.
-
-### 🔧 Instructions:
-1. Enter your **Subject Title**, **Grade**, and **SLOs** (one per line)
-2. Click **Generate Book Structure** (Title Page, TOC, SLOs)
-3. Click each **Chapter button** one by one to generate chapters
-4. Finally, click **Download** to export the complete textbook as a Word file
-""")
+    gr.Markdown("## 📘 AI Textbook Generator – Step-by-Step (APA Style)")
 
     with gr.Row():
-        title_input = gr.Textbox(label="Subject Title")
-        grade_input = gr.Textbox(label="Grade Level")
-    slos_input = gr.Textbox(label="Paste SLOs (one per line)", lines=10)
+        title = gr.Textbox(label="Subject Title")
+        grade = gr.Textbox(label="Grade Level")
 
-    structure_btn = gr.Button("🧠 Generate Book Structure")
-    structure_output = gr.Textbox(label="📄 Book Structure Preview", lines=20)
+    slos = gr.Textbox(label="Paste SLOs (One per Line)", lines=10)
+    generate_structure = gr.Button("🧠 Generate TOC + Preface")
+    structure_output = gr.Textbox(label="Generated Preface and TOC", lines=15)
 
-    structure_btn.click(fn=generate_intro, inputs=[title_input, grade_input, slos_input], outputs=structure_output)
+    generate_structure.click(fn=generate_book_intro, inputs=[title, grade, slos], outputs=structure_output)
 
     gr.Markdown("### 📥 Generate Chapters One by One")
-
-    chapter_output = gr.Textbox(label="📖 Chapter Preview", lines=25)
+    chapter_output = gr.Textbox(label="Generated Chapter Content", lines=25)
 
     with gr.Row():
-        for i in range(1, 8):
-            gr.Button(f"Chapter {i}").click(fn=generate_chapter, inputs=gr.Number(value=i, visible=False), outputs=chapter_output)
-
-    gr.Markdown("### 💾 Export Complete Textbook")
+        for i in range(1, 7):
+            gr.Button(f"Generate Chapter {i}").click(fn=generate_chapter, inputs=gr.Number(value=i, visible=False), outputs=chapter_output)
 
-    export_btn = gr.Button("⬇️ Download as Word Document")
+    gr.Markdown("### 💾 Download Complete Textbook")
+    download_btn = gr.Button("Download MS Word Textbook")
     download_file = gr.File()
 
-    export_btn.click(fn=export_book, outputs=download_file)
+    download_btn.click(fn=export_book, outputs=download_file)
 
-    gr.Markdown("---")
     gr.Markdown("Developed by **Najaf Ali Sharqi** | Powered by **Groq + Gradio**")
 
 demo.launch()