Update app.py

app.py

@@ -1,7 +1,7 @@
 #!/usr/bin/env python3
 # -*- coding: utf-8 -*-
 """
-3D Flipbook Viewer (Gradio) – 전체 소스
+3D Flipbook Viewer (Gradio) – 전체 소스 (수정 버전)
 최종 수정: 2025-05-18
 """
 
@@ -15,11 +15,12 @@ import json
 import logging
 import traceback
 from pathlib import Path
+from typing import Optional, List, Dict
 
 # 외부 라이브러리
 import gradio as gr
 from PIL import Image
-import fitz            # PyMuPDF
+import fitz  # PyMuPDF
 
 # ────────────────────────────
 # 로깅 설정
@@ -48,8 +49,8 @@ for d in [TEMP_DIR, UPLOAD_DIR, OUTPUT_DIR, THUMBS_DIR, HTML_DIR]:
 # ────────────────────────────
 # 유틸 함수
 # ────────────────────────────
-def create_thumbnail(src: str, dst: str, size=(300, 300)) -> str | None:
-    """원본 이미지를 썸네일로 저장"""
+def create_thumbnail(src: str, dst: str, size=(300, 300)) -> Optional[str]:
+    """원본 이미지를 썸네일로 저장 (이미지 열기 실패 시 None 리턴)"""
     try:
         with Image.open(src) as im:
             im.thumbnail(size, Image.LANCZOS)
@@ -59,11 +60,11 @@ def create_thumbnail(src: str, dst: str, size=(300, 300)) -> str | None:
         logging.error("Thumbnail error: %s", e)
         return None
 
-
 # ────────────────────────────
 # PDF → 이미지
 # ────────────────────────────
-def process_pdf(pdf_path: str, session_id: str) -> list[dict]:
+def process_pdf(pdf_path: str, session_id: str) -> List[Dict]:
+    """PDF 파일을 페이지 별 PNG로 변환하고 페이지 정보를 리스트로 리턴"""
     pages_info = []
     out_dir    = os.path.join(OUTPUT_DIR, session_id)
     th_dir     = os.path.join(THUMBS_DIR, session_id)
@@ -73,13 +74,16 @@ def process_pdf(pdf_path: str, session_id: str) -> list[dict]:
     try:
         pdf_doc = fitz.open(pdf_path)
         for idx, page in enumerate(pdf_doc):
-            pix = page.get_pixmap(matrix=fitz.Matrix(2, 2))   # 2× 해상도
+            # 해상도 향상을 위해 매트릭스 사용 (1.5배 정도)
+            mat = fitz.Matrix(1.5, 1.5)
+            pix = page.get_pixmap(matrix=mat)
             img_path = os.path.join(out_dir, f"page_{idx+1}.png")
             pix.save(img_path)
 
             thumb_path = os.path.join(th_dir, f"thumb_{idx+1}.png")
             create_thumbnail(img_path, thumb_path)
 
