Update app.py

app.py

@@ -13,13 +13,13 @@ client = Groq(api_key=os.environ["GROQ_API_KEY"])
 book_data = {
     "title": "",
     "grade": "",
-    "slos": [],
     "toc": [],
     "chapters": {},
     "preface": ""
 }
 
 # Clean formatting
+
 def clean_formatting(text):
     text = re.sub(r"\*\*(.*?)\*\*", r"\1", text)
     text = re.sub(r"\*(.*?)\*", r"\1", text)
@@ -28,20 +28,17 @@ def clean_formatting(text):
     return text.strip()
 
 # Generate TOC and Preface
-def generate_book_intro(title, grade, slos_input):
+
+def generate_book_intro(title, grade):
     book_data["title"] = title.strip()
     book_data["grade"] = grade.strip()
-    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}
+    2. A Table of Contents with 7 chapters. Title each chapter clearly according to a logical structure for beginners.
 
     Do NOT write any chapter content. Only return:
     - Preface
@@ -62,13 +59,12 @@ def generate_book_intro(title, grade, slos_input):
     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 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)
 
     prompt = f"""
     You are an expert textbook writer. Write Chapter {ch_num} for the book titled "{book_data['title']}" for Grade {book_data['grade']}. 
@@ -77,11 +73,11 @@ def generate_chapter(ch_num):
 
     Include the following sections:
     - Introduction
-    - 5 to 7 Student Learning Outcomes (SLOs):\n{slo_text}
+    - Student Learning Outcomes (5 to 7)
     - Detailed content aligned with each SLO
-    - Examples / Case Studies
-    - Chapter Summary
-    - 10 Review Questions or Activities (no answers)
+    - Activities or Case Studies
+    - Assessment (15 MCQs with no answers)
+    - Glossary
 
     Follow APA 7th edition tone. Do NOT use any asterisks, markdown, emojis, or non-printable characters.
     """
@@ -97,6 +93,7 @@ def generate_chapter(ch_num):
     return chapter
 
 # Export full textbook to Word
+
 def export_book():
     doc = Document()
     style = doc.styles['Normal']
@@ -133,17 +130,11 @@ def export_book():
         add_paragraph(line)
     doc.add_page_break()
 
-    # SLOs
-    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 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"]):
+            if any(keyword in line.lower() for keyword in ["introduction", "summary", "examples", "review", "student learning outcomes", "assessment", "glossary", "activities"]):
                 add_heading(line, level=2)
             else:
                 add_paragraph(line)
@@ -155,23 +146,22 @@ def export_book():
 
 # Gradio Interface
 with gr.Blocks() as demo:
-    gr.Markdown("## 📘 AI Textbook Generator – Step-by-Step (APA Style)")
+    gr.Markdown("## 📘 AI Textbook Generator – APA Style | Developed by Najaf Ali Sharqi")
 
     with gr.Row():
         title = gr.Textbox(label="Subject Title")
         grade = gr.Textbox(label="Grade Level")
 
-    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)
 
-    generate_structure.click(fn=generate_book_intro, inputs=[title, grade, slos], outputs=structure_output)
+    generate_structure.click(fn=generate_book_intro, inputs=[title, grade], outputs=structure_output)
 
     gr.Markdown("### 📥 Generate Chapters One by One")
     chapter_output = gr.Textbox(label="Generated Chapter Content", lines=25)
 
     with gr.Row():
-        for i in range(1, 7):
+        for i in range(1, 8):
             gr.Button(f"Generate Chapter {i}").click(fn=generate_chapter, inputs=gr.Number(value=i, visible=False), outputs=chapter_output)
 
     gr.Markdown("### 💾 Download Complete Textbook")
@@ -180,7 +170,6 @@ with gr.Blocks() as demo:
 
     download_btn.click(fn=export_book, outputs=download_file)
 
-    gr.Markdown("Developed by **Najaf Ali Sharqi** | Powered by **Groq + Gradio**")
+    gr.Markdown("Created with ❤️ using Gradio + Groq | All content APA 7th aligned")
 
-# 👇 This makes it work on Colab/Hugging Face
 demo.launch(share=True)