Update app.py

app.py

@@ -130,44 +130,44 @@ def process_images(image_paths, session_id):
     return pages_info
 
 def create_flipbook_from_pdf(pdf_file, view_mode="2d", skin="light"):
-    """Create a flipbook from uploaded PDF."""
+    """Create a flipbook from an uploaded PDF."""
+    session_id = str(uuid.uuid4())
+    debug_info = ""
+
+    if not pdf_file:
+        return (
+            "<div style='color:red;padding:20px;'>PDF 파일을 업로드해주세요.</div>",
+            "No file uploaded",
+        )
+
     try:
-        session_id = str(uuid.uuid4())
-        pages_info = []
-        debug_info = ""
-        
-        if pdf_file is not None:
-            # In Gradio, pdf_file is a file path string, not the actual content
-            pdf_path = pdf_file.name  # Get the file path
-            debug_info += f"PDF path: {pdf_path}\n"
-            
-            # Copy the PDF file to a location accessible by the public URL
-            pdf_public_path = os.path.join(HTML_DIR, f"pdf_{session_id}.pdf")
-            shutil.copy(pdf_path, pdf_public_path)
-            
-            # Create a simple HTML page with direct PDF loading
-            html_filename = f"flipbook_{session_id}.html"
-            html_path = os.path.join(HTML_DIR, html_filename)
-            
-            # Create HTML content that uses CDN resources
-            html_content = create_flipbook_html_page(f"pdf_{session_id}.pdf", session_id, view_mode, skin)
-            
-            # Write the HTML file
-            with open(html_path, 'w', encoding='utf-8') as f:
-                f.write(html_content)
-            
-            # Generate link to the HTML file
-            public_url = f"/public/flipbooks/{html_filename}"
-            flipbook_link = generate_flipbook_link(public_url, session_id, view_mode, skin)
-            
-            return flipbook_link, debug_info
-        else:
-            return """<div style="color: red; padding: 20px;">PDF 파일을 업로드해주세요.</div>""", "No file uploaded"
-    
+        pdf_path = pdf_file.name                      # Gradio File 객체의 실제 경로
+        debug_info += f"PDF path: {pdf_path}\n"
+
+        # 1) PDF 페이지를 이미지로 변환
+        pages_info = process_pdf(pdf_path, session_id)
+        debug_info += f"Number of pages: {len(pages_info)}\n"
+
+        if not pages_info:
+            return (
+                "<div style='color:red;padding:20px;'>PDF 처리 실패.</div>",
+                "No pages processed",
+            )
+
+        # 2) 이미지 리스트로 플립북 HTML 생성
+        iframe_html = generate_flipbook_html(
+            pages_info, session_id, view_mode, skin
+        )
+        return iframe_html, debug_info
+
     except Exception as e:
         error_msg = f"Error creating flipbook from PDF: {e}"
         print(error_msg)
-        return f"""<div style="color: red; padding: 20px;">오류가 발생했습니다: {str(e)}</div>""", error_msg
+        return (
+            f"<div style='color:red;padding:20px;'>오류가 발생했습니다: {e}</div>",
+            error_msg,
+        )
+
 
 def create_flipbook_from_images(images, view_mode="2d", skin="light"):
     """Create a flipbook from uploaded images."""