Upload rag.py with huggingface_hub

rag.py

@@ -6,7 +6,6 @@ RAG (Retrieval-Augmented Generation) 모듈
 """
 
 from sentence_transformers import SentenceTransformer
-from openai import OpenAI
 from config import OPENAI_API_KEY, OPENAI_MODEL, EMBEDDING_MODEL
 from database import search_documents
 from typing import List, Dict
@@ -14,8 +13,17 @@ from typing import List, Dict
 # 임베딩 모델 로드 (한국어 지원)
 embedding_model = SentenceTransformer(EMBEDDING_MODEL)
 
-# OpenAI 클라이언트
-client = OpenAI(api_key=OPENAI_API_KEY) if OPENAI_API_KEY else None
+# OpenAI 클라이언트 (lazy loading으로 gradio 충돌 방지)
+_openai_client = None
+
+
+def get_openai_client():
+    """OpenAI 클라이언트 lazy loading"""
+    global _openai_client
+    if _openai_client is None and OPENAI_API_KEY:
+        from openai import OpenAI
+        _openai_client = OpenAI(api_key=OPENAI_API_KEY)
+    return _openai_client
 
 
 def get_embedding(text: str) -> List[float]:
@@ -40,6 +48,8 @@ def retrieve_context(query: str, top_k: int = 3) -> str:
 
 def generate_response(query: str, context: str = "", history: List[Dict] = None) -> str:
     """OpenAI를 사용하여 응답 생성"""
+    client = get_openai_client()
+
     if not client:
         return "OpenAI API 연결이 필요합니다. API 키를 확인해주세요."