Update app.py

app.py

@@ -6,6 +6,8 @@ from langchain.chains import RetrievalQA
 from langchain.prompts import PromptTemplate
 from langchain.vectorstores import FAISS
 from langchain.embeddings import HuggingFaceEmbeddings
+from datetime import datetime
+import json
 
 # SSL 경고 제거
 urllib3.disable_warnings(urllib3.exceptions.InsecureRequestWarning)
@@ -18,6 +20,94 @@ vectorstores = {}
 embeddings = None
 combined_vectorstore = None
 
+# 대화 기록 저장을 위한 전역 변수 (메모리 기반)
+conversation_history = []
+
+def save_conversation(question, answer, sources=None):
+    """대화 기록을 저장하는 함수"""
+    global conversation_history
+    
+    timestamp = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
+    
+    conversation_entry = {
+        "timestamp": timestamp,
+        "question": question,
+        "answer": answer,
+        "sources": sources if sources else []
+    }
+    
+    conversation_history.append(conversation_entry)
+    
+    # 최대 100개까지만 저장 (메모리 절약)
+    if len(conversation_history) > 100:
+        conversation_history = conversation_history[-100:]
+
+def get_conversation_history():
+    """대화 기록을 반환하는 함수"""
+    global conversation_history
+    
+    if not conversation_history:
+        return "아직 대화 기록이 없습니다."
+    
+    history_text = "## 🕐 대화 기록\n\n"
+    
+    # 최근 기록부터 표시
+    for i, entry in enumerate(reversed(conversation_history), 1):
+        history_text += f"### {i}. {entry['timestamp']}\n"
+        history_text += f"**❓ 질문:** {entry['question']}\n\n"
+        history_text += f"**💬 답변:** {entry['answer']}\n\n"
+        
+        if entry['sources']:
+            history_text += f"**📚 출처:** {', '.join(entry['sources'])}\n\n"
+        
+        history_text += "---\n\n"
+    
+    return history_text
+
+def clear_history():
+    """대화 기록을 삭제하는 함수"""
+    global conversation_history
+    conversation_history = []
+    return "대화 기록이 삭제되었습니다."
+
+def extract_source_info(source_documents):
+    """소스 문서에서 출처 정보를 추출하는 함수"""
+    sources = []
+    
+    for doc in source_documents:
+        # 메타데이터에서 소스 정보 추출
+        if hasattr(doc, 'metadata') and doc.metadata:
+            source = doc.metadata.get('source', '')
+            if source:
+                # 파일명에서 규정명 추출 (예: "학사규정.pdf" -> "학사규정")
+                source_name = os.path.basename(source).replace('.pdf', '').replace('.txt', '')
+                if source_name and source_name not in sources:
+                    sources.append(source_name)
+        
+        # 문서 내용에서 규정명 패턴 찾기
+        if hasattr(doc, 'page_content'):
+            content = doc.page_content
+            # 일반적인 규정명 패턴들
+            import re
+            patterns = [
+                r'([가-힣\s]+규정)',
+                r'([가-힣\s]+규칙)',
+                r'([가-힣\s]+세칙)',
+                r'([가-힣\s]+지침)',
+                r'([가-힣\s]+운영요강)',
+                r'([가-힣\s]+관리요령)'
+            ]
+            
+            for pattern in patterns:
+                matches = re.findall(pattern, content)
+                for match in matches:
+                    clean_match = match.strip()
+                    if len(clean_match) > 2 and clean_match not in sources:
+                        sources.append(clean_match)
+                        break  # 첫 번째 매치만 사용
+    
+    return sources[:3]  # 최대 3개까지만 반환
+
 def debug_file_system():
     """파일 시스템 상태를 자세히 확인하는 함수"""
     import os
@@ -326,17 +416,19 @@ prompt = PromptTemplate(
 )
 
 def respond_with_groq(question, selected_q, model):
