Update main.py

main.py

@@ -5,6 +5,8 @@ import os
 import sys
 import logging
 import tempfile
+import traceback
+import time
 from fastapi import FastAPI, Request, HTTPException, Query, Body
 from fastapi.middleware.cors import CORSMiddleware
 from fastapi.responses import JSONResponse
@@ -13,9 +15,6 @@ import json
 import base64
 from io import BytesIO
 from PIL import Image
-import time
-import traceback
-from db_connector import fetch_found_items, count_found_items
 
 # 캐시 디렉토리 설정 및 최적화
 CACHE_DIRS = {
@@ -25,8 +24,17 @@ CACHE_DIRS = {
     'UPLOADS_DIR': '/tmp/uploads'
 }
 
-# 데이터베이스 관련 환경 변수 (기본값 설정)
+# 환경변수 설정
+for key, path in CACHE_DIRS.items():
+    os.environ[key] = path
+    os.makedirs(path, exist_ok=True)
+
+# 추가 환경변수 최적화
+os.environ['HF_HUB_DISABLE_TELEMETRY'] = '1'
+os.environ['TRANSFORMERS_VERBOSITY'] = 'error'
 
+# 데이터베이스 관련 환경 변수 (기본값 설정)
+# 실제 환경에서는 .env 파일이나 환경 변수로 설정해야 함
 os.environ.setdefault('DB_HOST', 'localhost')
 os.environ.setdefault('DB_PORT', '3306')
 os.environ.setdefault('DB_USER', 'username')  # 실제 사용시 변경 필요
@@ -36,48 +44,6 @@ os.environ.setdefault('DB_NAME', 'foundlost')
 # 애플리케이션 환경 설정 (development, production, test)
 os.environ.setdefault('APP_ENV', 'development')
 
-# 애플리케이션 시작 이벤트 수정
-@app.on_event("startup")
-async def startup_event():
-    """
-    애플리케이션 시작 시 실행되는 이벤트 핸들러
-    """
-    logger.info("애플리케이션 시작 중...")
-    try:
-        # 모델 사전 다운로드 (비동기적으로)
-        from models.clip_model import preload_clip_model
-        preload_clip_model()
-        logger.info("모델 사전 다운로드 완료")
-        
-        # 데이터베이스 연결 테스트
-        if os.getenv('APP_ENV') != 'test':
-            try:
-                from db_connector import get_db_connection
-                with get_db_connection() as connection:
-                    with connection.cursor() as cursor:
-                        cursor.execute("SELECT 1")
-                        result = cursor.fetchone()
-                        if result:
-                            logger.info("✅ 데이터베이스 연결 테스트 성공")
-                        else:
-                            logger.warning("⚠️ 데이터베이스 연결 테스트 결과 없음")
-            except Exception as db_error:
-                logger.error(f"❌ 데이터베이스 연결 테스트 실패: {str(db_error)}")
-                logger.error(traceback.format_exc())
-                
-    except Exception as e:
-        logger.error(f"시작 중 오류 발생: {str(e)}")
-        logger.error(traceback.format_exc())
-
-# 환경변수 설정
-for key, path in CACHE_DIRS.items():
-    os.environ[key] = path
-    os.makedirs(path, exist_ok=True)
-
-# 추가 환경변수 최적화
-os.environ['HF_HUB_DISABLE_TELEMETRY'] = '1'
-os.environ['TRANSFORMERS_VERBOSITY'] = 'error'
-
 # 로깅 설정 개선
 logging.basicConfig(
     level=logging.INFO,
@@ -166,53 +132,6 @@ def get_clip_model(force_reload=False):
             # 실패 시 None 반환
             return None
     return clip_model
-    
-# FastAPI 애플리케이션 생성
-app = FastAPI(
-    title="습득물 유사도 검색 API",
-    description="한국어 CLIP 모델을 사용하여 사용자 게시글과 습득물 간의 유사도를 계산하는 API",
-    version="1.0.0"
-)
-
-# CORS 미들웨어 설정
-app.add_middleware(
-    CORSMiddleware,
-    allow_origins=["*"],  
-    allow_credentials=True,
-    allow_methods=["*"],
-    allow_headers=["*"],
-)
-
-# 애플리케이션 시작 이벤트
-@app.on_event("startup")
-async def startup_event():
-    """
-    애플리케이션 시작 시 실행되는 이벤트 핸들러
-    """
-    logger.info("애플리케이션 시작 중...")
-    try:
-        # 모델 사전 다운로드 (비동기적으로)
-        from models.clip_model import preload_clip_model
-        preload_clip_model()
-        logger.info("모델 사전 다운로드 완료")
-    except Exception as e:
-        logger.error(f"시작 중 오류 발생: {str(e)}")
-        logger.error(traceback.format_exc())
-
-# 전역 예외 처리
-@app.exception_handler(Exception)
-async def global_exception_handler(request: Request, exc: Exception):
-    """
-    전역 예외 처리기
-    """
-    logger.error(f"요청 처리 중 예외 발생: {str(exc)}")
-    return JSONResponse(
-        status_code=500,
-        content={"success": False, "message": f"서버 오류가 발생했습니다: {str(exc)}"}
-    )
-
-# 유틸리티 모듈 임포트
-from utils.similarity import calculate_similarity, find_similar_items, CATEGORY_WEIGHT, ITEM_NAME_WEIGHT, COLOR_WEIGHT, CONTENT_WEIGHT
 
 # 내부적으로 습득물 목록을 가져오는 함수
 async def fetch_found_items(limit=100, offset=0):
@@ -272,6 +191,73 @@ async def fetch_found_items(limit=100, offset=0):
         
         # 오류 발생 시 빈 목록 반환
         return []
+    
+# FastAPI 애플리케이션 생성
+app = FastAPI(
+    title="습득물 유사도 검색 API",
+    description="한국어 CLIP 모델을 사용하여 사용자 게시글과 습득물 간의 유사도를 계산하는 API",
+    version="1.0.0"
+)
+
+# CORS 미들웨어 설정
+app.add_middleware(
+    CORSMiddleware,
+    allow_origins=["*"],  
+    allow_credentials=True,
+    allow_methods=["*"],
+    allow_headers=["*"],
+)
+
+# 애플리케이션 시작 이벤트
+@app.on_event("startup")
+async def startup_event():
+    """
+    애플리케이션 시작 시 실행되는 이벤트 핸들러
+    """
+    logger.info("애플리케이션 시작 중...")
+    try:
+        # 모델 사전 다운로드 (비동기적으로)
+        from models.clip_model import preload_clip_model
+        preload_clip_model()
+        logger.info("모델 사전 다운로드 완료")
+        
+        # 데이터베이스 연결 테스트
+        if os.getenv('APP_ENV') != 'test':
+            try:
+                from db_connector import get_db_connection
+                with get_db_connection() as connection:
+                    with connection.cursor() as cursor:
+                        cursor.execute("SELECT 1")
+                        result = cursor.fetchone()
+                        if result:
+                            logger.info("✅ 데이터베이스 연결 테스트 성공")
+                        else:
+                            logger.warning("⚠️ 데이터베이스 연결 테스트 결과 없음")
+            except Exception as db_error:
+                logger.error(f"❌ 데이터베이스 연결 테스트 실패: {str(db_error)}")
+                logger.error(traceback.format_exc())
+                
+    except Exception as e:
+        logger.error(f"시작 중 오류 발생: {str(e)}")
+        logger.error(traceback.format_exc())
+
+# 전역 예외 처리
+@app.exception_handler(Exception)
+async def global_exception_handler(request: Request, exc: Exception):
+    """
+    전역 예외 처리기
+    """
+    logger.error(f"요청 처리 중 예외 발생: {str(exc)}")
+    return JSONResponse(
+        status_code=500,
+        content={"success": False, "message": f"서버 오류가 발생했습니다: {str(exc)}"}
+    )
+
+# 유틸리티 모듈 임포트
+from utils.similarity import calculate_similarity, find_similar_items, CATEGORY_WEIGHT, ITEM_NAME_WEIGHT, COLOR_WEIGHT, CONTENT_WEIGHT
+
+# 총 데이터 개수 조회 함수 임포트
+from db_connector import count_found_items
 
 # API 엔드포인트 정의 - Spring Boot에 맞게 수정
 @app.post("/api/matching/find-similar", response_model=MatchingResponse)
@@ -463,7 +449,7 @@ async def find_similar_items_api(
         }
         
         return JSONResponse(status_code=500, content=error_response)
-        
+
 @app.get("/api/matching/test")
 async def test_endpoint():
     """