Update app.py

app.py

@@ -38,15 +38,6 @@ def debug_file_system():
                 try:
                     sub_items = os.listdir(item_path)
                     print(f"   └── 내용: {sub_items}")
-                    
-                    # 벡터스토어 관련 파일 크기 확인
-                    if item.startswith('vectorstore'):
-                        for sub_item in sub_items:
-                            sub_path = os.path.join(item_path, sub_item)
-                            if os.path.isfile(sub_path):
-                                size = os.path.getsize(sub_path)
-                                print(f"   └── {sub_item}: {size:,} bytes")
-                                
                 except Exception as e:
                     print(f"   └── 접근 불가: {e}")
             else:
@@ -57,6 +48,7 @@ def debug_file_system():
     
     # 환경 변수 확인
     print(f"🔑 GROQ_API_KEY 설정됨: {'GROQ_API_KEY' in os.environ}")
+    
     print("=" * 50)
 
 # 앱 시작 시 디버깅 실행
@@ -68,12 +60,8 @@ def find_vectorstore_folders():
     print(f"현재 디렉토리: {current_dir}")
     
     # 모든 파일과 폴더 확인
-    try:
-        all_items = os.listdir(current_dir)
-        print(f"현재 디렉토리 내 모든 항목들: {all_items}")
-    except Exception as e:
-        print(f"디렉토리 읽기 오류: {e}")
-        return []
+    all_items = os.listdir(current_dir)
+    print(f"현재 디렉토리 내 모든 항목들: {all_items}")
     
     # 예상되는 벡터스토어 폴더들
     expected_folders = ['vectorstore1', 'vectorstore2', 'vectorstore3']
@@ -91,18 +79,8 @@ def find_vectorstore_folders():
                 has_all_files = all(file in folder_contents for file in required_files)
                 
                 if has_all_files:
-                    # 파일 크기도 확인
-                    faiss_path = os.path.join(folder_path, 'index.faiss')
-                    pkl_path = os.path.join(folder_path, 'index.pkl')
-                    
-                    faiss_size = os.path.getsize(faiss_path) if os.path.exists(faiss_path) else 0
-                    pkl_size = os.path.getsize(pkl_path) if os.path.exists(pkl_path) else 0
-                    
-                    if faiss_size > 0 and pkl_size > 0:
-                        vectorstore_folders.append(folder_name)
-                        print(f"✅ {folder_name} - 모든 필수 파일 존재 (FAISS: {faiss_size:,}bytes, PKL: {pkl_size:,}bytes)")
-                    else:
-                        print(f"❌ {folder_name} - 파일이 비어있음 (FAISS: {faiss_size}bytes, PKL: {pkl_size}bytes)")
+                    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}")
@@ -114,7 +92,6 @@ def find_vectorstore_folders():
     
     if not vectorstore_folders:
         print("❌ 사용 가능한 벡터스토어 폴더를 찾을 수 없습니다")
-        print("💡 허깅페이스 스페이스에 벡터스토어 폴더들이 제대로 업로드되었는지 확인하세요")
     else:
         print(f"✅ 총 {len(vectorstore_folders)}개의 벡터스토어 폴더를 찾았습니다: {vectorstore_folders}")
     
@@ -124,29 +101,18 @@ def load_all_vectorstores():
     """모든 벡터스토어를 로드하고 통합하는 함수"""
     global vectorstores, embeddings, combined_vectorstore
     
-    print("🔄 임베딩 모델 로딩 중...")
-    try:
-        if not embeddings:
-            embeddings = HuggingFaceEmbeddings(
-                model_name="jhgan/ko-sbert-nli"
-            )
-        print("✅ 임베딩 모델 로드 완료")
-    except Exception as e:
-        print(f"❌ 임베딩 모델 로드 실패: {e}")
-        return False
+    if not embeddings:
+        embeddings = HuggingFaceEmbeddings(
+            model_name="sentence-transformers/all-MiniLM-L6-v2"
+        )
     
     # 벡터스토어 폴더들 찾기
     folders = find_vectorstore_folders()
     
-    if not folders:
-        print("❌ 로드할 벡터스토어 폴더가 없습니다")
-        return False
-    
     loaded_vectorstores = []
     
     for folder_name in folders:
         try:
-            print(f"🔄 {folder_name} 로딩 중...")
             vectorstore = FAISS.load_local(
                 f"./{folder_name}", 
                 embeddings, 
@@ -154,41 +120,24 @@ def load_all_vectorstores():
             )
             vectorstores[folder_name] = vectorstore
             loaded_vectorstores.append(vectorstore)
-            
-            # 벡터스토어 정보 출력
-            doc_count = vectorstore.index.ntotal
-            print(f"✅ {folder_name} 로드 완료 (문서 수: {doc_count})")
-            
+            print(f"✅ {folder_name} 로드 완료")
         except Exception as e:
             print(f"❌ {folder_name} 로드 실패: {e}")
-            import traceback
-            traceback.print_exc()
     
