Update app.py

app.py

@@ -12,6 +12,7 @@ GROQ_API_KEY = os.environ.get("GROQ_API_KEY")
 # 전역 변수들
 vectorstores = {}
 embeddings = None
+combined_vectorstore = None
 
 def debug_file_system():
     """파일 시스템 상태를 자세히 확인하는 함수"""
@@ -97,8 +98,8 @@ def find_vectorstore_folders():
     return vectorstore_folders
 
 def load_all_vectorstores():
-    """모든 벡터스토어를 로드하는 함수"""
-    global vectorstores, embeddings
+    """모든 벡터스토어를 로드하고 통합하는 함수"""
+    global vectorstores, embeddings, combined_vectorstore
     
     if not embeddings:
         embeddings = HuggingFaceEmbeddings(
@@ -108,20 +109,33 @@ def load_all_vectorstores():
     # 벡터스토어 폴더들 찾기
     folders = find_vectorstore_folders()
     
+    loaded_vectorstores = []
+    
     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
+            loaded_vectorstores.append(vectorstore)
+            print(f"✅ {folder_name} 로드 완료")
+        except Exception as e:
+            print(f"❌ {folder_name} 로드 실패: {e}")
+    
+    # 벡터스토어들을 통합 (첫 번째를 기본으로 하고 나머지를 merge)
+    if loaded_vectorstores:
+        combined_vectorstore = loaded_vectorstores[0]
+        for vs in loaded_vectorstores[1:]:
             try:
-                vectorstore = FAISS.load_local(
-                    f"./{folder_name}", 
-                    embeddings, 
-                    allow_dangerous_deserialization=True
-                )
-                vectorstores[folder_name] = vectorstore
-                print(f"✅ {folder_name} 로드 완료")
+                combined_vectorstore.merge_from(vs)
+                print(f"✅ 벡터스토어 통합 완료")
             except Exception as e:
-                print(f"❌ {folder_name} 로드 실패: {e}")
+                print(f"❌ 벡터스토어 통합 실패: {e}")
+        print(f"🎉 총 {len(loaded_vectorstores)}개의 벡터스토어가 통합되었습니다")
     
-    return list(vectorstores.keys())
+    return len(loaded_vectorstores) > 0
 
 # 질문 리스트 (기존과 동일)
 suggested_questions = [
@@ -167,7 +181,7 @@ prompt = PromptTemplate(
     input_variables=["context", "question"]
 )
 
-def respond_with_groq(question, selected_q, model, selected_vectorstore):
+def respond_with_groq(question, selected_q, model):
     """질문에 대한 답변을 생성하는 함수"""
     
     # 선택된 질문이 있으면 그것을 사용
@@ -180,18 +194,12 @@ def respond_with_groq(question, selected_q, model, selected_vectorstore):
     if not GROQ_API_KEY:
         return "❌ API 키가 설정되지 않았습니다. 관리자에게 문의하세요."
 
-    # 벡터스토어가 로드되지 않은 경우
-    if not vectorstores:
+    # 통합된 벡터스토어가 로드되지 않은 경우
+    if not combined_vectorstore:
         return "❌ 사용 가능한 벡터스토어가 없습니다. 관리자에게 문의하세요."
 
     try:
-        # 선택된 벡터스토어 확인
-        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}")
+        print(f"✅ 통합된 벡터스토어를 사용하여 검색 중...")
 
         # LLM 설정
         llm = ChatGroq(
@@ -205,7 +213,7 @@ def respond_with_groq(question, selected_q, model, selected_vectorstore):
         qa_chain = RetrievalQA.from_chain_type(
             llm=llm,
             chain_type="stuff",
-            retriever=current_vectorstore.as_retriever(search_kwargs={"k": 3}),
+            retriever=combined_vectorstore.as_retriever(search_kwargs={"k": 5}),
             chain_type_kwargs={"prompt": prompt},
             return_source_documents=True
         )
@@ -227,7 +235,7 @@ def update_question(selected):
     return ""
 
 # 앱 시작시 벡터스토어들 로드
-available_vectorstores = load_all_vectorstores()
+vectorstores_loaded = load_all_vectorstores()
 
 # Gradio 인터페이스 생성
 with gr.Blocks(title="한남대학교 Q&A") as interface:
@@ -240,28 +248,6 @@ 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="💡 자주 묻는 질문",
@@ -293,7 +279,7 @@ with gr.Blocks(title="한남대학교 Q&A") as interface:
     # 이벤트 연결
     submit_btn.click(
         fn=respond_with_groq,
-        inputs=[question_input, question_dropdown, model_choice, vectorstore_dropdown],
+        inputs=[question_input, question_dropdown, model_choice],
         outputs=output
     )