Update app.py

app.py

@@ -11,60 +11,80 @@ from langchain.embeddings import HuggingFaceEmbeddings
 GROQ_API_KEY = os.environ.get("GROQ_API_KEY")
 
 # 전역 변수들
-vectorstore = None
+vectorstores = {}
 embeddings = None
 
-import os
-
-def load_vectorstore():
-    """벡터스토어를 로드하는 함수"""
-    global vectorstore, embeddings
+def find_vectorstore_folders():
+    """현재 디렉토리에서 벡터스토어 폴더들을 찾는 함수"""
+    current_dir = os.getcwd()
+    print(f"현재 디렉토리: {current_dir}")
     
-    if vectorstore is None:
-        try:
-            embeddings = HuggingFaceEmbeddings(
-                model_name="sentence-transformers/all-MiniLM-L6-v2"
-            )
-            
-            # 절대 경로로 확인
-            vectorstore_path = os.path.join(os.getcwd(), "vectorstore")
-            print(f"벡터스토어 경로: {vectorstore_path}")
-            
-            # 필수 파일들 확인
-            required_files = ["index.faiss", "index.pkl"]
-            missing_files = []
-            
-            if os.path.exists(vectorstore_path):
-                existing_files = os.listdir(vectorstore_path)
-                print(f"존재하는 파일들: {existing_files}")
+    # 모든 파일과 폴더 확인
+    all_items = os.listdir(current_dir)
+    print(f"현재 디렉토리 내 모든 항목들: {all_items}")
+    
+    # 예상되는 벡터스토어 폴더들
+    expected_folders = ['vectorstore1', 'vectorstore2', 'vectorstore3']
+    vectorstore_folders = []
+    
+    for folder_name in expected_folders:
+        folder_path = os.path.join(current_dir, folder_name)
+        if os.path.exists(folder_path) and os.path.isdir(folder_path):
+            try:
+                folder_contents = os.listdir(folder_path)
+                print(f"📁 {folder_name} 폴더 내용: {folder_contents}")
                 
-                for file in required_files:
-                    if file not in existing_files:
-                        missing_files.append(file)
+                # 필수 파일들 확인
+                required_files = ['index.faiss', 'index.pkl']
+                has_all_files = all(file in folder_contents for file in required_files)
                 
-                if missing_files:
-                    print(f"❌ 누락된 파일들: {missing_files}")
-                    return None
-            else:
-                print("❌ vectorstore 폴더가 존재하지 않습니다")
-                return None
-            
-            vectorstore = FAISS.load_local(
-                vectorstore_path, 
-                embeddings, 
-                allow_dangerous_deserialization=True
-            )
-            print("✅ 벡터스토어 로드 완료")
-            
-        except Exception as e:
-            print(f"❌ 벡터스토어 로드 실패: {e}")
-            import traceback
-            traceback.print_exc()
-            vectorstore = None
+                if has_all_files:
+                    vectorstore_folders.append(folder_name)
+                    print(f"✅ {folder_name} - 모든 필수 파일 존재")
+                else:
+                    missing_files = [f for f in required_files if f not in folder_contents]
+                    print(f"❌ {folder_name} - 누락된 파일: {missing_files}")
+                    
+            except Exception as e:
+                print(f"❌ {folder_name} 폴더 확인 중 오류: {e}")
+        else:
+            print(f"❌ {folder_name} 폴더가 존재하지 않음")
     
-    return vectorstore
+    if not vectorstore_folders:
+        print("❌ 사용 가능한 벡터스토어 폴더를 찾을 수 없습니다")
+    else:
+        print(f"✅ 총 {len(vectorstore_folders)}개의 벡터스토어 폴더를 찾았습니다: {vectorstore_folders}")
+    
+    return vectorstore_folders
 
