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,191 @@ 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
+        
+        # 임시 PDF 파일 생성
+        pdf_buffer = io.BytesIO()
+        doc = SimpleDocTemplate(pdf_buffer, pagesize=letter)
+        styles = getSampleStyleSheet()
+        
+        # 내용을 문단으로 분할
+        content = []
+        
+        # 제목 추가
+        title_style = styles['Title']
+        content.append(Paragraph(title, title_style))
+        content.append(Spacer(1, 12))
+        
+        # 본문 텍스트 스타일
+        normal_style = styles['Normal']
+        
+        # 텍스트를 단락으로 분리하여 추가
+        paragraphs = text_content.split('\n\n')
+        for para in paragraphs:
+            if para.strip():
+                p = Paragraph(para.replace('\n', '<br/>'), 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를 사용하여 텍스트를 더 구조화된 형식으로 변환
+async def enhance_text_with_ai(text_content: str, title: str) -> str:
+    try:
+        # API 키가 없거나 유효하지 않은 경우 원본 텍스트 반환
+        if not HAS_VALID_API_KEY or not openai_client:
+            return text_content
+        
+        # 텍스트가 짧은 경우 원본 반환
+        if len(text_content) < 100:
+            return text_content
+        
+        # 컨텍스트 크기를 고려하여 텍스트가 너무 길면 앞부분만 사용
+        max_context_length = 60000
+        if len(text_content) > max_context_length:
+            text_to_process = text_content[:max_context_length] + "...(이하 생략)"
+        else:
+            text_to_process = text_content
+        
+        # OpenAI API 호출하여 텍스트 구조화
+        try:
+            system_prompt = """
+            당신은 텍스트를 잘 구조화된 전자책 형식으로 변환하는 전문가입니다.
+            제공된 원본 텍스트를 분석하고, 다음과 같이 개선해주세요:
+            
+            1. 논리적인 섹션으로 나누고 적절한 제목과 부제목 추가
+            2. 단락을 자연스럽게 구성하고 가독성 개선
+            3. 중요한 내용은 강조하거나 요약하여 표시
+            4. 원본 내용의 의미와 맥락은 유지하면서 깔끔하게 정리
+            5. 맞춤법과 문법 오류 수정
+            
+            원본 내용을 모두 유지하되, 잘 정리된 전자책처럼 보이도록 구조화해주세요.
+            수정본을 직접 반환하고, 원본 내용을 모두 포함해야 합니다.
+            """
+            
+            response = openai_client.chat.completions.create(
+                model="gpt-4.1-mini",
+                messages=[
+                    {"role": "system", "content": system_prompt},
+                    {"role": "user", "content": f"제목: {title}\n\n원본 텍스트:\n{text_to_process}"}
+                ],
+                temperature=0.7,
+                max_tokens=4000,
+                timeout=60.0
+            )
+            
+            enhanced_text = response.choices[0].message.content
+            return enhanced_text
+            
+        except Exception as api_error:
+            logger.error(f"AI 텍스트 향상 오류: {api_error}")
+            # 오류 발생 시 원본 텍스트 반환
+            return text_content
+            
+    except Exception as e:
+        logger.error(f"AI 텍스트 향상 오류: {e}")
+        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'):
+            text_content = content.decode('utf-8', errors='replace')
+        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 +1155,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 +1179,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 +1551,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 +2313,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 +2642,6 @@ async function submitQuestion(question) {
   }
 }
 
-
-
-
     
     // DOM이 로드되면 실행
     document.addEventListener('DOMContentLoaded', function() {
@@ -2473,6 +2676,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 +2796,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,8 +3893,7 @@ async function submitQuestion(question) {
     }
   </script>
 </body>
-</html>
-"""
+</html>            
 
 if __name__ == "__main__":
     uvicorn.run("app:app", host="0.0.0.0", port=int(os.getenv("PORT", 7860)))