Update app.py

app.py

@@ -9,6 +9,13 @@ import threading
 import concurrent.futures
 from openai import OpenAI
 import fitz  # PyMuPDF
+import tempfile
+from reportlab.lib.pagesizes import letter
+from reportlab.pdfgen import canvas
+from reportlab.platypus import SimpleDocTemplate, Paragraph, Spacer
+from reportlab.lib.styles import getSampleStyleSheet
+import io
+import docx2txt
 
 # 로깅 설정
 logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(name)s - %(levelname)s - %(message)s')
@@ -948,6 +955,199 @@ async def upload_pdf(file: UploadFile = File(...)):
             status_code=500
         )
 
+# 텍스트 파일을 PDF로 변환하는 함수
+async def convert_text_to_pdf(text_content: str, title: str) -> str:
+    try:
+        # 제목에서 유효한 파일명 생성
+        import re
+        safe_title = re.sub(r'[^\w\-_\. ]', '_', title)
+        if not safe_title:
+            safe_title = "aibook"
+        
+        # 타임스탬프 추가로 고유한 파일명 생성
+        timestamp = int(time.time())
+        filename = f"{safe_title}_{timestamp}.pdf"
+        
+        # 영구 저장소의 파일 경로
+        file_path = PERMANENT_PDF_DIR / filename
+        
+        # 한글 폰트 등록 - 업로드된 MaruBuri-SemiBold.ttf 사용
+        from reportlab.pdfbase import pdfmetrics
+        from reportlab.pdfbase.ttfonts import TTFont
+        
+        # 폰트 경로 설정 (app.py와 같은 디렉토리에 있는 폰트 사용)
+        font_path = BASE / "MaruBuri-SemiBold.ttf"
+        
+        # 폰트 등록
+        font_name = "MaruBuri"
+        if font_path.exists():
+            pdfmetrics.registerFont(TTFont(font_name, str(font_path)))
+            logger.info(f"한글 폰트 등록 성공: {font_path}")
+        else:
+            font_name = "Helvetica"
+            logger.warning(f"한글 폰트 파일을 찾을 수 없습니다: {font_path}. 기본 폰트를 사용합니다.")
+        
+        # 임시 PDF 파일 생성
+        pdf_buffer = io.BytesIO()
+        
+        # 한글 지원을 위한 스타일 설정
+        from reportlab.lib.pagesizes import letter
+        from reportlab.platypus import SimpleDocTemplate, Paragraph, Spacer
+        from reportlab.lib.styles import getSampleStyleSheet, ParagraphStyle
+        from reportlab.lib.enums import TA_CENTER, TA_LEFT
+        
+        doc = SimpleDocTemplate(pdf_buffer, pagesize=letter, encoding='utf-8')
+        
+        # 사용자 정의 스타일 생성
+        title_style = ParagraphStyle(
+            name='CustomTitle',
+            fontName=font_name,
+            fontSize=18,
+            leading=22,
+            alignment=TA_CENTER,
+            spaceAfter=20
+        )
+        
+        normal_style = ParagraphStyle(
+            name='CustomNormal',
+            fontName=font_name,
+            fontSize=12,
+            leading=15,
+            alignment=TA_LEFT,
+            spaceBefore=6,
+            spaceAfter=6
+        )
+        
+        # 내용을 문단으로 분할
+        content = []
+        
+        # 제목 추가
+        content.append(Paragraph(title, title_style))
+        content.append(Spacer(1, 20))
+        
+        # 텍스트를 단락으로 분리하여 추가
+        paragraphs = text_content.split('\n\n')
+        for para in paragraphs:
+            if para.strip():
+                # XML 특수문자 이스케이프 처리
+                from xml.sax.saxutils import escape
+                safe_para = escape(para.replace('\n', '<br/>'))
+                p = Paragraph(safe_para, normal_style)
+                content.append(p)
+                content.append(Spacer(1, 10))
+        
+        # PDF 생성
+        doc.build(content)
+        
+        # 파일로 저장
+        with open(file_path, 'wb') as f:
+            f.write(pdf_buffer.getvalue())
+        
+        # 메인 디렉토리에도 복사
+        with open(PDF_DIR / filename, 'wb') as f:
+            f.write(pdf_buffer.getvalue())
+        
+        # PDF ID 생성 및 메타데이터 저장
+        pdf_id = generate_pdf_id(filename)
+        pdf_metadata[pdf_id] = str(file_path)
+        save_pdf_metadata()
+        
+        # 백그라운드에서 캐싱 시작
+        asyncio.create_task(cache_pdf(str(file_path)))
+        
+        return {
+            "path": str(file_path),
+            "filename": filename,
+            "id": pdf_id
+        }
+        
+    except Exception as e:
+        logger.error(f"텍스트를 PDF로 변환 중 오류: {e}")
+        raise e
+
+
+# AI를 사용하여 텍스트를 더 구조화된 형식으로 변환 (OpenAI 제거 버전)
+async def enhance_text_with_ai(text_content: str, title: str) -> str:
+    # 원본 텍스트 그대로 반환 (AI 향상 기능 비활성화)
+    return text_content    
+
+
+
+# 텍스트 파일을 PDF로 변환하는 엔드포인트
+@app.post("/api/text-to-pdf")
+async def text_to_pdf(file: UploadFile = File(...)):
+    try:
+        # 지원하는 파일 형식 확인
+        filename = file.filename.lower()
+        if not (filename.endswith('.txt') or filename.endswith('.docx') or filename.endswith('.doc')):
+            return JSONResponse(
+                content={"success": False, "message": "지원하는 파일 형식은 .txt, .docx, .doc입니다."}, 
+                status_code=400
+            )
+        
+        # 파일 내용 읽기
+        content = await file.read()
+        
+        # 파일 타입에 따라 텍스트 추출
+        if filename.endswith('.txt'):
+            # 인코딩 자동 감지 시도
+            encodings = ['utf-8', 'euc-kr', 'cp949', 'latin1']
+            text_content = None
+            
+            for encoding in encodings:
+                try:
+                    text_content = content.decode(encoding, errors='strict')
+                    logger.info(f"텍스트 파일 인코딩 감지: {encoding}")
+                    break
+                except UnicodeDecodeError:
+                    continue
+            
+            if text_content is None:
+                # 모든 인코딩 시도 실패 시 기본적으로 UTF-8로 시도하고 오류는 대체 문자로 처리
+                text_content = content.decode('utf-8', errors='replace')
+                logger.warning("텍스트 파일 인코딩을 감지할 수 없어 UTF-8으로 시도합니다.")
+                
+        elif filename.endswith('.docx') or filename.endswith('.doc'):
+            # 임시 파일로 저장
+            with tempfile.NamedTemporaryFile(delete=False, suffix=os.path.splitext(filename)[1]) as temp_file:
+                temp_file.write(content)
+                temp_path = temp_file.name
+            
+            try:
+                # docx2txt로 텍스트 추출
+                text_content = docx2txt.process(temp_path)
+            finally:
+                # 임시 파일 삭제
+                os.unlink(temp_path)
+        
+        # 파일명에서 제목 추출 (확장자 제외)
+        title = os.path.splitext(filename)[0]
+        
+        # AI로 텍스트 내용 향상
+        enhanced_text = await enhance_text_with_ai(text_content, title)
+        
+        # 텍스트를 PDF로 변환
+        pdf_info = await convert_text_to_pdf(enhanced_text, title)
+        
+        return JSONResponse(
+            content={
+                "success": True, 
+                "path": pdf_info["path"], 
+                "name": os.path.splitext(pdf_info["filename"])[0],
+                "id": pdf_info["id"],
+                "viewUrl": f"/view/{pdf_info['id']}"
+            },
+            status_code=200
+        )
+    except Exception as e:
+        import traceback
+        error_details = traceback.format_exc()
+        logger.error(f"텍스트를 PDF로 변환 중 오류: {str(e)}\n{error_details}")
+        return JSONResponse(
+            content={"success": False, "message": str(e)}, 
+            status_code=500
+        )
+
 # 관리자 인증 엔드포인트
 @app.post("/api/admin-login")
 async def admin_login(password: str = Form(...)):