-    """질문에 대한 답변을 생성하는 함수 - 개선 버전"""
+    """질문에 대한 답변을 생성하는 함수 - 개선 버전 (출처 정보 포함)"""
     
     # 선택된 질문이 있으면 그것을 사용
     if selected_q != "직접 입력":
         question = selected_q
 
     if not question.strip():
-        return "질문을 입력해주세요."
+        return "질문을 입력해주세요.", ""
 
     if not GROQ_API_KEY:
-        return "❌ GROQ API 키가 설정되지 않았습니다. Hugging Face Spaces의 Settings에서 GROQ_API_KEY를 설정해주세요."
+        error_msg = "❌ GROQ API 키가 설정되지 않았습니다. Hugging Face Spaces의 Settings에서 GROQ_API_KEY를 설정해주세요."
+        save_conversation(question, error_msg)
+        return error_msg, get_conversation_history()
 
     # 통합된 벡터스토어가 로드되지 않은 경우 재시도
     if not combined_vectorstore:
@@ -345,7 +437,7 @@ def respond_with_groq(question, selected_q, model):
         if not success:
             # 디버깅 정보 출력
             debug_file_system()
-            return """❌ 벡터스토어를 로드할 ��� 없습니다. 
+            error_msg = """❌ 벡터스토어를 로드할 수 없습니다. 
             
 가능한 원인:
 1. 벡터스토어 파일이 올바르게 업로드되지 않음
@@ -355,6 +447,8 @@ def respond_with_groq(question, selected_q, model):
 1. vectorstore 폴더들이 제대로 업로드되었는지 확인
 2. 각 폴더에 index.faiss와 index.pkl 파일이 있는지 확인
 3. Git LFS를 사용해 큰 파일들을 관리해보세요"""
+            save_conversation(question, error_msg)
+            return error_msg, get_conversation_history()
 
     try:
         print(f"🔍 질문: {question}")
@@ -380,7 +474,9 @@ def respond_with_groq(question, selected_q, model):
             print(f"🔍 검색된 문서 수: {len(docs)}")
         except Exception as e:
             print(f"❌ 검색 오류: {e}")
-            return f"❌ 문서 검색 중 오류가 발생했습니다: {str(e)}"
+            error_msg = f"❌ 문서 검색 중 오류가 발생했습니다: {str(e)}"
+            save_conversation(question, error_msg)
+            return error_msg, get_conversation_history()
 
         # QA 체인 생성
         qa_chain = RetrievalQA.from_chain_type(
@@ -393,18 +489,38 @@ def respond_with_groq(question, selected_q, model):
 
         # 답변 생성
         result = qa_chain({"query": question})
-        return result['result']
+        answer = result['result']
+        source_documents = result.get('source_documents', [])
+        
+        # 출처 정보 추출
+        sources = extract_source_info(source_documents)
+        
+        # 답변에 출처 정보 추가
+        if sources:
+            source_text = f"\n\n📚 **출처:** {', '.join(sources)}"
+            final_answer = answer + source_text
+        else:
+            final_answer = answer
+            sources = []
+        
+        # 대화 기록 저장
+        save_conversation(question, final_answer, sources)
+        
+        return final_answer, get_conversation_history()
 
     except Exception as e:
         import traceback
         error_details = traceback.format_exc()
         print(f"❌ 상세 오류 정보:\n{error_details}")
-        return f"""❌ 답변 생성 중 오류가 발생했습니다: {str(e)}
+        error_msg = f"""❌ 답변 생성 중 오류가 발생했습니다: {str(e)}
 디버깅 정보:
 - 벡터스토어 로드됨: {combined_vectorstore is not None}
 - API 키 설정됨: {GROQ_API_KEY is not None}
 - 모델: {model}
 관리자에게 위 정보와 함께 문의해주세요."""
+        
+        save_conversation(question, error_msg)
+        return error_msg, get_conversation_history()
 
 def update_question(selected):
     """드롭다운 선택 시 질문을 업데이트하는 함수"""
@@ -412,6 +528,11 @@ def update_question(selected):
         return selected
     return ""
 
