Update app.py

app.py

@@ -74,11 +74,16 @@ def text_to_speech(text):
         print("TTS error:", e)
         return None
 
-def generate_pdf(images, explanations):
+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()
-        story = [Paragraph("AI-Generated Story Scenes", styles["Title"]), Spacer(1, 12)]
+        
+        # Add the title at the top
+        story = [
+            Paragraph(title, styles["Title"]),
+            Spacer(1, 24)
+        ]
 
         for i, img in enumerate(images):
             if img:
@@ -94,6 +99,7 @@ def generate_pdf(images, explanations):
         doc.build(story)
         return tmp.name
 
+
 # -------------------- Core Logic --------------------
 
 def generate_scene(num_scenes, theme, char_count, character_names, dialogue,
@@ -127,11 +133,11 @@ def generate_scene(num_scenes, theme, char_count, character_names, dialogue,
 
     return image, summary, explanation, images, summaries, explanations, status, gr.update(visible=done_visible)
 
-def finalize_story(images, explanations):
+def finalize_story(images, explanations, title):
     if not images or not explanations:
         return None, None
 
-    pdf = generate_pdf(images, explanations)
+    pdf = generate_pdf(images, explanations, title=title)
 
     with tempfile.NamedTemporaryFile(delete=False, suffix=".txt", mode="w", encoding="utf-8") as txt:
         for i, exp in enumerate(explanations):
@@ -140,6 +146,7 @@ def finalize_story(images, explanations):
 
     return txt_path, pdf
 
+
 def ai_write_scene(theme, total_scenes, scene_summaries):
     try:
         scene_index = len(scene_summaries) + 1
@@ -406,9 +413,10 @@ with gr.Blocks(
 
     done_btn.click(
         fn=finalize_story,
-        inputs=[scene_images, scene_explanations],
+        inputs=[scene_images, scene_explanations, theme],
         outputs=[txt_file, pdf_file]
-    )
+)
+
 
     tts_btn.click(
         fn=text_to_speech,