-# 질문 리스트
+def load_all_vectorstores():
+    """모든 벡터스토어를 로드하는 함수"""
+    global vectorstores, embeddings
+    
+    if not embeddings:
+        embeddings = HuggingFaceEmbeddings(
+            model_name="sentence-transformers/all-MiniLM-L6-v2"
+        )
+    
+    # 벡터스토어 폴더들 찾기
+    folders = find_vectorstore_folders()
+    
+    for folder_name in folders:
+        if folder_name not in vectorstores:
+            try:
+                vectorstore = FAISS.load_local(
+                    f"./{folder_name}", 
+                    embeddings, 
+                    allow_dangerous_deserialization=True
+                )
+                vectorstores[folder_name] = vectorstore
+                print(f"✅ {folder_name} 로드 완료")
+            except Exception as e:
+                print(f"❌ {folder_name} 로드 실패: {e}")
+    
+    return list(vectorstores.keys())
+
+# 질문 리스트 (기존과 동일)
 suggested_questions = [
     '교원 신규 임용은 어떻게 하나요?',
     '교원 연구년 기간은 어떻게 되나요?',
@@ -88,24 +108,19 @@ suggested_questions = [
     '안전사고예방계획은 어디에 제출해야 하나요?'
 ]
 
-# 프롬프트 템플릿
+# 프롬프트 템플릿 (기존과 동일)
 prompt_template = """당신은 한남대학교 규정집 도우미입니다.
 반드시 한국어로만 답변해주세요. 영어나 다른 언어는 절대 사용하지 마세요.
-
 주어진 문서 내용을 바탕으로 질문에 대해 정확하고 친절하게 한국어로 답변해주세요.
-
 참고 문서:
 {context}
-
 질문: {question}
-
 답변 지침:
 - 이용자를 반기는 인사로 시작하세요
 - 반드시 한국어로만 답변하세요
 - 정중하고 친근한 말투를 사용하세요
 - 구체적이고 도움이 되는 정보를 제공하세요
 - 문서에서 답을 찾을 수 없으면 "죄송하지만 해당 정보를 규정집에서 찾을 수 없습니다"라고 답변하세요
-
 한국어 답변:"""
 
 prompt = PromptTemplate(
@@ -113,7 +128,7 @@ prompt = PromptTemplate(
     input_variables=["context", "question"]
 )
 
-def respond_with_groq(question, selected_q, model):
+def respond_with_groq(question, selected_q, model, selected_vectorstore):
     """질문에 대한 답변을 생성하는 함수"""
     
     # 선택된 질문이 있으면 그것을 사용
@@ -126,11 +141,18 @@ def respond_with_groq(question, selected_q, model):
     if not GROQ_API_KEY:
         return "❌ API 키가 설정되지 않았습니다. 관리자에게 문의하세요."
 
+    # 벡터스토어가 로드되지 않은 경우
+    if not vectorstores:
+        return "❌ 사용 가능한 벡터스토어가 없습니다. 관리자에게 문의하세요."
+
     try:
-        # 벡터스토어 로드
-        current_vectorstore = load_vectorstore()
-        if not current_vectorstore:
-            return "❌ 벡터스토어를 로드할 수 없습니다."
+        # 선택된 벡터스토어 확인
+        if selected_vectorstore not in vectorstores:
+            available_stores = list(vectorstores.keys())
+            return f"❌ 선택된 벡터스토어({selected_vectorstore})를 찾을 수 없습니다. 사용 가능한 스토어: {available_stores}"
+
+        current_vectorstore = vectorstores[selected_vectorstore]
+        print(f"✅ 사용 중인 벡터스토어: {selected_vectorstore}")
 
         # LLM 설정
         llm = ChatGroq(
@@ -154,6 +176,9 @@ def respond_with_groq(question, selected_q, model):
         return result['result']
 
     except Exception as e:
+        import traceback
+        error_details = traceback.format_exc()
+        print(f"❌ 상세 오류 정보:\n{error_details}")
         return f"❌ 오류가 발생했습니다: {str(e)}"
 
 def update_question(selected):
@@ -162,6 +187,9 @@ def update_question(selected):
         return selected
     return ""
 
+# 앱 시작시 벡터스토어들 로드
+available_vectorstores = load_all_vectorstores()
+
 # Gradio 인터페이스 생성
 with gr.Blocks(title="한남대학교 Q&A") as interface:
     gr.HTML("""
@@ -173,6 +201,28 @@ with gr.Blocks(title="한남대학교 Q&A") as interface:
 
     with gr.Row():
         with gr.Column(scale=1):
+            # 벡터스토어 선택 드롭다운 추가
+            if available_vectorstores:
+                # 사용자 친화적인 이름으로 매핑
+                vectorstore_display_names = {
+                    'vectorstore1': '📚 데이터베이스 1',
+                    'vectorstore2': '📚 데이터베이스 2', 
+                    'vectorstore3': '📚 데이터베이스 3'
+                }
+                display_choices = [vectorstore_display_names.get(vs, vs) for vs in available_vectorstores]
+                
+                vectorstore_dropdown = gr.Dropdown(
+                    choices=list(zip(display_choices, available_vectorstores)),
+                    label="📚 데이터베이스 선택",
+                    value=available_vectorstores[0] if available_vectorstores else None
+                )
+            else:
+                vectorstore_dropdown = gr.Dropdown(
+                    choices=[("벡터스토어 없음", "none")],
+                    label="📚 데이터베이스 선택",
+                    value="none"
+                )
+
             question_dropdown = gr.Dropdown(
                 choices=["직접 입력"] + suggested_questions,
                 label="💡 자주 묻는 질문",
@@ -204,7 +254,7 @@ with gr.Blocks(title="한남대학교 Q&A") as interface:
     # 이벤트 연결
     submit_btn.click(
         fn=respond_with_groq,
-        inputs=[question_input, question_dropdown, model_choice],
+        inputs=[question_input, question_dropdown, model_choice, vectorstore_dropdown],
         outputs=output
     )
 
@@ -216,4 +266,4 @@ with gr.Blocks(title="한남대학교 Q&A") as interface:
 
 # 앱 실행
 if __name__ == "__main__":
-    interface.launch()
+    interface.launch()