Merge branch 'main' into fix/cuda_gpu

Author: BcKmini

.github/workflows/ci.yml

@@ -0,0 +1,41 @@
+name: CI
+
+on:
+  pull_request:
+    branches: [ main ]
+  push:
+    branches: [ main ]
+    paths:
+      - '**/*.py'
+      - 'requirements.txt'
+      - '.github/workflows/ci.yml'
+
+jobs:
+  lint-test:
+    runs-on: ubuntu-latest
+    timeout-minutes: 10
+    steps:
+      - uses: actions/checkout@v4
+
+      - uses: actions/setup-python@v5
+        with:
+          python-version: '3.11'
+          cache: 'pip'
+
+      - name: Install deps
+        run: |
+          python -m pip install --upgrade pip
+          pip install -r requirements.txt
+
+      - name: Syntax check
+        run: |
+          python -m py_compile $(git ls-files '*.py' | tr '\n' ' ')
+
+      - name: Import smoke
+        run: |
+          python - << 'PY'
+          from importlib import import_module
+          import_module('main')
+          print('Import OK')
+          PY
+

.github/workflows/docker.yml

@@ -0,0 +1,35 @@
+name: Docker Build & Push (Backend)
+
+on:
+  push:
+    branches: [ main ]
+    paths:
+      - '**'
+      - '!README.md'
+
+jobs:
+  docker:
+    runs-on: ubuntu-latest
+    permissions:
+      contents: read
+      packages: write
+    steps:
+      - uses: actions/checkout@v4
+
+      - name: Log in to GHCR
+        uses: docker/login-action@v3
+        with:
+          registry: ghcr.io
+          username: ${{ github.actor }}
+          password: ${{ secrets.GITHUB_TOKEN }}
+
+      - name: Build and push
+        uses: docker/build-push-action@v6
+        with:
+          context: .
+          file: ./Dockerfile
+          push: true
+          tags: |
+            ghcr.io/${{ github.repository }}:backend-latest
+            ghcr.io/${{ github.repository }}:backend-${{ github.sha }}
+

.gitignore

@@ -4,4 +4,4 @@ nohup.out
 __pycache__/
 venv
 ./venv
