Add application file

app.py

@@ -861,7 +861,7 @@ class AutoRAGChatApp:
         def process_audio_auto(audio, chat_history):
             """녹음 완료 시 자동으로 STT 처리 후 질의 처리"""
             if audio is None:
-                return chat_history, None, gr.update(value=None, interactive=True), ""
+                return chat_history, gr.update(interactive=True), "녹음 후 자동으로 처리됩니다", ""
 
             # 처리 중 상태 메시지
             processing_msg = "음성 처리 중..."
@@ -871,8 +871,7 @@ class AutoRAGChatApp:
                 import scipy.io.wavfile as wav
 
                 # 상태 업데이트
-                yield chat_history, gr.update(value=processing_msg, interactive=False), gr.update(
-                    interactive=False), "음성을 텍스트로 변환 중..."
+                yield chat_history, gr.update(interactive=False), processing_msg, "음성을 텍스트로 변환 중..."
 
                 # numpy 배열을 WAV 파일로 저장
                 with tempfile.NamedTemporaryFile(suffix=".wav", delete=False) as temp_file:
@@ -900,7 +899,7 @@ class AutoRAGChatApp:
                     print(f"[STT] {error_msg}")
                     new_history = list(chat_history)
                     new_history.append(["음성 메시지", error_msg])
-                    yield new_history, None, gr.update(interactive=True), ""
+                    yield new_history, gr.update(interactive=True), error_msg, ""
                     return
 
                 # 인식된 텍스트 추출
@@ -910,14 +909,14 @@ class AutoRAGChatApp:
                     print("[STT] 인식된 텍스트 없음")
                     new_history = list(chat_history)
                     new_history.append(["음성 메시지", error_msg])
-                    yield new_history, None, gr.update(interactive=True), ""
+                    yield new_history, gr.update(interactive=True), error_msg, ""
                     return
 
                 print(f"[STT] 인식된 텍스트: {recognized_text}")
 
                 # 상태 업데이트
-                yield chat_history, gr.update(value=f"인식된 텍스트: {recognized_text}\n\n응답 생성 중...",
-                                              interactive=False), gr.update(interactive=False), "응답 생성 중..."
+                yield chat_history, gr.update(
+                    interactive=False), f"인식된 텍스트: {recognized_text}\n\n응답 생성 중...", "응답 생성 중..."
 
                 # 인식된 텍스트로 쿼리 처리 (음성 메시지 접두어 추가)
                 query = f"🎤 {recognized_text}"
@@ -940,13 +939,13 @@ class AutoRAGChatApp:
                             response = f"문서를 로드할 수 없습니다. 'documents' 폴더에 PDF 파일이 있는지 확인하세요.\n오류 정보: {init_result}"
                             new_history = list(chat_history)
                             new_history.append([query, response])
-                            yield new_history, None, gr.update(interactive=True), ""
+                            yield new_history, gr.update(interactive=True), response, ""
                             return
                     except Exception as e:
                         response = f"문서 로드 중 오류 발생: {str(e)}"
                         new_history = list(chat_history)
                         new_history.append([query, response])
-                        yield new_history, None, gr.update(interactive=True), ""
+                        yield new_history, gr.update(interactive=True), response, ""
                         return
                 else:
                     print("음성 쿼리 처리: 문서가 이미 로드되어 있습니다.")
@@ -973,7 +972,7 @@ class AutoRAGChatApp:
                     new_history.append([query, response])  # 리스트 형식으로 추가
 
                     # 최종 상태 업데이트
-                    yield new_history, None, gr.update(interactive=True), ""
+                    yield new_history, gr.update(interactive=True), f"처리 완료: {recognized_text}", ""
 
                 except Exception as e:
                     error_msg = f"오류 발생: {str(e)}"
@@ -983,7 +982,7 @@ class AutoRAGChatApp:
                     # 메시지 형식에 맞게 추가
                     new_history = list(chat_history)  # 기존 리스트를 복사
                     new_history.append([query, error_msg])  # 리스트 형식으로 추가
-                    yield new_history, None, gr.update(interactive=True), ""
+                    yield new_history, gr.update(interactive=True), error_msg, ""
 
             except Exception as e:
                 error_msg = f"음성 처리 중 오류 발생: {str(e)}"
@@ -992,7 +991,7 @@ class AutoRAGChatApp:
                 traceback.print_exc()
                 new_history = list(chat_history)
                 new_history.append(["음성 메시지", error_msg])
-                yield new_history, None, gr.update(interactive=True), ""
+                yield new_history, gr.update(interactive=True), error_msg, ""
 
         with gr.Blocks(title="음성인식 기능이 추가된 PDF 문서 기반 RAG 챗봇") as app:
             gr.Markdown("# 음성인식 기능이 추가된 PDF 문서 기반 RAG 챗봇")