+def handle_clear_history():
+    """기록 삭제 버튼 핸들러"""
+    clear_msg = clear_history()
+    return clear_msg, ""
+
 # 앱 시작시 벡터스토어들 로드
 print("🚀 앱 시작 - 벡터스토어 로딩 중...")
 vectorstores_loaded = load_all_vectorstores()
@@ -440,41 +561,64 @@ with gr.Blocks(title="한남대학교 Q&A") as interface:
     </div>
     """)
 
-    with gr.Row():
-        with gr.Column(scale=1):
-            question_dropdown = gr.Dropdown(
-                choices=["직접 입력"] + suggested_questions,
-                label="💡 자주 묻는 질문",
-                value="직접 입력"
-            )
+    # 탭 생성
+    with gr.Tabs():
+        # 메인 Q&A 탭
+        with gr.TabItem("💬 질문하기"):
+            with gr.Row():
+                with gr.Column(scale=1):
+                    question_dropdown = gr.Dropdown(
+                        choices=["직접 입력"] + suggested_questions,
+                        label="💡 자주 묻는 질문",
+                        value="직접 입력"
+                    )
 
-            question_input = gr.Textbox(
-                label="❓ 질문을 입력하세요",
-                placeholder="예: 졸업 요건은 무엇인가요?",
-                lines=3
-            )
+                    question_input = gr.Textbox(
+                        label="❓ 질문을 입력하세요",
+                        placeholder="예: 졸업 요건은 무엇인가요?",
+                        lines=3
+                    )
 
-            submit_btn = gr.Button("답변 받기", variant="primary", size="lg")
+                    submit_btn = gr.Button("답변 받기", variant="primary", size="lg")
 
-            model_choice = gr.Radio(
-                choices=["llama3-70b-8192", "llama3-8b-8192"],
-                label="🤖 AI 모델 선택",
-                value="llama3-70b-8192"
-            )
+                    model_choice = gr.Radio(
+                        choices=["llama3-70b-8192", "llama3-8b-8192"],
+                        label="🤖 AI 모델 선택",
+                        value="llama3-70b-8192"
+                    )
 
-        with gr.Column(scale=2):
-            output = gr.Textbox(
-                label="💬 답변",
-                lines=15,
-                max_lines=20,
-                show_copy_button=True
-            )
+                with gr.Column(scale=2):
+                    output = gr.Textbox(
+                        label="💬 답변",
+                        lines=15,
+                        max_lines=20,
+                        show_copy_button=True
+                    )
+
+        # 대화 기록 탭
+        with gr.TabItem("📋 대화 기록"):
+            with gr.Row():
+                with gr.Column():
+                    gr.HTML("""
+                    <div style="text-align: center; padding: 15px; background-color: #f8f9fa; border-radius: 8px; margin-bottom: 15px;">
+                        <h3>🕐 이전 대화 기록</h3>
+                        <p>지금까지의 질문과 답변 내역을 확인할 수 있습니다.</p>
+                    </div>
+                    """)
+                    
+                    refresh_btn = gr.Button("🔄 기록 새로고침", variant="secondary")
+                    clear_history_btn = gr.Button("🗑️ 기록 삭제", variant="stop")
+                    
+                    history_output = gr.Markdown(
+                        value="아직 대화 기록이 없습니다.",
+                        label="대화 기록"
+                    )
 
     # 이벤트 연결
     submit_btn.click(
         fn=respond_with_groq,
         inputs=[question_input, question_dropdown, model_choice],
-        outputs=output
+        outputs=[output, history_output]
     )
 
     question_dropdown.change(
@@ -482,6 +626,16 @@ with gr.Blocks(title="한남대학교 Q&A") as interface:
         inputs=question_dropdown,
         outputs=question_input
     )
+    
+    refresh_btn.click(
+        fn=get_conversation_history,
+        outputs=history_output
+    )
+    
+    clear_history_btn.click(
+        fn=handle_clear_history,
+        outputs=[history_output, output]
+    )
 
 # 앱 실행
 if __name__ == "__main__":