Update app.py

app.py

@@ -347,7 +347,7 @@ def create_pdf_preview(pdf_path: Path):
 
 # --- Main API Function ---
 def generate_pdfs_api(files, ai_content, layouts, fonts, num_columns, header_text, footer_text, font_size_body, font_size_h1, font_size_h2, font_size_h3, margin_top, margin_bottom, margin_left, margin_right, progress=gr.Progress(track_tqdm=True)):
-    if not files and not ai_content.strip(): raise gr.Error("Please conjure some content or upload an image before alchemizing!")
+    if not files and (not ai_content or "Golem awaits" in ai_content): raise gr.Error("Please conjure some content or upload a file before alchemizing!")
     if not layouts: raise gr.Error("You must select a scroll (page layout)!")
     if not fonts: raise gr.Error("A scribe needs a font! Please choose one.")
     if not EMOJI_FONT_PATH: raise gr.Error("CRITICAL: Cannot generate PDFs. 'NotoColorEmoji-Regular.ttf' not found. Please add it to the app directory.")
@@ -355,10 +355,34 @@ def generate_pdfs_api(files, ai_content, layouts, fonts, num_columns, header_tex
     shutil.rmtree(OUTPUT_DIR, ignore_errors=True); shutil.rmtree(PREVIEW_DIR, ignore_errors=True)
     OUTPUT_DIR.mkdir(); PREVIEW_DIR.mkdir()
 
-    md_content = ai_content if ai_content and ai_content.strip() else ""
-    image_files = [Path(f.name) for f in files if Path(f.name).suffix.lower() in ['.png', '.jpg', '.jpeg']] if files else []
-
-    log_updates, generated_pdf_paths = "", []
+    # Separate uploaded files by type
+    image_files, pdf_files, txt_files = [], [], []
+    if files:
+        for f in files:
+            file_path = Path(f.name)
+            ext = file_path.suffix.lower()
+            if ext in ['.png', '.jpg', '.jpeg', '.gif', '.bmp', '.tiff']:
+                image_files.append(file_path)
+            elif ext == '.pdf':
+                pdf_files.append(file_path)
+            elif ext == '.txt':
+                txt_files.append(file_path)
+
+    log_updates = ""
+    # Consolidate all text content
+    all_text_content = []
+    if ai_content and "Golem awaits" not in ai_content:
+        all_text_content.append(ai_content)
+
+    for txt_path in txt_files:
+        try:
+            all_text_content.append(txt_path.read_text(encoding='utf-8'))
+        except Exception as e:
+            log_updates += f"⚠️ Failed to read text file {txt_path.name}: {e}\n"
+    
+    md_content = "\n\n---\n\n".join(all_text_content)
+    
+    generated_pdf_paths = []
     
     # Clear emoji cache for each run to save memory
     EMOJI_IMAGE_CACHE.clear()
@@ -367,6 +391,7 @@ def generate_pdfs_api(files, ai_content, layouts, fonts, num_columns, header_tex
         for font_name in progress.tqdm(fonts, desc=f" enchanting scrolls with {layout_name}..."):
             merger = PdfWriter()
             
+            # 1. Process main text content (AI + TXT)
             if md_content:
                 md_buffer = io.BytesIO()
                 story, title = markdown_to_story(md_content, font_name, EMOJI_FONT_NAME, font_size_body, font_size_h1, font_size_h2, font_size_h3)
@@ -387,6 +412,7 @@ def generate_pdfs_api(files, ai_content, layouts, fonts, num_columns, header_tex
                 md_buffer.seek(0)
                 merger.append(fileobj=md_buffer)
 
+            # 2. Append image files
             for img_path in image_files:
                 try:
                     with Image.open(img_path) as img: img_width_px, img_height_px = img.size
@@ -398,6 +424,14 @@ def generate_pdfs_api(files, ai_content, layouts, fonts, num_columns, header_tex
                     merger.append(fileobj=img_buffer)
                 except Exception as e: log_updates += f"⚠️ Failed to process image {img_path.name}: {e}\n"
 
+            # 3. Append PDF files
+            for pdf_path in pdf_files:
+                try:
+                    merger.append(str(pdf_path))
+                except Exception as e:
+                    log_updates += f"⚠️ Failed to merge PDF {pdf_path.name}: {e}\n"
+
+            # 4. Write the final merged PDF
             if len(merger.pages) > 0:
                 time_str = datetime.datetime.now().strftime('%H%M%S')
                 clean_layout = layout_name.replace(' ', '')
@@ -455,11 +489,11 @@ with gr.Blocks(theme=theme, title="The PDF Alchemist") as demo:
                     gr.Markdown("### 🤖 The Idea Golem")
                     ai_prompt = gr.Textbox(label="Command the Golem", placeholder="e.g., 'A recipe for a dragon's breath chili...'")
                     generate_ai_btn = gr.Button("🧠 Animate Golem!")
-                    ai_content_output = gr.Textbox(label="Golem's Manuscript (Editable)", lines=15, interactive=True)
+                    ai_content_output = gr.Textbox(label="Golem's Manuscript (Editable)", lines=15, interactive=True, value="# The Golem awaits your command!\n\n")
                     gr.Markdown("<hr style='border-color: #374151;'>")
                     gr.Markdown("### 📤 Or, Upload Your Treasures")
-                    uploaded_files = gr.File(label="Upload Images (.png, .jpg)", file_count="multiple", file_types=[".png", ".jpg", ".jpeg"])
-                    gr.Markdown("*<p style='font-size:0.8rem; color: #9ca3af;'>Note: The Golem's manuscript takes precedence. Images are always added.</p>*")
+                    uploaded_files = gr.File(label="Upload Files (Images, PDFs, TXT)", file_count="multiple", file_types=['.png', '.jpg', '.jpeg', '.gif', '.bmp', '.tiff', '.pdf', '.txt'])
+                    gr.Markdown("*<p style='font-size:0.8rem; color: #9ca3af;'>Note: The Golem's manuscript and .txt files are combined. Images and PDF pages are appended after the text content.</p>*")
                 
                 with gr.TabItem("② Define Your Canvas 🎨"):
                     gr.Markdown("### 📐 Choose Your Scroll")