Spaces:

ginipick
/

AI-BOOK

Running on CPU Upgrade

App Files Files Community

ginipick commited on about 1 month ago

Commit

e9d99da

verified ·

1 Parent(s): dfb5abe

Create app.py

Browse files

Files changed (1) hide show

app.py +447 -0

app.py ADDED Viewed

	@@ -0,0 +1,447 @@

+from fastapi import FastAPI
+from fastapi.responses import HTMLResponse
+from fastapi.staticfiles import StaticFiles
+import pathlib, os, uvicorn, base64
+BASE = pathlib.Path(__file__).parent
+app = FastAPI()
+app.mount("/static", StaticFiles(directory=BASE), name="static")
+# PDF 디렉토리 설정
+PDF_DIR = BASE / "pdf"
+if not PDF_DIR.exists():
+    PDF_DIR.mkdir(parents=True)
+# PDF 파일 목록 가져오기
+def get_pdf_files():
+    pdf_files = []
+    if PDF_DIR.exists():
+        pdf_files = [f for f in PDF_DIR.glob("*.pdf")]
+    return pdf_files
+# PDF 썸네일 생성 및 프로젝트 데이터 준비
+def generate_pdf_projects():
+    projects_data = []
+    pdf_files = get_pdf_files()
+    for pdf_file in pdf_files:
+        projects_data.append({
+            "path": str(pdf_file),
+            "name": pdf_file.stem
+        })
+    return projects_data
+HTML = """
+<!doctype html><html lang="ko"><head>
+<meta charset="utf-8"><title>FlipBook Space</title>
+<link rel="stylesheet" href="/static/flipbook.css">
+<script src="/static/three.js"></script>
+<script src="/static/iscroll.js"></script>
+<script src="/static/mark.js"></script>
+<script src="/static/mod3d.js"></script>
+<script src="/static/pdf.js"></script>
+<script src="/static/flipbook.js"></script>
+<script src="/static/flipbook.book3.js"></script>
+<script src="/static/flipbook.scroll.js"></script>
+<script src="/static/flipbook.swipe.js"></script>
+<script src="/static/flipbook.webgl.js"></script>
+<style>
+body{margin:0;background:#f0f0f0;font-family:sans-serif}
+header{max-width:960px;margin:0 auto;padding:18px 20px;display:flex;align-items:center}
+#homeBtn{display:none;width:38px;height:38px;border:none;border-radius:50%;cursor:pointer;
+         background:#0077c2;color:#fff;font-size:20px;margin-right:12px}
+#homeBtn:hover{background:#005999}
+h2{margin:0;font-size:1.5rem;font-weight:600}
+#home,#viewerPage{max-width:960px;margin:0 auto;padding:0 20px 40px}
+.grid{display:grid;grid-template-columns:repeat(auto-fill,180px);gap:16px;margin-top:24px}
+.card{
+  background:#fff url('/static/book2.jpg') no-repeat center center;
+  background-size: 169%;  /* 배경 이미지를 현재보다 30% 더 키움 (130% * 1.3 = 169%) */
+  border:1px solid #ccc;
+  border-radius:6px;
+  cursor:pointer;
+  box-shadow:0 2px 4px rgba(0,0,0,.12);
+  width: 180px;
+  height: 240px;
+  position: relative;
+  display: flex;
+  flex-direction: column;
+  align-items: center;
+  justify-content: center;
+}
+.card img{
+  width:55%;  /* 썸네일 크기 줄임 (배경이 커져서 비율 유지) */
+  height:auto;
+  object-fit:contain;
+  position:absolute;  /* 절대 위치로 변경 */
+  top:50%;  /* 상단에서 50% */
+  left:50%;  /* 좌측에서 50% */
+  transform: translate(-50%, -70%);  /* 정중앙에서 20% 위로 이동 */
+  border: 1px solid #ddd;
+  box-shadow: 0 2px 5px rgba(0,0,0,0.2);
+}
+.card p{
+  text-align:center;
+  margin:6px 0;
+  position: absolute;
+  bottom: 10px;
+  left: 50%;
+  transform: translateX(-50%);
+  background: rgba(255, 255, 255, 0.7);
+  padding: 4px 8px;
+  border-radius: 4px;
+  width: 85%;
+  white-space: nowrap;
+  overflow: hidden;
+  text-overflow: ellipsis;
+  max-width: 150px;
+  font-size: 11px; /* 기본 16px에서 약 30% 감소 */
+}
+button.upload{all:unset;cursor:pointer;border:1px solid #bbb;padding:8px 14px;border-radius:6px;background:#fff;margin:0 8px}
+#viewer{
+  width:80%;
+  height:80vh; /* 전체 화면 높이의 80%로 축소 */
+  max-width:80%; /* 최대 너비 80%로 제한 */
+  margin:0;
+  background:#fff;
+  border:1px solid #ddd;
+  border-radius:8px;
+  position:fixed; /* 고정 위치 */
+  top:50%;
+  left:50%;
+  transform:translate(-50%, -50%); /* 중앙 정렬 */
+  z-index:1000; /* 최상위 표시 */
+  box-shadow:0 4px 20px rgba(0,0,0,0.15);
+}
+</style></head><body>
+<header>
+  <button id="homeBtn" title="홈으로">⌂</button>
+  <h2>My FlipBook Projects</h2>
+</header>
+<section id="home">
+  <div>
+    <label class="upload">📷 이미지 <input id="imgInput" type="file" accept="image/*" multiple hidden></label>
+    <label class="upload">📄 PDF <input id="pdfInput" type="file" accept="application/pdf" hidden></label>
+  </div>
+  <div class="grid" id="grid"></div>
+</section>
+<section id="viewerPage" style="display:none">
+  <div id="viewer"></div>
+</section>
+<script>
+let projects=[], fb=null;
+const grid=$id('grid'), viewer=$id('viewer');
+pdfjsLib.GlobalWorkerOptions.workerSrc='/static/pdf.worker.js';
+// 서버에서 미리 로드된 PDF 프로젝트
+let serverProjects = [];
+/* 🔊 오디오 unlock – 내장 Audio 와 같은 MP3 경로 사용 */
+['click','touchstart'].forEach(evt=>{
+  document.addEventListener(evt,function u(){new Audio('static/turnPage2.mp3')
+        .play().then(a=>a.pause()).catch(()=>{});document.removeEventListener(evt,u,{capture:true});},
+        {once:true,capture:true});
+});
+/* ── 유틸 ── */
+function $id(id){return document.getElementById(id)}
+function addCard(i,thumb,title){
+  const d=document.createElement('div');
+  d.className='card';
+  d.onclick=()=>open(i);
+  // 제목 10글자 제한 및 말줄임표 처리
+  const displayTitle = title ?
+    (title.length > 10 ? title.substring(0, 10) + '...' : title) :
+    '프로젝트 ' + (i+1);
+  d.innerHTML=`<img src="${thumb}"><p title="${title || '프로젝트 ' + (i+1)}">${displayTitle}</p>`;
+  grid.appendChild(d);
+}
+/* ── 이미지 업로드 ── */
+$id('imgInput').onchange=e=>{
+  const files=[...e.target.files]; if(!files.length) return;
+  const pages=[],tot=files.length;let done=0;
+  files.forEach((f,i)=>{const r=new FileReader();r.onload=x=>{pages[i]={src:x.target.result,thumb:x.target.result};
+      if(++done===tot) save(pages);};r.readAsDataURL(f);});
+};
+/* ── PDF 업로드 ── */
+$id('pdfInput').onchange=e=>{
+  const file=e.target.files[0]; if(!file) return;
+  const fr=new FileReader();
+  fr.onload=v=>{
+    pdfjsLib.getDocument({data:v.target.result}).promise.then(async pdf=>{
+      const pages=[];
+      for(let p=1;p<=pdf.numPages;p++){
+        const pg=await pdf.getPage(p), vp=pg.getViewport({scale:1});
+        const c=document.createElement('canvas');c.width=vp.width;c.height=vp.height;
+        await pg.render({canvasContext:c.getContext('2d'),viewport:vp}).promise;
+        pages.push({src:c.toDataURL(),thumb:c.toDataURL()});
+      }
+      save(pages, file.name.replace('.pdf', ''));
+    });
+  };fr.readAsArrayBuffer(file);
+};
+/* ── 프로젝트 저장 ── */
+function save(pages, title){
+  const id=projects.push(pages)-1;
+  addCard(id,pages[0].thumb, title);
+}
+/* ── 서버 PDF 로드 ── */
+async function loadServerPDFs() {
+  try {
+    const response = await fetch('/api/pdf-projects');
+    serverProjects = await response.json();
+    // 서버 PDF 로드 및 썸네일 생성
+    for(let i = 0; i < serverProjects.length; i++) {
+      const project = serverProjects[i];
+      const response = await fetch(`/api/pdf-thumbnail?path=${encodeURIComponent(project.path)}`);
+      const data = await response.json();
+      if(data.thumbnail) {
+        const pages = [{
+          src: data.thumbnail,
+          thumb: data.thumbnail,
+          path: project.path
+        }];
+        save(pages, project.name);
+      }
+    }
+  } catch(error) {
+    console.error('서버 PDF 로드 실패:', error);
+  }
+}
+/* ── 카드 → FlipBook ── */
+function open(i){
+  toggle(false);
+  const pages = projects[i];
+  // 로컬 프로젝트 또는 서버 PDF 로드
+  if(fb){fb.destroy();viewer.innerHTML='';}
+  if(pages[0].path) {
+    // 로딩 표시
+    viewer.innerHTML = '<div style="position:absolute;top:50%;left:50%;transform:translate(-50%,-50%);text-align:center;"><div style="border:4px solid #f3f3f3;border-top:4px solid #3498db;border-radius:50%;width:50px;height:50px;margin:0 auto;animation:spin 2s linear infinite;"></div><p style="margin-top:20px;font-size:16px;">PDF 로딩 중...</p></div>';
+    // 스타일 추가
+    if (!document.getElementById('loadingStyle')) {
+      const style = document.createElement('style');
+      style.id = 'loadingStyle';
+      style.textContent = '@keyframes spin{0%{transform:rotate(0deg)}100%{transform:rotate(360deg)}}';
+      document.head.appendChild(style);
+    }
+    // 서버 PDF 파일 로드
+    fetch(`/api/pdf-content?path=${encodeURIComponent(pages[0].path)}`)
+      .then(response => {
+        if (!response.ok) {
+          throw new Error('PDF 로드 실패: ' + response.statusText);
+        }
+        return response.arrayBuffer();
+      })
+      .then(pdfData => {
+        // PDF 데이터 로드 확인 로깅
+        console.log('PDF 데이터 로드 완료:', pdfData.byteLength + ' 바이트');
+        return pdfjsLib.getDocument({data: pdfData}).promise;
+      })
+      .then(async pdf => {
+        console.log('PDF 문서 로드 완료. 페이지 수:', pdf.numPages);
+        const pdfPages = [];
+        const progressElement = viewer.querySelector('p');
+        for(let p = 1; p <= pdf.numPages; p++) {
+          if (progressElement) {
+            progressElement.textContent = `PDF 페이지 로딩 중... (${p}/${pdf.numPages})`;
+          }
+          try {
+            const pg = await pdf.getPage(p);
+            const vp = pg.getViewport({scale: 1});
+            const c = document.createElement('canvas');
+            c.width = vp.width;
+            c.height = vp.height;
+            await pg.render({canvasContext: c.getContext('2d'), viewport: vp}).promise;
+            pdfPages.push({src: c.toDataURL(), thumb: c.toDataURL()});
+          } catch (pageError) {
+            console.error(`페이지 ${p} 렌더링 오류:`, pageError);
+          }
+        }
+        console.log('모든 페이지 렌더링 완료:', pdfPages.length);
+        if (pdfPages.length > 0) {
+          createFlipBook(pdfPages);
+        } else {
+          throw new Error('PDF에서 페이지를 추출할 수 없습니다.');
+        }
+      })
+      .catch(error => {
+        console.error('PDF 처리 중 오류 발생:', error);
+        viewer.innerHTML = `<div style="position:absolute;top:50%;left:50%;transform:translate(-50%,-50%);text-align:center;"><p style="color:red;font-size:16px;">PDF를 로드하는 중 오류가 발생했습니다:<br>${error.message}</p><button id="backBtn" style="margin-top:20px;padding:10px 20px;background:#0077c2;color:white;border:none;border-radius:4px;cursor:pointer;">홈으로 돌아가기</button></div>`;
+        document.getElementById('backBtn').addEventListener('click', function() {
+          toggle(true);
+        });
+      });
+  } else {
+    // 업로드된 프로젝트 보기
+    console.log('로컬 업로드된 프로젝트 렌더링:', pages.length + '페이지');
+    createFlipBook(pages);
+  }
+}
+function createFlipBook(pages) {
+  console.log('FlipBook 생성 시작. 페이지 수:', pages.length);
+  try {
+    fb = new FlipBook(viewer, {
+      pages: pages,
+      viewMode: 'webgl',
+      autoSize: true,
+      flipDuration: 800,
+      backgroundColor: '#fff',
+      /* 🔊 내장 사운드 */
+      sound: true,
+      assets: {flipMp3: 'static/turnPage2.mp3', hardFlipMp3: 'static/turnPage2.mp3'},
+      controlsProps: {enableFullscreen: true, thumbnails: true}
+    });
+    console.log('FlipBook 생성 완료');
+  } catch (error) {
+    console.error('FlipBook 생성 중 오류 발생:', error);
+    alert('FlipBook을 생성하는 중 오류가 발생했습니다: ' + error.message);
+  }
+}
+/* ── 네비게이션 ── */
+$id('homeBtn').onclick=()=>{
+  if(fb) {
+    fb.destroy();
+    viewer.innerHTML = '';
+    fb = null;
+  }
+  toggle(true);
+};
+function toggle(showHome){
+  $id('home').style.display=showHome?'block':'none';
+  $id('viewerPage').style.display=showHome?'none':'block';
+  $id('homeBtn').style.display=showHome?'none':'inline-block';
+  // 추가: 전체 화면 모드에서 homeBtn 위치 조정
+  if(!showHome) {
+    $id('homeBtn').style.position = 'fixed';
+    $id('homeBtn').style.top = '20px';
+    $id('homeBtn').style.left = '20px';
+    $id('homeBtn').style.zIndex = '1001';  // 뷰어보다 높은 z-index
+    $id('homeBtn').style.fontSize = '24px'; // 크기 증가
+    $id('homeBtn').style.width = '48px';
+    $id('homeBtn').style.height = '48px';
+    // 배경 오버레이 추가
+    document.body.style.backgroundColor = '#3a3a3a';
+  } else {
+    $id('homeBtn').style.position = '';
+    $id('homeBtn').style.top = '';
+    $id('homeBtn').style.left = '';
+    $id('homeBtn').style.zIndex = '';
+    $id('homeBtn').style.fontSize = '';
+    $id('homeBtn').style.width = '';
+    $id('homeBtn').style.height = '';
+    // 배경 원래대로
+    document.body.style.backgroundColor = '#f0f0f0';
+  }
+}
+// 페이지 로드 시 서버 PDF 로드
+window.addEventListener('DOMContentLoaded', loadServerPDFs);
+</script>
+</body></html>
+"""
+# API 엔드포인트: PDF 프로젝트 목록
+@app.get("/api/pdf-projects")
+async def get_pdf_projects():
+    return generate_pdf_projects()
+# API 엔드포인트: PDF 썸네일 생성
+@app.get("/api/pdf-thumbnail")
+async def get_pdf_thumbnail(path: str):
+    try:
+        import fitz  # PyMuPDF
+        # PDF 파일 열기
+        doc = fitz.open(path)
+        # 첫 페이지 가져오기
+        if doc.page_count > 0:
+            page = doc[0]
+            # 썸네일용 이미지 렌더링 (해상도 조정)
+            pix = page.get_pixmap(matrix=fitz.Matrix(0.5, 0.5))
+            img_data = pix.tobytes("png")
+            # Base64 인코딩
+            b64_img = base64.b64encode(img_data).decode('utf-8')
+            return {"thumbnail": f"data:image/png;base64,{b64_img}"}
+        return {"thumbnail": None}
+    except Exception as e:
+        return {"error": str(e), "thumbnail": None}
+@app.get("/api/pdf-content")
+async def get_pdf_content(path: str):
+    try:
+        # 파일 존재 여부 확인
+        pdf_path = pathlib.Path(path)
+        if not pdf_path.exists():
+            return {"error": f"파일을 찾을 수 없습니다: {path}"}, 404
+        # 파일 읽기
+        with open(path, "rb") as pdf_file:
+            content = pdf_file.read()
+        # 파일명 처리 - URL 인코딩으로 한글 등 특수 문자 처리
+        import urllib.parse
+        filename = pdf_path.name
+        encoded_filename = urllib.parse.quote(filename)
+        # 응답 헤더 설정 - RFC 6266 표준 사용
+        headers = {
+            "Content-Type": "application/pdf",
+            "Content-Disposition": f"inline; filename=\"{encoded_filename}\"; filename*=UTF-8''{encoded_filename}"
+        }
+        # 파일 콘텐츠 직접 반환 (dict가 아닌 Response 객체)
+        from fastapi.responses import Response
+        return Response(content=content, media_type="application/pdf")
+    except Exception as e:
+        import traceback
+        error_details = traceback.format_exc()
+        print(f"PDF 콘텐츠 로드 오류: {str(e)}\n{error_details}")
+        # 오류 응답 반환 (JSON 형식)
+        from fastapi.responses import JSONResponse
+        return JSONResponse(content={"error": str(e)}, status_code=500)
+@app.get("/", response_class=HTMLResponse)
+async def root():
+    return HTML
+if __name__ == "__main__":
+    uvicorn.run("app:app", host="0.0.0.0", port=int(os.getenv("PORT", 7860)))