Update app.py

app.py

@@ -95,14 +95,13 @@ def process_images(image_paths, session_id):
     
     return pages_info
 
-def create_flipbook(upload_type, pdf_file=None, images=None, view_mode="webgl", skin="light"):
-    """Create a flipbook from uploaded PDF or images."""
+def create_flipbook_from_pdf(pdf_file, view_mode="webgl", skin="light"):
+    """Create a flipbook from uploaded PDF."""
     try:
         session_id = str(uuid.uuid4())
         pages_info = []
         
-        # Process based on upload type
-        if upload_type == "pdf" and pdf_file is not None:
+        if pdf_file is not None:
             # Save PDF to temp directory
             pdf_path = os.path.join(UPLOAD_DIR, f"{session_id}.pdf")
             with open(pdf_path, "wb") as f:
@@ -110,21 +109,39 @@ def create_flipbook(upload_type, pdf_file=None, images=None, view_mode="webgl",
                 
             # Process PDF
             pages_info = process_pdf(pdf_path, session_id)
-        elif upload_type == "images" and images is not None:
+        else:
+            return """<div style="color: red; padding: 20px;">Please upload a PDF file.</div>"""
+            
+def create_flipbook_from_images(images, view_mode="webgl", skin="light"):
+    """Create a flipbook from uploaded images."""
+    try:
+        session_id = str(uuid.uuid4())
+        pages_info = []
+        
+        if images is not None and len(images) > 0:
             # Process images
             pages_info = process_images(images, session_id)
         else:
-            return """<div style="color: red; padding: 20px;">Please upload a PDF file or images.</div>"""
+        if not pages_info:
+            return """<div style="color: red; padding: 20px;">Failed to process the uploaded images. Please try again.</div>"""
+        
+        # Create and return HTML for the flipbook
+        flipbook_html = generate_flipbook_html(pages_info, session_id, view_mode, skin)
+        return flipbook_html
+    
+    except Exception as e:
+        print(f"Error creating flipbook from images: {e}")
+        return f"""<div style="color: red; padding: 20px;">An error occurred: {str(e)}</div>"""
         
         if not pages_info:
-            return """<div style="color: red; padding: 20px;">Failed to process the uploaded file(s). Please try again.</div>"""
+            return """<div style="color: red; padding: 20px;">Failed to process the uploaded PDF. Please try again.</div>"""
         
         # Create and return HTML for the flipbook
         flipbook_html = generate_flipbook_html(pages_info, session_id, view_mode, skin)
         return flipbook_html
     
     except Exception as e:
-        print(f"Error creating flipbook: {e}")
+        print(f"Error creating flipbook from PDF: {e}")
         return f"""<div style="color: red; padding: 20px;">An error occurred: {str(e)}</div>"""
 
 def generate_flipbook_html(pages_info, session_id, view_mode, skin):
@@ -204,38 +221,53 @@ with gr.Blocks(title="3D Flipbook Viewer") as demo:
     with gr.Tabs():
         with gr.TabItem("PDF Upload"):
             pdf_file = gr.File(label="Upload PDF", file_types=[".pdf"])
+            
+            with gr.Accordion("Advanced Settings", open=False):
+                pdf_view_mode = gr.Dropdown(
+                    choices=["webgl", "3d", "2d", "swipe"], 
+                    value="webgl", 
+                    label="View Mode"
+                )
+                pdf_skin = gr.Dropdown(
+                    choices=["light", "dark", "gradient"], 
+                    value="light", 
+                    label="Skin"
+                )
+                
             pdf_create_btn = gr.Button("Create Flipbook from PDF")
+            pdf_output = gr.HTML(label="Flipbook Output")
+            
+            # Set up PDF event handler
+            pdf_create_btn.click(
+                fn=create_flipbook_from_pdf, 
+                inputs=[pdf_file, pdf_view_mode, pdf_skin], 
+                outputs=pdf_output
+            )
             
         with gr.TabItem("Image Upload"):
             images = gr.File(label="Upload Images", file_types=["image"], file_count="multiple")
+            
+            with gr.Accordion("Advanced Settings", open=False):
+                img_view_mode = gr.Dropdown(
+                    choices=["webgl", "3d", "2d", "swipe"], 
+                    value="webgl", 
+                    label="View Mode"
+                )
+                img_skin = gr.Dropdown(
+                    choices=["light", "dark", "gradient"], 
+                    value="light", 
+                    label="Skin"
+                )
+                
             img_create_btn = gr.Button("Create Flipbook from Images")
-    
-    with gr.Accordion("Advanced Settings", open=False):
-        view_mode = gr.Dropdown(
-            choices=["webgl", "3d", "2d", "swipe"], 
-            value="webgl", 
-            label="View Mode"
-        )
-        skin = gr.Dropdown(
-            choices=["light", "dark", "gradient"], 
-            value="light", 
-            label="Skin"
-        )
-    
-    output = gr.HTML(label="Flipbook Output")
-    
-    # Set up event handlers
-    pdf_create_btn.click(
-        fn=create_flipbook, 
-        inputs=[gr.Textbox(value="pdf", visible=False), pdf_file, None, view_mode, skin], 
-        outputs=output
-    )
-    
-    img_create_btn.click(
-        fn=create_flipbook, 
-        inputs=[gr.Textbox(value="images", visible=False), None, images, view_mode, skin], 
-        outputs=output
-    )
+            img_output = gr.HTML(label="Flipbook Output")
+            
+            # Set up image event handler
+            img_create_btn.click(
+                fn=create_flipbook_from_images, 
+                inputs=[images, img_view_mode, img_skin], 
+                outputs=img_output
+            )
     
 # Launch the app
 if __name__ == "__main__":