Update app.py

app.py

@@ -1,63 +1,151 @@
-import gradio as gr
-import fitz  # PyMuPDF
-from openai import OpenAI
+# app.py
+
 import os
+from groq import Groq
+import fitz  # PyMuPDF
+import gradio as gr
 
-# Groq API 키 환경변수 설정 (본인 키로 변경)
+# 환경변수에 Groq API 키 설정 (로컬 실행 시 미리 설정하거나 직접 변경 필요)
 os.environ["GROQ_API_KEY"] = "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"
+client = Groq(api_key=os.environ["GROQ_API_KEY"])
+
+# PDF 파일 경로 리스트
+pdf_paths = [
+    "./pdf/의왕단오축제 _ 의왕문화원.pdf",
+    "./pdf/성인 문화학교 _ 의왕문화원.pdf",
+    "./pdf/횡성문화원_문화학교.pdf",
+    "./pdf/발달장애인 교육사업 _ 꿈꾸는 마을.pdf",
+    "./pdf/횡성의지역축제 - 횡성축제소개.pdf",
+    "./pdf/어린이 문화학교 _ 의왕문화원.pdf"
+]
+
+# PDF 텍스트 추출 함수
+def extract_texts_from_pdfs(pdf_paths):
+    docs = []
+    for path in pdf_paths:
+        try:
+            with fitz.open(path) as doc:
+                text = "".join(page.get_text() for page in doc)
+                docs.append({"filename": path, "text": text})
+                print(f"✅ 텍스트 추출 성공: {path}")
+        except Exception as e:
+            print(f"❌ 오류: {path} -> {e}")
+    return docs
+
+# 문서 데이터 로드
+all_documents = extract_texts_from_pdfs(pdf_paths)
 
-# Groq 클라이언트 초기화
-client = OpenAI(
-    base_url="https://api.groq.com/openai/v1",
-    api_key=os.environ["GROQ_API_KEY"]
-)
-
-def extract_text_from_pdf(file):
-    """업로드된 PDF 파일에서 텍스트 추출"""
-    doc = fitz.open(file.name)
-    text = ""
-    for page in doc:
-        text += page.get_text()
-    return text
-
-def ask_question(question, context):
-    """Groq API 호출로 질문에 답변 생성"""
+# 키워드 기반 문서 검색
+def get_documents_by_keyword(documents, keyword):
+    matches = []
+    for doc in documents:
+        if keyword.lower() in doc["text"].lower():
+            matches.append(doc["text"])
+    return "\n\n".join(matches)
+
+# Groq API로 질문 답변
+def ask_question_with_context(question, context):
+    if not context:
+        return "❌ 관련 키워드를 포함하는 문서를 찾을 수 없습니다."
     response = client.chat.completions.create(
         model="llama3-8b-8192",
         messages=[
-            {"role": "system", "content": "너는 업로드한 문서를 설명하는 한국어 도우미야. 항상 한국어로 답변해."},
-            {"role": "user", "content": f"{question}\n\n[문서 내용 일부]\n{context[:3000]}"}
+            {"role": "system", "content": "너는 업로드한 문서를 바탕으로 설명하는 한국어 도우미야."},
+            {"role": "user", "content": f"{question}\n\n[문서 내용]\n{context[:3000]}"}
         ]
     )
     return response.choices[0].message.content
 
