Update app.py

app.py

@@ -10,7 +10,7 @@ from groq import Groq
 os.environ["GROQ_API_KEY"] = "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"
 client = Groq(api_key=os.environ["GROQ_API_KEY"])
 
-# ✅ 2. PDF 파일 경로 리스트 (pdfs 폴더 기준)
+# ✅ 2. PDF 파일 경로 리스트
 pdf_paths = [
     "pdfs/의왕단오축제 _ 의왕문화원.pdf",
     "pdfs/성인 문화학교 _ 의왕문화원.pdf",
@@ -20,7 +20,7 @@ pdf_paths = [
     "pdfs/어린이 문화학교 _ 의왕문화원.pdf"
 ]
 
-# ✅ 3. PDF 텍스트 추출
+# ✅ 3. 텍스트 추출
 def extract_texts_from_pdfs(pdf_paths):
     chunks = []
     for path in pdf_paths:
@@ -40,7 +40,7 @@ def extract_texts_from_pdfs(pdf_paths):
 
 docs = extract_texts_from_pdfs(pdf_paths)
 
-# ✅ 4. 문서 임베딩 및 FAISS 색인 구축
+# ✅ 4. 임베딩 및 FAISS 색인
 embed_model = SentenceTransformer("jhgan/ko-sroberta-multitask")
 texts = [doc["text"] for doc in docs]
 sources = [doc["source"] for doc in docs]
@@ -49,7 +49,7 @@ embeddings = embed_model.encode(texts, convert_to_numpy=True, show_progress_bar=
 index = faiss.IndexFlatL2(embeddings.shape[1])
 index.add(np.array(embeddings))
 
-# ✅ 5. 유사 문서 검색 함수
+# ✅ 5. 유사 문서 검색
 def search_similar_docs(query, top_k=3):
     query_emb = embed_model.encode([query])[0]
     scores, indices = index.search(np.array([query_emb]), top_k)
@@ -58,7 +58,7 @@ def search_similar_docs(query, top_k=3):
         results.append(docs[idx]["text"])
     return "\n\n".join(results)
 
-# ✅ 6. Groq 기반 응답 생성
+# ✅ 6. Groq 응답
 def ask_with_groq(question, context):
     response = client.chat.completions.create(
         model="llama3-8b-8192",
@@ -69,13 +69,13 @@ def ask_with_groq(question, context):
     )
     return response.choices[0].message.content
 
-# ✅ 7. Gradio 인터페이스 함수
+# ✅ 7. 챗봇 인터페이스
 def chatbot_interface(question):
     context = search_similar_docs(question)
     answer = ask_with_groq(question, context)
     return answer
 
-# ✅ 8. 예시 질문 리스트
+# ✅ 8. 추천 질문 리스트
 suggested_questions = [
     "의왕단오축제는 언제 열리나요?",
     "성인 문화학교에서 제공하는 수업 종류는 무엇인가요?",
@@ -84,27 +84,31 @@ suggested_questions = [
     "횡성축제의 개최 시기와 일정은 어떻게 되나요?",
 ]
 
-# ✅ 9. Gradio UI 구성
+# ✅ 9. Gradio UI
 with gr.Blocks() as demo:
     gr.Markdown("# 📘 오아시스 문서 기반 문화 프로그램 챗봇 🤖 '뮤지스(Musesis)'")
     gr.Markdown("질문을 입력하면 유사 문서를 검색해 답변해줍니다.")
 
     with gr.Row():
-        question_box = gr.Textbox(placeholder="질문을 입력하세요", label="질문")
-    with gr.Row():
-        submit_btn = gr.Button("질문하기")
-    with gr.Row():
-        answer_box = gr.Textbox(label="답변", lines=10)
-
-    with gr.Row():
-        gr.Markdown("### 💡 추천 질문")
-    with gr.Column():
-        for q in suggested_questions:
-            btn = gr.Button(q)
-            btn.click(fn=chatbot_interface, inputs=gr.Textbox(value=q, visible=False), outputs=answer_box)
+        with gr.Column(scale=3):
+            question_box = gr.Textbox(placeholder="질문을 입력하세요", label="질문", lines=1)
+            with gr.Row():
+                submit_btn = gr.Button("질문하기")
+                clear_btn = gr.Button("기록 지우기")
+            answer_box = gr.Textbox(label="답변", lines=10)
+        with gr.Column(scale=1):
+            with gr.Accordion("💡 추천 질문 보기 (펼치기/접기)", open=False):
+                for q in suggested_questions:
+                    btn = gr.Button(q, scale=1)
+                    btn.click(fn=chatbot_interface, inputs=gr.Textbox(value=q, visible=False), outputs=answer_box)
 
+    # ✅ 버튼 클릭 및 Enter로 질문 실행
     submit_btn.click(fn=chatbot_interface, inputs=question_box, outputs=answer_box)
+    question_box.submit(fn=chatbot_interface, inputs=question_box, outputs=answer_box)
+
+    # ✅ 기록 지우기
+    clear_btn.click(lambda: ("", ""), inputs=[], outputs=[question_box, answer_box])
 
-# ✅ 10. 앱 실행
+# ✅ 10. 실행
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch()