Update app.py

app.py

@@ -1,13 +1,15 @@
 #!/usr/bin/env python3
 # -*- coding: utf-8 -*-
 """
-3D Flipbook Viewer (Gradio) – 전체 소스 (SyntaxError 수정 버전)
+3D Flipbook Viewer (Gradio) – 최종 수정 버전
 최종 수정: 2025-05-18
+
+- f-string 구문에서 JS의 { }를 {{ }}로 이스케이프 처리
+- Python 3.9 이하 호환(typing.Optional, typing.List)
+- Gradio가 생성한 임시 파일 → temp/uploads 폴더로 복사 후 처리
+- /public 폴더에 최종 HTML 생성 (정적 서빙 별도 설정 필요)
 """
 
-# ────────────────────────────
-# 기본 모듈
-# ────────────────────────────
 import os
 import shutil
 import uuid
@@ -17,7 +19,6 @@ import traceback
 from pathlib import Path
 from typing import Optional, List, Dict
 
-# 외부 라이브러리
 import gradio as gr
 from PIL import Image
 import fitz  # PyMuPDF
@@ -26,28 +27,28 @@ import fitz  # PyMuPDF
 # 로깅 설정
 # ────────────────────────────
 logging.basicConfig(
-    level=logging.INFO,                       # 필요하면 DEBUG
+    level=logging.INFO,
     format="%(asctime)s [%(levelname)s] %(message)s",
-    filename="app.log",                       # 동일 디렉터리에 로그 파일 생성
+    filename="app.log",
     filemode="a",
 )
 logging.info("🚀 Flipbook app started")
 
 # ────────────────────────────
-# 상수 / 경로
+# 폴더 경로 설정
 # ────────────────────────────
 TEMP_DIR   = "temp"
 UPLOAD_DIR = os.path.join(TEMP_DIR, "uploads")
 OUTPUT_DIR = os.path.join(TEMP_DIR, "output")
 THUMBS_DIR = os.path.join(OUTPUT_DIR, "thumbs")
-HTML_DIR   = os.path.join("public", "flipbooks")   # 웹으로 노출되는 위치
+HTML_DIR   = os.path.join("public", "flipbooks")
 
-# 디렉터리 보장
+# 폴더 생성
 for d in [TEMP_DIR, UPLOAD_DIR, OUTPUT_DIR, THUMBS_DIR, HTML_DIR]:
     os.makedirs(d, exist_ok=True)
 
 # ────────────────────────────
-# 유틸 함수
+# 유틸 함수: 이미지 썸네일
 # ────────────────────────────
 def create_thumbnail(src: str, dst: str, size=(300, 300)) -> Optional[str]:
     """원본 이미지를 썸네일로 저장 (이미지 열기 실패 시 None 리턴)"""
@@ -61,10 +62,10 @@ def create_thumbnail(src: str, dst: str, size=(300, 300)) -> Optional[str]:
         return None
 
 # ────────────────────────────
-# PDF → 이미지
+# PDF 처리 함수: PDF → 이미지
 # ────────────────────────────
 def process_pdf(pdf_path: str, session_id: str) -> List[Dict]:
-    """PDF 파일을 페이지 별 PNG로 변환하고 페이지 정보를 리스트로 리턴"""
+    """PDF 파일을 페이지 별 PNG로 변환하고, 페이지 정보 리스트 반환"""
     pages_info = []
     out_dir    = os.path.join(OUTPUT_DIR, session_id)
     th_dir     = os.path.join(THUMBS_DIR, session_id)
@@ -74,7 +75,7 @@ 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):
-            # 해상도 향상을 위해 매트릭스 사용 (1.5배 정도)
+            # 해상도(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")
@@ -83,7 +84,7 @@ def process_pdf(pdf_path: str, session_id: str) -> List[Dict]:
             thumb_path = os.path.join(th_dir, f"thumb_{idx+1}.png")
             create_thumbnail(img_path, thumb_path)
 
-            # 첫 페이지에만 예시로 오버레이 HTML 제공
+            # 첫 페이지 예시로 오버레이 HTML
             html_overlay = (
                 """
                 <div style="position:absolute;top:50px;left:50px;
@@ -116,10 +117,10 @@ def process_pdf(pdf_path: str, session_id: str) -> List[Dict]:
         return []
 
 # ────────────────────────────
-# 이미지 업로드 처리
+# 이미지 처리 함수
 # ────────────────────────────
 def process_images(img_paths: List[str], session_id: str) -> List[Dict]:
-    """업로드된 이미지를 복사/썸네일 생성 후 페이지 정보로 리턴"""
+    """이미지들을 temp/output으로 복사, 썸네일 생성, 페이지 정보 반환"""
     pages_info = []
     out_dir = os.path.join(OUTPUT_DIR, session_id)
     th_dir  = os.path.join(THUMBS_DIR, session_id)
@@ -134,7 +135,7 @@ def process_images(img_paths: List[str], session_id: str) -> List[Dict]:
             thumb = os.path.join(th_dir, f"thumb_{i+1}.png")
             create_thumbnail(dst, thumb)
 
-            # 페이지별 간단한 오버레이 예시
+            # 페이지별 오버레이 예시
             if i == 0:
                 html_overlay = """
                 <div style="position:absolute;top:50px;left:50px;
@@ -188,8 +189,11 @@ def generate_flipbook_html(
     view_mode: str,
     skin: str
 ) -> str:
-    """페이지 정보를 3D Flipbook에 적용할 HTML 파일로 만들고 링크 반환"""
-    # htmlContent가 None인 경우는 JSON에서 제거
+    """
+    3D Flipbook 용 HTML 파일 생성 후, HTML 링크(버튼) 블록을 반환
+    - f-string 안에서 JS의 { }는 {{ }}로 이스케이프
+    """
+    # htmlContent=None 제거
     for p in pages_info:
         if p.get("htmlContent") is None:
             p.pop("htmlContent", None)
@@ -198,7 +202,6 @@ def generate_flipbook_html(
     html_file  = f"flipbook_{session_id}.html"
     html_path  = os.path.join(HTML_DIR, html_file)
 
-    # f-string 내부에서 JS용 { }를 표현하려면 {{ }} 로 써야 함
     html = f"""
 <!DOCTYPE html>
 <html lang="ko">
@@ -232,7 +235,6 @@ def generate_flipbook_html(
 
 <script>
 document.addEventListener('DOMContentLoaded', () => {{
-  // hide() 함수: 로딩 메시지 제거
   const hide = () => {{
     document.getElementById('loading').style.display = 'none';
   }};
@@ -270,6 +272,7 @@ document.addEventListener('DOMContentLoaded', () => {{
     Path(html_path).write_text(html, encoding="utf-8")
     public_url = f"/public/flipbooks/{html_file}"
 
+    # 사용자에게 돌려줄 HTML 블록
     return f"""
 <div style="text-align:center;padding:20px;background:#f9f9f9;border-radius:5px">
   <h2 style="margin:0;color:#333">플립북이 준비되었습니다!</h2>
@@ -283,7 +286,7 @@ document.addEventListener('DOMContentLoaded', () => {{
 """
 
 # ────────────────────────────
-# 콜백: PDF 업로드
+# PDF 업로드 콜백
 # ────────────────────────────
 def create_flipbook_from_pdf(
     pdf_file: Optional[gr.File],
@@ -300,24 +303,24 @@ def create_flipbook_from_pdf(
         )
 
     try:
-        # Gradio가 넘겨준 임시 PDF 경로
+        # Gradio 임시 경로
         uploaded_temp_path = pdf_file.name
 
-        # 서버 내 임시 업로드 폴더에 안전하게 복사
+        # temp/uploads 폴더에 복사
         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 → 페이지 이미지 변환
+        # 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 생성
+        # 플립북 HTML
         html_block = generate_flipbook_html(pages_info, session_id, view_mode, skin)
         return html_block, "\n".join(debug)
 
@@ -331,7 +334,7 @@ def create_flipbook_from_pdf(
         )
 
 # ────────────────────────────
-# 콜백: 이미지 업로드
+# 이미지 업로드 콜백
 # ────────────────────────────
 def create_flipbook_from_images(
     images: Optional[List[gr.File]],
@@ -348,10 +351,9 @@ def create_flipbook_from_images(
         )
 
     try:
-        # 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)
@@ -360,14 +362,14 @@ def create_flipbook_from_images(
 
         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 생성
+        # 플립북 HTML
         html_block = generate_flipbook_html(pages_info, session_id, view_mode, skin)
         return html_block, "\n".join(debug)
 
@@ -391,16 +393,8 @@ with gr.Blocks(title="3D Flipbook Viewer") as demo:
         with gr.TabItem("PDF 업로드"):
             pdf_file = gr.File(label="PDF 파일", file_types=[".pdf"])
             with gr.Accordion("고급 설정", open=False):
-                pdf_view = gr.Radio(
-                    ["webgl", "3d", "2d", "swipe"],
-                    value="2d",
-                    label="뷰 모드",
-                )
-                pdf_skin = gr.Radio(
-                    ["light", "dark", "gradient"],
-                    value="light",
-                    label="스��",
-                )
+                pdf_view = gr.Radio(["webgl", "3d", "2d", "swipe"], value="2d", label="뷰 모드")
+                pdf_skin = gr.Radio(["light", "dark", "gradient"], value="light", label="스킨")
             pdf_btn   = gr.Button("PDF → 플립북", variant="primary")
             pdf_out   = gr.HTML()
             pdf_dbg   = gr.Textbox(label="디버그", lines=10)
@@ -413,22 +407,10 @@ with gr.Blocks(title="3D Flipbook Viewer") as demo:
 
         # 이미지 탭
         with gr.TabItem("이미지 업로드"):
-            imgs = gr.File(
-                label="이미지 파일들",
-                file_types=["image"],
-                file_count="multiple",
-            )
+            imgs = gr.File(label="이미지 파일들", file_types=["image"], file_count="multiple")
             with gr.Accordion("고급 설정", open=False):
-                img_view = gr.Radio(
-                    ["webgl", "3d", "2d", "swipe"],
-                    value="2d",
-                    label="뷰 모드",
-                )
-                img_skin = gr.Radio(
-                    ["light", "dark", "gradient"],
-                    value="light",
-                    label="스킨",
-                )
+                img_view = gr.Radio(["webgl", "3d", "2d", "swipe"], value="2d", label="뷰 모드")
+                img_skin = gr.Radio(["light", "dark", "gradient"], value="light", label="스킨")
             img_btn = gr.Button("이미지 → 플립북", variant="primary")
             img_out = gr.HTML()
             img_dbg = gr.Textbox(label="디버그", lines=10)
@@ -450,5 +432,5 @@ with gr.Blocks(title="3D Flipbook Viewer") as demo:
 # 실행
 # ────────────────────────────
 if __name__ == "__main__":
-    # 필요한 경우 share=True 등 인자로 추가 가능
+    # share=True 등 옵션을 넣어 배포/공유 가능
     demo.launch(debug=True)