-# 상태 유지용 전역 변수 (추출된 문서 텍스트)
-global_document_text = ""
+# 자동 키워드 추출
+def auto_keyword_extraction(message):
+    keywords = ["단오축제", "문화학교", "횡성축제", "꿈꾸는 마을", "발달장애인", "의왕문화원", "횡성문화원"]
+    for k in keywords:
+        if k in message:
+            return k
+    return ""
+
+# Gradio 챗봇 상태 저장용
+chat_history = []
+
+def chatbot_fn(message, keyword):
+    global chat_history
+    message = message.strip()
+    keyword = keyword.strip()
+
+    # 키워드 자동 추출
+    if not keyword:
+        keyword = auto_keyword_extraction(message)
+        if not keyword:
+            chat_history.append(("🙋‍♂️ " + message, "❌ 키워드를 입력하지 않았고 자동 추출도 실패했습니다. 키워드를 입력해주세요."))
+            return "", chat_history
+
+    # 문서 검색 및 응답
+    context = get_documents_by_keyword(all_documents, keyword)
+    response = ask_question_with_context(message, context)
+    chat_history.append(("🙋‍♂️ " + message, "🤖 " + response))
+    return "", chat_history
+
+def clear_history():
+    global chat_history
+    chat_history = []
+    return chat_history
+
+# 추천 질문 리스트
+suggested_questions = [
+    '의왕단오축제는 언제 열리나요?',
+    '의왕단오축제 주요 프로그램은 무엇인가요?',
+    '의왕문화원의 성인 문화학교 교육 대상은 누구인가요?',
+    '성인 문화학교에서 제공하는 수업 종류는 무엇인가요?',
+    '횡성문화원의 문화학교는 어떤 교육을 진행하나요?',
+    '발달장애인 교육사업 "꿈꾸는 마을"의 주요 목표는 무엇인가요?',
+    '꿈꾸는 마을 프로그램 참가 자격은 어떻게 되나요?',
+    '횡성지역축제의 대표 축제는 무엇인가요?',
+    '횡성축제의 개최 시기와 일정은 어떻게 되나요?',
+    '의왕문화원의 어린이 문화학교는 몇 세부터 참여 가능한가요?',
+    '어린이 문화학교에서 진행하는 주요 프로그램은 무엇인가요?',
+    '의왕문화원 문화 프로그램은 어디에서 진행되나요?',
+    '각 프로그램의 참가 신청 방법과 일정은 어떻게 되나요?',
+    '발달장애인 교육사업에서 제공하는 지원 서비스는 무엇인가요?',
+    '의왕단오축제 참가비는 얼마인가요?',
+    '횡성문화원 문화학교는 몇 개의 과정을 운영하나요?',
+    '어린이 문화학교 행사 일정은 어떻게 되나요?',
+    '문화 프로그램이 지역사회에 미치는 긍정적 영향은 무엇인가요?',
+    '프로그램 안전관리 및 코로나 방역 대책은 어떻게 되어 있나요?',
+    '문화 프로그램 참여 후 제공되는 피드백 절차가 있나요?'
+]
 
-def upload_and_extract(pdf_file):
-    global global_document_text
-    text = extract_text_from_pdf(pdf_file)
-    global_document_text = text
-    return "PDF 텍스트가 성공적으로 추출되었습니다. 질문을 입력하세요."
+# Gradio UI 구성
+with gr.Blocks(title="🤖 뮤지스 챗봇(Musesis)") as demo:
+    gr.Markdown("## 📘 오아시스 문서 기반 문화 프로그램 챗봇 🤖 '뮤지스(Musesis)'")
+    gr.Markdown("업로드된 문서를 바탕으로 질문에 답변합니다. 키워드를 입력하지 않아도 자동으로 인식됩니다.")
 
-def answer_question(question):
-    global global_document_text
-    if not global_document_text:
-        return "먼저 PDF를 업로드해서 텍스트를 추출해주세요."
-    return ask_question(question, global_document_text)
+    chatbot = gr.Chatbot(label="🤖 챗봇 응답", height=400)
 
-with gr.Blocks() as demo:
-    gr.Markdown("# 📄 PDF 기반 한국어 질문 답변 챗봇 (Groq API)")
+    with gr.Row():
+        keyword_input = gr.Textbox(label="🔍 키워드 (선택)", placeholder="예: 단오축제, 문화학교")
+        user_input = gr.Textbox(label="✉️ 질문", placeholder="질문을 입력하세요", lines=1)
 
-    pdf_input = gr.File(label="PDF 파일 업로드", file_types=[".pdf"])
-    upload_btn = gr.Button("PDF 텍스트 추출")
-    upload_status = gr.Textbox(label="상태", interactive=False)
+    with gr.Row():
+        send_btn = gr.Button("질문하기")
+        clear_btn = gr.Button("대화 초기화")
 
-    question_input = gr.Textbox(label="질문 입력", placeholder="질문을 입력하세요")
-    answer_output = gr.Textbox(label="챗봇 답변", interactive=False)
+    send_btn.click(chatbot_fn, inputs=[user_input, keyword_input], outputs=[user_input, chatbot])
+    user_input.submit(chatbot_fn, inputs=[user_input, keyword_input], outputs=[user_input, chatbot])
+    clear_btn.click(clear_history, outputs=chatbot)
 
-    upload_btn.click(upload_and_extract, inputs=pdf_input, outputs=upload_status)
-    question_input.submit(answer_question, inputs=question_input, outputs=answer_output)
+    with gr.Accordion("💡 추천 질문 목록 (클릭 시 바로 질문)", open=False):
+        for q in suggested_questions:
+            btn = gr.Button(q)
+            btn.click(
+                fn=lambda x=q: chatbot_fn(x, ""),
+                inputs=[],
+                outputs=[user_input, chatbot]
+            )
+            btn.click(
+                fn=lambda x=q: x,
+                inputs=None,
+                outputs=user_input
+            )
 
 if __name__ == "__main__":
     demo.launch()