Update app.py

app.py

@@ -78,12 +78,8 @@ def generate_pdf(images, explanations, title="AI-Generated Story Scenes"):
     with tempfile.NamedTemporaryFile(delete=False, suffix=".pdf") as tmp:
         doc = SimpleDocTemplate(tmp.name, pagesize=letter)
         styles = getSampleStyleSheet()
-        
-        # Add the title at the top
-        story = [
-            Paragraph(title, styles["Title"]),
-            Spacer(1, 24)
-        ]
+        story = [Paragraph(title, styles["Title"]), Spacer(1, 12)]
+
 
         for i, img in enumerate(images):
             if img:
@@ -137,7 +133,7 @@ def finalize_story(images, explanations, title):
     if not images or not explanations:
         return None, None
 
-    pdf = generate_pdf(images, explanations, title=title)
+    pdf = generate_pdf(images, explanations,title)
 
     with tempfile.NamedTemporaryFile(delete=False, suffix=".txt", mode="w", encoding="utf-8") as txt:
         for i, exp in enumerate(explanations):
@@ -146,7 +142,6 @@ def finalize_story(images, explanations, title):
 
     return txt_path, pdf
 
-
 def ai_write_scene(theme, total_scenes, scene_summaries):
     try:
         scene_index = len(scene_summaries) + 1
@@ -321,6 +316,7 @@ with gr.Blocks(
     gr.Markdown("Describe your story one scene at a time, with AI-generated images, summaries, and explanations.")
 
     with gr.Accordion("🧩 Story Setup", open=True):
+        title = gr.Textbox(label="📖 Story Title", placeholder="e.g. The Enchanted Forest")
         scene_total = gr.Number(label="🔢 Number of Scenes", precision=0, value=3, elem_id="scene-count")
         theme = gr.Textbox(label="🌍 Global Theme", placeholder="e.g. A magical forest full of glowing creatures", elem_id="theme-input")
 
@@ -413,9 +409,10 @@ with gr.Blocks(
 
     done_btn.click(
         fn=finalize_story,
-        inputs=[scene_images, scene_explanations, theme],
+        inputs=[scene_images, scene_explanations, title],
         outputs=[txt_file, pdf_file]
-)
+    )
+
 
 
     tts_btn.click(