@@ -963,9 +1163,17 @@ async def delete_pdf(path: str):
         if not pdf_file.exists():
             return {"success": False, "message": "파일을 찾을 수 없습니다"}
         
-        # PDF 파일 삭제
+        # PDF 파일명 가져오기
+        filename = pdf_file.name
+        
+        # PDF 파일 삭제 (영구 저장소에서)
         pdf_file.unlink()
         
+        # 메인 디렉토리에서도 동일한 파일이 있으면 삭제 (버그 수정)
+        main_file_path = PDF_DIR / filename
+        if main_file_path.exists():
+            main_file_path.unlink()
+        
         # 관련 캐시 파일 삭제
         pdf_name = pdf_file.stem
         cache_path = get_cache_path(pdf_name)
@@ -979,7 +1187,7 @@ async def delete_pdf(path: str):
         # 메타데이터에서 해당 파일 ID 제거
         to_remove = []
         for pid, fpath in pdf_metadata.items():
-            if os.path.basename(fpath) == pdf_file.name:
+            if os.path.basename(fpath) == filename:
                 to_remove.append(pid)
         
         for pid in to_remove:
@@ -1351,7 +1559,9 @@ HTML = """
     font-weight: 500;
     display: flex;
     align-items: center;
-    box-shadow: var(--shadow-sm);
+
+
+box-shadow: var(--shadow-sm);
     transition: var(--transition);
     position: relative;
     overflow: hidden;
@@ -2111,7 +2321,11 @@ HTML = """
       <button class="upload" id="pdfUploadBtn">
         <i class="fas fa-file-pdf"></i> PDF Upload
       </button>