-    # 벡터스토어들을 통합
+    # 벡터스토어들을 통합 (첫 번째를 기본으로 하고 나머지를 merge)
     if loaded_vectorstores:
-        print("🔄 벡터스토어 통합 중...")
-        try:
-            combined_vectorstore = loaded_vectorstores[0]
-            total_docs = combined_vectorstore.index.ntotal
-            
-            for i, vs in enumerate(loaded_vectorstores[1:], 1):
+        combined_vectorstore = loaded_vectorstores[0]
+        for vs in loaded_vectorstores[1:]:
+            try:
                 combined_vectorstore.merge_from(vs)
-                total_docs += vs.index.ntotal
-                print(f"✅ 벡터스토어 {i+1} 통합 완료")
-                
-            print(f"🎉 총 {len(loaded_vectorstores)}개의 벡터스토어가 통합되었습니다 (총 문서 수: {total_docs})")
-            return True
-            
-        except Exception as e:
-            print(f"❌ 벡터스토어 통합 실패: {e}")
-            import traceback
-            traceback.print_exc()
-            return False
-    else:
-        print("❌ 로드된 벡터스토어가 없습니다")
-        return False
+                print(f"✅ 벡터스토어 통합 완료")
+            except Exception as e:
+                print(f"❌ 벡터스토어 통합 실패: {e}")
+        print(f"🎉 총 {len(loaded_vectorstores)}개의 벡터스토어가 통합되었습니다")
+    
+    return len(loaded_vectorstores) > 0
 
-# 질문 리스트
+# 질문 리스트 (기존과 동일)
 suggested_questions = [
     '교원 신규 임용은 어떻게 하나요?',
     '교원 연구년 기간은 어떻게 되나요?',
@@ -212,23 +161,19 @@ suggested_questions = [
     '안전사고예방계획은 어디에 제출해야 하나요?'
 ]
 
-# 프롬프트 템플릿
+# 프롬프트 템플릿 (기존과 동일)
 prompt_template = """당신은 한남대학교 규정집 도우미입니다.
 반드시 한국어로만 답변해주세요. 영어나 다른 언어는 절대 사용하지 마세요.
 주어진 문서 내용을 바탕으로 질문에 대해 정확하고 친절하게 한국어로 답변해주세요.
-
 참고 문서:
 {context}
-
 질문: {question}
-
 답변 지침:
 - 이용자를 반기는 인사로 시작하세요
 - 반드시 한국어로만 답변하세요
 - 정중하고 친근한 말투를 사용하세요
 - 구체적이고 도움이 되는 정보를 제공하세요
 - 문서에서 답을 찾을 수 없으면 "죄송하지만 해당 정보를 규정집에서 찾을 수 없습니다"라고 답변하세요
-
 한국어 답변:"""
 
 prompt = PromptTemplate(
@@ -251,10 +196,10 @@ def respond_with_groq(question, selected_q, model):
 
     # 통합된 벡터스토어가 로드되지 않은 경우
     if not combined_vectorstore:
-        return "❌ 사용 가능한 벡터스토어가 없습니다. 벡터스토어 파일들이 제대로 업로드되었는지 확인하세요."
+        return "❌ 사용 가능한 벡터스토어가 없습니다. 관리자에게 문의하세요."
 
     try:
-        print(f"✅ 통합된 벡터스토어를 사용하여 검색 중... (질문: {question})")
+        print(f"✅ 통합된 벡터스토어를 사용하여 검색 중...")
 
         # LLM 설정
         llm = ChatGroq(
@@ -275,7 +220,6 @@ def respond_with_groq(question, selected_q, model):
 
         # 답변 생성
         result = qa_chain({"query": question})
-        print(f"✅ 답변 생성 완료")
         return result['result']
 
     except Exception as e:
@@ -291,14 +235,8 @@ def update_question(selected):
     return ""
 
 # 앱 시작시 벡터스토어들 로드
-print("🚀 애플리케이션 시작 - 벡터스토어 로딩 중...")
 vectorstores_loaded = load_all_vectorstores()
 
-if vectorstores_loaded:
-    print("🎉 모든 벡터스토어가 성공적으로 로드되었습니다!")
-else:
-    print("❌ 벡터스토어 로드에 실패했습니다. 파일 업로드를 확인하세요.")
-
 # Gradio 인터페이스 생성
 with gr.Blocks(title="한남대학교 Q&A") as interface:
     gr.HTML("""
@@ -307,15 +245,6 @@ with gr.Blocks(title="한남대학교 Q&A") as interface:
         <p>한남대학교 규정집에 대한 질문에 답변해드립니다.</p>
     </div>
     """)
-    
-    # 시스템 상태 표시
-    with gr.Row():
-        system_status = gr.Textbox(
-            label="🔧 시스템 상태",
-            value=get_system_status(),
-            lines=4,
-            interactive=False
-        )
 
     with gr.Row():
         with gr.Column(scale=1):