Update app.py

app.py

@@ -111,7 +111,18 @@ def create_flipbook_from_pdf(pdf_file, view_mode="webgl", skin="light"):
             pages_info = process_pdf(pdf_path, session_id)
         else:
             return """<div style="color: red; padding: 20px;">Please upload a PDF file.</div>"""
-            
+        
+        if not pages_info:
+            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 from PDF: {e}")
+        return f"""<div style="color: red; padding: 20px;">An error occurred: {str(e)}</div>"""
+
 def create_flipbook_from_images(images, view_mode="webgl", skin="light"):
     """Create a flipbook from uploaded images."""
     try:
@@ -122,6 +133,8 @@ def create_flipbook_from_images(images, view_mode="webgl", skin="light"):
             # Process images
             pages_info = process_images(images, session_id)
         else:
+            return """<div style="color: red; padding: 20px;">Please upload at least one image.</div>"""
+        
         if not pages_info:
             return """<div style="color: red; padding: 20px;">Failed to process the uploaded images. Please try again.</div>"""
         
@@ -132,17 +145,6 @@ def create_flipbook_from_images(images, view_mode="webgl", skin="light"):
     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 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 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):
     """Generate HTML for the flipbook."""