Update app.py

app.py

@@ -1,25 +1,24 @@
-# app.py
-
 import os
 from groq import Groq
 import fitz  # PyMuPDF
 import gradio as gr
 
-# 환경변수에 Groq API 키 설정 (로컬 실행 시 미리 설정하거나 직접 변경 필요)
-os.environ["GROQ_API_KEY"] = "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"
-client = Groq(api_key=os.environ["GROQ_API_KEY"])
+# Hugging Face에서 제공하는 Secrets로 API 키 가져오기
+api_key = os.environ.get("GROQ_API_KEY")
+if not api_key:
+    raise ValueError("❌ GROQ_API_KEY가 환경변수로 설정되어 있지 않습니다.")
+client = Groq(api_key=api_key)
 
-# PDF 파일 경로 리스트
+# PDF 파일 리스트 (Spaces의 Files 탭에서 파일 직접 업로드 필요)
 pdf_paths = [
-    "./pdf/의왕단오축제 _ 의왕문화원.pdf",
-    "./pdf/성인 문화학교 _ 의왕문화원.pdf",
-    "./pdf/횡성문화원_문화학교.pdf",
-    "./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:
@@ -32,10 +31,8 @@ def extract_texts_from_pdfs(pdf_paths):
             print(f"❌ 오류: {path} -> {e}")
     return docs
 
-# 문서 데이터 로드
 all_documents = extract_texts_from_pdfs(pdf_paths)
 
-# 키워드 기반 문서 검색
 def get_documents_by_keyword(documents, keyword):
     matches = []
     for doc in documents:
@@ -43,7 +40,6 @@ def get_documents_by_keyword(documents, keyword):
             matches.append(doc["text"])
     return "\n\n".join(matches)
 
-# Groq API로 질문 답변
 def ask_question_with_context(question, context):
     if not context:
         return "❌ 관련 키워드를 포함하는 문서를 찾을 수 없습니다."
@@ -56,7 +52,6 @@ def ask_question_with_context(question, context):
     )
     return response.choices[0].message.content
 
-# 자동 키워드 추출
 def auto_keyword_extraction(message):
     keywords = ["단오축제", "문화학교", "횡성축제", "꿈꾸는 마을", "발달장애인", "의왕문화원", "횡성문화원"]
     for k in keywords:
@@ -64,7 +59,6 @@ def auto_keyword_extraction(message):
             return k
     return ""
 
-# Gradio 챗봇 상태 저장용
 chat_history = []
 
 def chatbot_fn(message, keyword):
@@ -72,14 +66,12 @@ def chatbot_fn(message, keyword):
     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))
@@ -90,7 +82,6 @@ def clear_history():
     chat_history = []
     return chat_history
 
-# 추천 질문 리스트
 suggested_questions = [
     '의왕단오축제는 언제 열리나요?',
     '의왕단오축제 주요 프로그램은 무엇인가요?',
@@ -102,19 +93,8 @@ suggested_questions = [
     '횡성지역축제의 대표 축제는 무엇인가요?',
     '횡성축제의 개최 시기와 일정은 어떻게 되나요?',
     '의왕문화원의 어린이 문화학교는 몇 세부터 참여 가능한가요?',
-    '어린이 문화학교에서 진행하는 주요 프로그램은 무엇인가요?',
-    '의왕문화원 문화 프로그램은 어디에서 진행되나요?',
-    '각 프로그램의 참가 신청 방법과 일정은 어떻게 되나요?',
-    '발달장애인 교육사업에서 제공하는 지원 서비스는 무엇인가요?',
-    '의왕단오축제 참가비는 얼마인가요?',
-    '횡성문화원 문화학교는 몇 개의 과정을 운영하나요?',
-    '어린이 문화학교 행사 일정은 어떻게 되나요?',
-    '문화 프로그램이 지역사회에 미치는 긍정적 영향은 무엇인가요?',
-    '프로그램 안전관리 및 코로나 방역 대책은 어떻게 되어 있나요?',
-    '문화 프로그램 참여 후 제공되는 피드백 절차가 있나요?'
 ]
 
-# Gradio UI 구성
 with gr.Blocks(title="🤖 뮤지스 챗봇(Musesis)") as demo:
     gr.Markdown("## 📘 오아시스 문서 기반 문화 프로그램 챗봇 🤖 '뮤지스(Musesis)'")
     gr.Markdown("업로드된 문서를 바탕으로 질문에 답변합니다. 키워드를 입력하지 않아도 자동으로 인식됩니다.")
@@ -133,19 +113,10 @@ with gr.Blocks(title="🤖 뮤지스 챗봇(Musesis)") as demo:
     user_input.submit(chatbot_fn, inputs=[user_input, keyword_input], outputs=[user_input, chatbot])
     clear_btn.click(clear_history, outputs=chatbot)
 
-    with gr.Accordion("💡 추천 질문 목록 (클릭 시 바로 질문)", open=False):
+    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()
+            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)
+
+demo.launch()