-uploads/*
+uploads/*

Dockerfile

@@ -0,0 +1,20 @@
+FROM python:3.11-slim
+
+ENV PYTHONDONTWRITEBYTECODE=1 \
+    PYTHONUNBUFFERED=1
+
+WORKDIR /app
+
+# System dependencies for pdf2image
+RUN apt-get update && apt-get install -y --no-install-recommends \
+    poppler-utils \
+    && rm -rf /var/lib/apt/lists/*
+
+COPY Backend/requirements.txt ./requirements.txt
+RUN pip install --no-cache-dir -r requirements.txt
+
+COPY Backend/ ./
+
+EXPOSE 8080
+CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "8080"]
+

Makefile

@@ -1,3 +1,3 @@
 # Makefile
 run:
-	uvicorn main:app --host 0.0.0.0 --port 8080 --reload
+	uvicorn main:app --host 0.0.0.0 --port 8080 --reload --http h11

README.md

@@ -0,0 +1,55 @@
+# Noteflow Backend (FastAPI)
+
+## Overview
+- FastAPI backend for Noteflow
+- OCR pipeline supports images, PDF, DOC/DOCX, HWP (via utilities and system tools)
+
+## Run (local)
+```
+python -m venv .venv
+source .venv/bin/activate
+pip install -r requirements.txt
+uvicorn main:app --host 0.0.0.0 --port 8080 --reload
+```
+
+Env (optional):
+- `SECRET_KEY`, `ACCESS_TOKEN_EXPIRE_MINUTES`
+- Database URLs if you connect a DB (current code uses provided models)
+
+## OCR system tools (optional but recommended)
+- PyMuPDF (Python) used by default for PDF text extraction
+- Optional fallbacks/tools:
+  - Poppler (`pdftoppm`) for `pdf2image`
+  - LibreOffice (`soffice`) for .doc → .pdf
+  - `hwp5txt` for .hwp text extraction
+- If missing, the API still returns 200 with `warnings` explaining limitations.
+
+## API Highlights
+- `POST /api/v1/files/ocr` — OCR and create note (accepts file + optional `folder_id`, `langs`, `max_pages`)
+- `POST /api/v1/files/upload` — Upload files to folder
+- `POST /api/v1/files/audio` — STT from audio, create/append to note
+
+## CI (GitHub Actions)
+- This folder includes `.github/workflows/ci.yml` to lint/smoke-test on push/PR.
+- Python 3.11, `pip install -r requirements.txt`, syntax check and import smoke.
+
+## Docker (optional; for later)
+- Dockerfile included. Build & run locally:
+```
+docker build -t noteflow-backend .
+docker run --rm -p 8080:8080 noteflow-backend
+```
+- GitHub Actions container build:
+  - `.github/workflows/docker.yml` pushes to GHCR:
+    - `ghcr.io/<owner>/<repo>:backend-latest`
+    - `ghcr.io/<owner>/<repo>:backend-<sha>`
+- Deployment example (SSH) once you’re ready:
+```
+docker login ghcr.io -u <USER> -p <TOKEN>
+docker pull ghcr.io/<owner>/<repo>:backend-latest
+docker run -d --name backend --restart=always -p 8080:8080 ghcr.io/<owner>/<repo>:backend-latest
+```
+
+## Notes
+- If you split this folder into its own repository root, the included `.github/workflows/*.yml` files will work as-is.
+- OCR uses model-first path (EasyOCR + TrOCR) and falls back to tesseract when available.

db.py

@@ -1,14 +1,19 @@
 from sqlalchemy import create_engine
 from sqlalchemy.orm import sessionmaker
+from models import Base  # ensure models are imported so metadata knows all tables
+import os
 
-DATABASE_URL = "mysql+mysqlconnector://noteflow:NoteFlow123!@localhost/noteflow"
-engine = create_engine(DATABASE_URL)
+DATABASE_URL = os.getenv("DATABASE_URL", "mysql+mysqlconnector://noteflow:NoteFlow123!@localhost/noteflow")
+
+engine = create_engine(DATABASE_URL, pool_pre_ping=True)
 SessionLocal = sessionmaker(autocommit=False, autoflush=False, bind=engine)
 
-# 데이터베이스 의존성
+def init_db():
+    Base.metadata.create_all(bind=engine)
+
 def get_db():
     db = SessionLocal()
     try:
         yield db
     finally:
-        db.close()
+        db.close()

main.py

@@ -1,57 +1,54 @@
-# src/main.py
+# Backend/main.py
 import os
 from dotenv import load_dotenv
 # 환경 변수를 최대한 빨리 로드하여 GPU 설정(CUDA_VISIBLE_DEVICES)이 라우터 임포트 전에 적용되도록 함
 load_dotenv()
 from fastapi import FastAPI
 from fastapi.middleware.cors import CORSMiddleware
+from fastapi.staticfiles import StaticFiles
+
+from db import init_db
 from routers.auth import router as auth_router
 from routers.note import router as note_router
-from routers.folder import router as folder_router 
-from fastapi.staticfiles import StaticFiles
-from routers.file import router as file_router 
-import logging
-import uvicorn  
+from routers.folder import router as folder_router
+from routers.checklist import router as checklist_router
+from routers.file import router as file_router
+
 
 # 1) 환경변수 로드 (상단에서 선 로드됨)
 
-# 2) 로깅 설정
-logging.basicConfig(level=logging.INFO)
-logger = logging.getLogger(__name__)
+import uvicorn
 
-# 3) FastAPI 앱 생성
-app = FastAPI()
 
-# 4) CORS 설정
-origins = [
-    "http://localhost:5174",
-]
+load_dotenv()
+
+app = FastAPI()
 
 app.add_middleware(
     CORSMiddleware,
-    allow_origins=["*"],
+    allow_origins=["*"],  # 개발 중 전체 허용
     allow_credentials=True,
     allow_methods=["*"],
     allow_headers=["*"],
 )
 
-# 5) 라우터 등록
+# 정적 파일(업로드) 서빙
+os.makedirs(os.path.join(os.path.dirname(__file__), "uploads"), exist_ok=True)
+app.mount("/static", StaticFiles(directory=os.path.join(os.path.dirname(__file__), "uploads")), name="static")
+
+# 라우터 등록
 app.include_router(auth_router)
 app.include_router(note_router)
-app.include_router(folder_router)  
+app.include_router(folder_router)
 app.include_router(file_router)
+app.include_router(checklist_router)
 
-# 6) 루트 엔드포인트
 @app.get("/")
-def read_root():
+def root():
     return {"message": "mini"}
 
-# 7) 실행 설정
+# 앱 시작 시(uvicorn main:app) 한 번만 테이블 생성 (개발용)
+init_db()
+
 if __name__ == "__main__":
-    uvicorn.run(
-        "main:app",
-        host="0.0.0.0",
-        port=8080,
-        reload=True,
-        env_file=".env"
-    )
+    uvicorn.run("main:app", host="0.0.0.0", port=8080, reload=True)

models/__init__.py

@@ -1 +1,8 @@
-from .user import User
+from .base import Base
+from .user import User
+from .folder import Folder
+from .note import Note
+from .file import File
+from .checklist import Checklist
+
+__all__ = ["Base", "User", "Folder", "Note", "File", "Checklist"]

models/base.py

@@ -1,3 +1,5 @@
 from sqlalchemy.ext.declarative import declarative_base
+from sqlalchemy.orm import declarative_base
+
 
 Base = declarative_base()