using pdf preview

app.py

@@ -1,4 +1,5 @@
 import gradio as gr
+from gradio_pdf import PDF
 import numpy as np
 import random
 import torch
@@ -668,8 +669,6 @@ def create_multi_page_pdf(session_manager: SessionManager) -> str:
 @spaces.GPU(duration=180)  # Increased duration for up to 6 images
 def infer_page(
     prompt,
-    seed=42,
-    randomize_seed=False,
     guidance_scale=1.0,
     num_inference_steps=8,
     style_preset="no_style",
@@ -684,8 +683,6 @@ def infer_page(
 
     Args:
         prompt (str): The text prompt to generate images from.
-        seed (int): The seed for the random number generator for reproducibility.
-        randomize_seed (bool): If True, a random seed is used for each image.
         guidance_scale (float): Corresponds to `true_cfg_scale`.
         num_inference_steps (int): The number of denoising steps.
         style_preset (str): The key of the style preset to apply.
@@ -696,7 +693,7 @@ def infer_page(
         progress (gr.Progress): A Gradio Progress object to track generation.
 
     Returns:
-        tuple: Updated session state, PDF path, preview image, page info, and updated button label.
+        tuple: Updated session state, PDF path, and updated button label.
     """
     # Initialize or retrieve session
     if session_state is None or "session_id" not in session_state:
@@ -710,7 +707,7 @@ def infer_page(
 
     # Check page limit
     if session_manager.metadata["total_pages"] >= 128:
-        return session_state, None, None, "Maximum page limit (128) reached!", f"Page limit reached"
+        return session_state, None, None, f"Page limit reached"
 
     generated_images = []
     used_seeds = []
@@ -723,7 +720,7 @@ def infer_page(
     for i in range(int(num_images)):
         progress((i + 0.5) / num_images, f"Generating image {i+1} of {num_images} for page {session_manager.metadata['total_pages'] + 1}")
 
-        current_seed = seed + i if not randomize_seed else random.randint(0, MAX_SEED)
+        current_seed = random.randint(0, MAX_SEED)  # Always randomize seed
 
         # Get optimal aspect ratio based on position in layout
         position_data = get_layout_position_for_image(layout, int(num_images), i)
@@ -763,15 +760,11 @@ def infer_page(
     # Update session state
     session_state["page_count"] = page_num
 
-    # Prepare page info
-    seeds_str = ", ".join(str(s) for s in used_seeds)
-    page_info = f"Page {page_num} added\nSeeds: {seeds_str}\nTotal pages: {page_num}"
-
     # Next button label
     next_page_num = page_num + 1
     button_label = f"Generate page {next_page_num}" if next_page_num <= 128 else "Page limit reached"
 
-    return session_state, pdf_path, generated_images[0] if generated_images else None, page_info, button_label
+    return session_state, pdf_path, pdf_path, button_label
 
 # New inference function with automatic aspect ratio
 def infer_single_auto(
@@ -921,27 +914,13 @@ with gr.Blocks(css=css) as demo:
                 )
 
         with gr.Row():
-            with gr.Column(scale=1):
-                result_preview = gr.Image(label="Preview", show_label=True, type="pil")
-            with gr.Column(scale=1):
-                pdf_output = gr.File(label="Download PDF", show_label=True)
-                page_info = gr.Textbox(label="Page Info", show_label=True, interactive=False, lines=3)
+            with gr.Column():
+                pdf_preview = PDF(label="PDF Preview", show_label=True, height=600, elem_id="pdf-preview")
+                pdf_output = gr.File(label="Download PDF", show_label=True, elem_id="pdf-download")
                 gr.Markdown("""**Note:** Your images and PDF are saved for up to 24 hours.
                 You can continue adding pages (up to 128) by clicking the generate button.""")
 
         with gr.Accordion("Advanced Settings", open=False):
-            seed = gr.Slider(
-                label="Seed",
-                minimum=0,
-                maximum=MAX_SEED,
-                step=1,
-                value=0,
-            )
-
-            randomize_seed = gr.Checkbox(label="Randomize seed", value=True)
-
-            # Removed prompt_enhance checkbox - story generation is now always enabled
-
             with gr.Row():
                 guidance_scale = gr.Slider(
                     label="Guidance scale (True CFG Scale)",
@@ -1004,8 +983,6 @@ with gr.Blocks(css=css) as demo:
         fn=infer_page,
         inputs=[
             prompt,
-            seed,
-            randomize_seed,
             guidance_scale,
             num_inference_steps,
             style_preset,
@@ -1014,19 +991,19 @@ with gr.Blocks(css=css) as demo:
             layout_dropdown,
             session_state,
         ],
-        outputs=[session_state, pdf_output, result_preview, page_info, run_button],
+        outputs=[session_state, pdf_output, pdf_preview, run_button],
     )
 
     # Reset button functionality
     def reset_session():
         new_state = {"session_id": str(uuid.uuid4()), "page_count": 0}
-        return new_state, None, None, "", "Generate page 1"
+        return new_state, None, None, "Generate page 1"
 
     # Connect the reset button
     reset_button.click(
         fn=reset_session,
         inputs=[],
-        outputs=[session_state, pdf_output, result_preview, page_info, run_button]
+        outputs=[session_state, pdf_output, pdf_preview, run_button]
     )
 
 if __name__ == "__main__":

requirements.txt

@@ -7,4 +7,5 @@ sentencepiece
 dashscope
 pyyaml
 reportlab
-pypdf2
+pypdf2
+gradio_pdf==0.0.22