Add application file

app.py

@@ -490,7 +490,7 @@ class AutoRAGChatApp:
             chat_history.append((query, error_msg))
             return "", chat_history
 
-    def process_voice_query(self, audio, chat_history: List[Tuple[str, str]]) -> Tuple[str, List[Tuple[str, str]]]:
+    def process_voice_query(self, audio, chat_history: List[Tuple[str, str]]) -> List[Tuple[str, str]]:
         """
         음성 쿼리 처리
 
@@ -499,7 +499,7 @@ class AutoRAGChatApp:
             chat_history: 대화 기록
 
         Returns:
-            응답 및 업데이트된 대화 기록
+            업데이트된 대화 기록
         """
         if audio is None:
             return "", chat_history
@@ -546,7 +546,40 @@ class AutoRAGChatApp:
             print(f"[STT] 인식된 텍스트: {recognized_text}")
 
             # 인식된 텍스트로 쿼리 처리 (음성 메시지 접두어 추가)
-            return self.process_query(f"🎤 {recognized_text}", chat_history)
+            query = f"🎤 {recognized_text}"
+
+            # RAG 체인 실행 및 응답 생성
+            if not self.is_initialized:
+                response = "문서 로드가 초기화되지 않았습니다. 자동 로드를 시도합니다."
+                chat_history.append((query, response))
+
+                # 자동 로드 시도
+                try:
+                    self.auto_process_documents()
+                    if not self.is_initialized:
+                        response = "문서를 로드할 수 없습니다. 'documents' 폴더에 PDF 파일이 있는지 확인하세요."
+                        chat_history.append((query, response))
+                        return chat_history
+                except Exception as e:
+                    response = f"문서 로드 중 오류 발생: {str(e)}"
+                    chat_history.append((query, response))
+                    return chat_history
+
+            try:
+                # RAG 체인 실행 및 응답 생성
+                start_time = time.time()
+                response = self.rag_chain.run(query)
+                end_time = time.time()
+
+                query_time = end_time - start_time
+                print(f"쿼리 처리 시간: {query_time:.2f}초")
+
+                chat_history.append((query, response))
+                return chat_history
+            except Exception as e:
+                error_msg = f"오류 발생: {str(e)}"
+                chat_history.append((query, error_msg))
+                return chat_history
 
         except Exception as e:
             error_msg = f"음성 처리 중 오류 발생: {str(e)}"
@@ -658,7 +691,11 @@ class AutoRAGChatApp:
             voice_submit_btn.click(
                 fn=self.process_voice_query,
                 inputs=[audio_input, chatbot],
-                outputs=[audio_input, chatbot]
+                outputs=[chatbot]
+            ).then(
+                fn=lambda: None,  # Audio 컴포넌트 초기화 함수
+                inputs=None,
+                outputs=[audio_input]
             )
 
             # 대화 초기화 버튼