+      <button class="upload" id="textToAIBookBtn">
+        <i class="fas fa-file-alt"></i> Text to AI-Book
+      </button>
       <input id="pdfInput" type="file" accept="application/pdf" style="display:none">
+      <input id="textInput" type="file" accept=".txt,.docx,.doc" style="display:none">
     </div>
     
     <div class="section-title">Projects</div>
@@ -2436,9 +2650,6 @@ async function submitQuestion(question) {
   }
 }
 
-
-
-
     
     // DOM이 로드되면 실행
     document.addEventListener('DOMContentLoaded', function() {
@@ -2473,6 +2684,26 @@ async function submitQuestion(question) {
         console.error("PDF 업로드 요소를 찾을 수 없음");
       }
       
+      // 텍스트 업로드 버튼
+      const textBtn = document.getElementById('textToAIBookBtn');
+      const textInput = document.getElementById('textInput');
+      
+      if (textBtn && textInput) {
+        // 버튼 클릭 시 파일 입력 트리거
+        textBtn.addEventListener('click', function() {
+          textInput.click();
+        });
+        
+        // 파일 선택 시 처리
+        textInput.addEventListener('change', function(e) {
+          const file = e.target.files[0];
+          if (!file) return;
+          
+          // 서버에 텍스트 파일 업로드 (영구 저장소에 PDF로 변환하여 저장)
+          uploadTextToServer(file);
+        });
+      }
+      
       // 서버 PDF 로드 및 캐시 상태 확인
       loadServerPDFs();
       
@@ -2573,6 +2804,40 @@ async function submitQuestion(question) {
       }
     }
 
+    // 서버에 텍스트 파일을 업로드하여 PDF로 변환하는 함수
+    async function uploadTextToServer(file) {
+      try {
+        showLoading("텍스트 분석 및 PDF 변환 중...");
+        
+        const formData = new FormData();
+        formData.append('file', file);
+        
+        const response = await fetch('/api/text-to-pdf', {
+          method: 'POST',
+          body: formData
+        });
+        
+        const result = await response.json();
+        
+        if (result.success) {
+          hideLoading();
+          
+          // 업로드 성공 시 서버 PDF 리스트 리로드
+          await loadServerPDFs();
+          
+          // 성공 메시지
+          showMessage("텍스트가 성공적으로 PDF로 변환되었습니다! 공유 URL: " + result.viewUrl);
+        } else {
+          hideLoading();
+          showError("변환 실패: " + (result.message || "알 수 없는 오류"));
+        }
+      } catch (error) {
+        console.error("텍스트 변환 오류:", error);
+        hideLoading();
+        showError("텍스트를 PDF로 변환하는 중 오류가 발생했습니다.");
+      }
+    }
+
     function addCard(i, thumb, title, isCached = false, pdfId = null) {
       const d = document.createElement('div');
       d.className = 'card fade-in';
@@ -3636,7 +3901,7 @@ async function submitQuestion(question) {
     }
   </script>
 </body>
-</html>
+</html>            
 """
 
 if __name__ == "__main__":