+            # 첫 페이지에만 예시로 오버레이 HTML 제공
             html_overlay = (
                 """
                 <div style="position:absolute;top:50px;left:50px;
@@ -111,11 +115,11 @@ def process_pdf(pdf_path: str, session_id: str) -> list[dict]:
         logging.error("process_pdf() failed: %s", e)
         return []
 
-
 # ────────────────────────────
 # 이미지 업로드 처리
 # ────────────────────────────
-def process_images(img_paths: list[str], session_id: str) -> list[dict]:
+def process_images(img_paths: List[str], session_id: str) -> List[Dict]:
+    """업로드된 이미지를 복사/썸네일 생성 후 페이지 정보로 리턴"""
     pages_info = []
     out_dir = os.path.join(OUTPUT_DIR, session_id)
     th_dir  = os.path.join(THUMBS_DIR, session_id)
@@ -128,8 +132,9 @@ def process_images(img_paths: list[str], session_id: str) -> list[dict]:
             shutil.copy(src, dst)
 
             thumb = os.path.join(th_dir, f"thumb_{i+1}.png")
-            create_thumbnail(src, thumb)
+            create_thumbnail(dst, thumb)
 
+            # 페이지별 간단한 오버레이 예시
             if i == 0:
                 html_overlay = """
                 <div style="position:absolute;top:50px;left:50px;
@@ -174,14 +179,17 @@ def process_images(img_paths: list[str], session_id: str) -> list[dict]:
 
     return pages_info
 
-
 # ────────────────────────────
 # 플립북 HTML 생성
 # ────────────────────────────
 def generate_flipbook_html(
-    pages_info: list[dict], session_id: str, view_mode: str, skin: str
+    pages_info: List[Dict],
+    session_id: str,
+    view_mode: str,
+    skin: str
 ) -> str:
-    # None 값은 JSON 직렬화 전에 제거
+    """페이지 정보를 3D Flipbook에 적용할 HTML 파일로 만들고 링크 반환"""
+    # htmlContent가 None인 경우는 JSON에서 제거
     for p in pages_info:
         if p.get("htmlContent") is None:
             p.pop("htmlContent", None)
@@ -198,6 +206,7 @@ def generate_flipbook_html(
 <meta name="viewport" content="width=device-width,initial-scale=1">
 <title>3D Flipbook</title>
 
+<!-- 3D Flipbook 관련 CSS/JS -->
 <link rel="stylesheet" href="/public/libs/flipbook/css/flipbook.style.css">
 <script src="/public/libs/flipbook/js/flipbook.min.js"></script>
 <script src="/public/libs/flipbook/js/flipbook.webgl.min.js"></script>
@@ -224,33 +233,40 @@ def generate_flipbook_html(
 document.addEventListener('DOMContentLoaded',()=>{
   const hide=()=>{{document.getElementById('loading').style.display='none'}};
   try{{
-    const options={{pages:{pages_json},
-                    viewMode:"{view_mode}",
-                    skin:"{skin}",
-                    responsiveView:true,
-                    singlePageMode:false,
-                    singlePageModeIfMobile:true,
-                    pageFlipDuration:1,
-                    thumbnailsOnStart:true,
-                    btnThumbs:{{enabled:true}},
-                    btnPrint:{{enabled:true}},
-                    btnDownloadPages:{{enabled:true}},
-                    btnDownloadPdf:{{enabled:true}},
-                    btnShare:{{enabled:true}},
-                    btnSound:{{enabled:true}},
-                    btnExpand:{{enabled:true}} }};
-    new FlipBook(document.getElementById('flipbook-container'),options);
-    setTimeout(hide,1000);
-  }}catch(e){{console.error(e);alert('플립북 초기화 오류:'+e.message);}}
+    const options = {{
+      pages: {pages_json},
+      viewMode: "{view_mode}",
+      skin: "{skin}",
+      responsiveView: true,
+      singlePageMode: false,
+      singlePageModeIfMobile: true,
+      pageFlipDuration: 1,
+      thumbnailsOnStart: true,
+      btnThumbs: {{enabled:true}},
+      btnPrint: {{enabled:true}},
+      btnDownloadPages: {{enabled:true}},
+      btnDownloadPdf: {{enabled:true}},
+      btnShare: {{enabled:true}},
+      btnSound: {{enabled:true}},
+      btnExpand: {{enabled:true}}
+    }};
+    new FlipBook(document.getElementById('flipbook-container'), options);
+    setTimeout(hide, 1000);
+  }} catch(e) {{
+    console.error(e);
+    alert('플립북 초기화 오류:' + e.message);
+  }}
 });
 </script>
-</body></html>
+</body>
+</html>
 """
 
     Path(html_path).write_text(html, encoding="utf-8")
+
     public_url = f"/public/flipbooks/{html_file}"
 
-    # 사용자에게 돌려줄 링크 덩어리
+    # 사용자에게 돌려줄 버튼 형태 링크
     return f"""
 <div style="text-align:center;padding:20px;background:#f9f9f9;border-radius:5px">
   <h2 style="margin:0;color:#333">플립북이 준비되었습니다!</h2>
@@ -263,15 +279,16 @@ document.addEventListener('DOMContentLoaded',()=>{
 </div>
 """
 
-
 # ────────────────────────────
 # 콜백: PDF 업로드
 # ────────────────────────────
 def create_flipbook_from_pdf(
-    pdf_file: gr.File | None, view_mode="2d", skin="light"
+    pdf_file: Optional[gr.File],
+    view_mode: str = "2d",
+    skin: str = "light"
 ):
     session_id = str(uuid.uuid4())
-    debug: list[str] = []
+    debug: List[str] = []
 
     if not pdf_file:
         return (
@@ -280,18 +297,25 @@ def create_flipbook_from_pdf(
         )
 
     try:
-        pdf_path = pdf_file.name
-        debug.append(f"PDF path: {pdf_path}")
+        # Gradio가 넘겨준 임시 PDF 경로
+        uploaded_temp_path = pdf_file.name
+
+        # 서버 내 임시 업로드 폴더에 안전하게 복사
+        filename_only = os.path.basename(uploaded_temp_path)
+        pdf_path = os.path.join(UPLOAD_DIR, filename_only)
+        shutil.copyfile(uploaded_temp_path, pdf_path)
 
+        debug.append(f"Copied PDF to: {pdf_path}")
+
+        # PDF → 페이지 이미지 변환
         pages_info = process_pdf(pdf_path, session_id)
         debug.append(f"Extracted pages: {len(pages_info)}")
 
         if not pages_info:
             raise RuntimeError("PDF 처리 결과가 비어 있습니다.")
 
-        html_block = generate_flipbook_html(
-            pages_info, session_id, view_mode, skin
-        )
+        # 플립북 HTML 생성
+        html_block = generate_flipbook_html(pages_info, session_id, view_mode, skin)
         return html_block, "\n".join(debug)
 
     except Exception as e:
@@ -303,15 +327,16 @@ def create_flipbook_from_pdf(
             "\n".join(debug),
         )
 
-
 # ────────────────────────────
 # 콜백: 이미지 업로드
 # ────────────────────────────
 def create_flipbook_from_images(
-    images: list[gr.File] | None, view_mode="2d", skin="light"
+    images: Optional[List[gr.File]],
+    view_mode: str = "2d",
+    skin: str = "light"
 ):
     session_id = str(uuid.uuid4())
-    debug: list[str] = []
+    debug: List[str] = []
 
     if not images:
         return (
@@ -320,18 +345,27 @@ def create_flipbook_from_images(
         )
 
     try:
-        img_paths = [f.name for f in images]
+        # Gradio가 넘겨준 임시 이미지 경로들
+        img_paths = []
+        for fobj in images:
+            # 안전하게 temp 폴더에 복사
+            uploaded_temp_path = fobj.name
+            filename_only = os.path.basename(uploaded_temp_path)
+            local_img_path = os.path.join(UPLOAD_DIR, filename_only)
+            shutil.copyfile(uploaded_temp_path, local_img_path)
+            img_paths.append(local_img_path)
+
         debug.append(f"Images: {img_paths}")
 
+        # 이미지 → 페이지 정보 변환
         pages_info = process_images(img_paths, session_id)
         debug.append(f"Processed: {len(pages_info)}")
 
         if not pages_info:
             raise RuntimeError("이미지 처리 실패")
 
-        html_block = generate_flipbook_html(
-            pages_info, session_id, view_mode, skin
-        )
+        # 플립북 HTML 생성
+        html_block = generate_flipbook_html(pages_info, session_id, view_mode, skin)
         return html_block, "\n".join(debug)
 
     except Exception as e:
@@ -343,7 +377,6 @@ def create_flipbook_from_images(
             "\n".join(debug),
         )
 
-
 # ────────────────────────────
 # Gradio UI
 # ────────────────────────────
@@ -413,5 +446,6 @@ with gr.Blocks(title="3D Flipbook Viewer") as demo:
 # ────────────────────────────
 # 실행
 # ────────────────────────────
-if __name__ == \"__main__\":
-    demo.launch(debug=True)   # share=True 필요 시 추가
+if __name__ == "__main__":
+    # 필요한 경우 share=True 등 인자로 추가 가능
+    demo.launch(debug=True)