Add application file

app.py

@@ -0,0 +1,277 @@
+"""
+RAG 챗봇 웹 클라이언트 애플리케이션
+
+Docker 기반 RAG API와 통신하는 웹 인터페이스를 제공합니다.
+VITO STT 기능을 통한 음성 질의도 지원합니다.
+"""
+
+import os
+import json
+import logging
+import tempfile
+import requests
+from flask import Flask, request, jsonify, render_template, send_from_directory
+from werkzeug.utils import secure_filename
+from dotenv import load_dotenv
+
+from utils.vito_stt import VitoSTT
+
+# 로거 설정
+logging.basicConfig(
+    format='%(asctime)s - %(name)s - %(levelname)s - %(message)s',
+    level=logging.INFO
+)
+logger = logging.getLogger(__name__)
+
+# 환경 변수 로드
+load_dotenv()
+
+# Flask 앱 초기화
+app = Flask(__name__)
+
+# 최대 파일 크기 설정 (10MB)
+app.config['MAX_CONTENT_LENGTH'] = 10 * 1024 * 1024
+app.config['UPLOAD_FOLDER'] = os.path.join(os.path.dirname(os.path.abspath(__file__)), 'uploads')
+
+# 업로드 폴더가 없으면 생성
+os.makedirs(app.config['UPLOAD_FOLDER'], exist_ok=True)
+
+# RAG API 설정
+API_BASE_URL = os.getenv('API_BASE_URL', 'http://localhost:8000')
+
+# 허용되는 오디오 파일 확장자
+ALLOWED_AUDIO_EXTENSIONS = {'mp3', 'wav', 'ogg', 'm4a'}
+
+# 허용되는 문서 파일 확장자
+ALLOWED_DOC_EXTENSIONS = {'txt', 'md', 'pdf', 'docx', 'csv'}
+
+# VITO STT 클라이언트 초기화
+stt_client = VitoSTT()
+
+def allowed_audio_file(filename):
+    """파일이 허용된 오디오 확장자를 가지는지 확인"""
+    return '.' in filename and filename.rsplit('.', 1)[1].lower() in ALLOWED_AUDIO_EXTENSIONS
+
+def allowed_doc_file(filename):
+    """파일이 허용된 문서 확장자를 가지는지 확인"""
+    return '.' in filename and filename.rsplit('.', 1)[1].lower() in ALLOWED_DOC_EXTENSIONS
+
+@app.route('/')
+def index():
+    """메인 페이지"""
+    return render_template('index.html')
+
+@app.route('/chat')
+def chat():
+    """채팅 페이지"""
+    return render_template('chat.html')
+
+@app.route('/knowledge')
+def knowledge():
+    """지식베이스 관리 페이지"""
+    return render_template('knowledge.html')
+
+@app.route('/api/chat', methods=['POST'])
+def api_chat():
+    """텍스트 채팅 API"""
+    try:
+        data = request.get_json()
+        if not data or 'query' not in data:
+            return jsonify({"error": "쿼리가 제공되지 않았습니다."}), 400
+        
+        query = data['query']
+        logger.info(f"쿼리 처리 중: {query}")
+        
+        # RAG API 호출
+        response = requests.post(
+            f"{API_BASE_URL}/rag",
+            json={
+                "query": query,
+                "retriever_type": data.get("retriever_type", "reranker"),
+                "top_k": data.get("top_k", 3),
+                "temperature": data.get("temperature", 0.7)
+            }
+        )
+        
+        # API 응답 확인
+        if response.status_code != 200:
+            logger.error(f"API 오류: {response.status_code} - {response.text}")
+            return jsonify({"error": f"API 오류: {response.text}"}), response.status_code
+        
+        # API 응답 반환
+        result = response.json()
+        logger.info(f"API 응답 성공: {len(result['answer'])} 문자")
+        return jsonify(result)
+        
+    except requests.RequestException as e:
+        logger.error(f"API 통신 오류: {str(e)}")
+        return jsonify({"error": f"API 서버 연결 오류: {str(e)}"}), 503
+        
+    except Exception as e:
+        logger.error(f"채팅 처리 중 오류: {str(e)}", exc_info=True)
+        return jsonify({"error": f"처리 중 오류 발생: {str(e)}"}), 500
+
+@app.route('/api/voice', methods=['POST'])
+def api_voice():
+    """음성 채팅 API - 오디오 파일을 텍스트로 변환하고 RAG API에 질의"""
+    logger.info("음성 채팅 요청 처리 중...")
+    
+    # 오디오 파일 확인
+    if 'audio' not in request.files:
+        logger.error("오디오 파일이 제공되지 않음")
+        return jsonify({"error": "오디오 파일이 제공되지 않았습니다."}), 400
+    
+    audio_file = request.files['audio']
+    
+    if audio_file.filename == '':
+        logger.error("오디오 파일명이 비어있음")
+        return jsonify({"error": "오디오 파일이 선택되지 않았습니다."}), 400
+    
+    if not allowed_audio_file(audio_file.filename):
+        logger.error(f"허용되지 않는 오디오 파일 형식: {audio_file.filename}")
+        return jsonify({"error": "허용되지 않는 오디오 파일 형식입니다."}), 400
+    
+    try:
+        # 오디오 파일 임시 저장
+        filename = secure_filename(audio_file.filename)
+        temp_dir = tempfile.mkdtemp()
+        temp_path = os.path.join(temp_dir, filename)
+        audio_file.save(temp_path)
+        
+        logger.info(f"오디오 파일 임시 저장: {temp_path}")
+        
+        # 파일을 바이트로 읽기
+        with open(temp_path, 'rb') as f:
+            audio_bytes = f.read()
+        
+        # VITO STT를 사용하여 음성 인식
+        stt_result = stt_client.transcribe_audio(audio_bytes)
+        
+        # 임시 파일 삭제
+        os.unlink(temp_path)
+        os.rmdir(temp_dir)
+        
+        if not stt_result["success"]:
+            logger.error(f"STT 오류: {stt_result.get('error', '알 수 없는 오류')}")
+            return jsonify({"error": f"음성 인식 오류: {stt_result.get('error', '알 수 없는 오류')}"}), 500
+        
+        transcription = stt_result["text"]
+        if not transcription:
+            logger.warning("STT 결과가 비어있음")
+            return jsonify({"error": "음성에서 텍스트를 인식하지 못했습니다."}), 400
+        
+        logger.info(f"음성 인식 결과: {transcription}")
+        
+        # RAG API 호출
+        try:
+            response = requests.post(
+                f"{API_BASE_URL}/rag",
+                json={
+                    "query": transcription,
+                    "retriever_type": "reranker",
+                    "top_k": 3
+                }
+            )
+            
+            if response.status_code != 200:
+                logger.error(f"RAG API 오류: {response.status_code} - {response.text}")
+                return jsonify({
+                    "transcription": transcription,
+                    "error": f"RAG API 오류: {response.text}"
+                }), response.status_code
+            
+            # API 결과에 음성 인식 결과 추가
+            result = response.json()
+            result["transcription"] = transcription
+            
+            logger.info("음성 처리 및 RAG 응답 성공")
+            return jsonify(result)
+            
+        except requests.RequestException as e:
+            logger.error(f"RAG API 통신 오류: {str(e)}")
+            return jsonify({
+                "transcription": transcription,
+                "error": f"RAG API 서버 연결 오류: {str(e)}"
+            }), 503
+        
+    except Exception as e:
+        logger.error(f"음성 처리 중 오류: {str(e)}", exc_info=True)
+        return jsonify({"error": f"음성 처리 중 오류 발생: {str(e)}"}), 500
+
+@app.route('/api/documents', methods=['GET'])
+def api_get_documents():
+    """문서 목록 조회 API"""
+    try:
+        # RAG API 호출
+        response = requests.get(f"{API_BASE_URL}/cache/stats")
+        
+        if response.status_code != 200:
+            logger.error(f"문서 목록 조회 API 오류: {response.status_code} - {response.text}")
+            return jsonify({"error": f"문서 목록 조회 API 오류: {response.text}"}), response.status_code
+        
+        # 캐시 통계 정보 반환
+        return jsonify(response.json())
+        
+    except requests.RequestException as e:
+        logger.error(f"API 통신 오류: {str(e)}")
+        return jsonify({"error": f"API 서버 연결 오류: {str(e)}"}), 503
+        
+    except Exception as e:
+        logger.error(f"문서 목록 조회 중 오류: {str(e)}", exc_info=True)
+        return jsonify({"error": f"처리 중 오류 발생: {str(e)}"}), 500
+
+@app.route('/api/upload', methods=['POST'])
+def api_upload_document():
+    """문서 업로드 API"""
+    # 파일 확인
+    if 'document' not in request.files:
+        return jsonify({"error": "문서 파일이 제공되지 않았습니다."}), 400
+    
+    doc_file = request.files['document']
+    
+    if doc_file.filename == '':
+        return jsonify({"error": "문서 파일이 선택되지 않았습니다."}), 400
+    
+    if not allowed_doc_file(doc_file.filename):
+        return jsonify({"error": "허용되지 않는 문서 파일 형식입니다."}), 400
+    
+    try:
+        # 파일 임시 저장
+        filename = secure_filename(doc_file.filename)
+        temp_dir = tempfile.mkdtemp()
+        temp_path = os.path.join(temp_dir, filename)
+        doc_file.save(temp_path)
+        
+        logger.info(f"문서 파일 임시 저장: {temp_path}")
+        
+        # 파일을 멀티파트 폼으로 업로드
+        with open(temp_path, 'rb') as f:
+            files = {'document': (filename, f)}
+            response = requests.post(f"{API_BASE_URL}/documents/upload", files=files)
+        
+        # 임시 파일 삭제
+        os.unlink(temp_path)
+        os.rmdir(temp_dir)
+        
+        if response.status_code not in (200, 201):
+            logger.error(f"문서 업로드 API 오류: {response.status_code} - {response.text}")
+            return jsonify({"error": f"문서 업로드 API 오류: {response.text}"}), response.status_code
+        
+        return jsonify(response.json())
+        
+    except requests.RequestException as e:
+        logger.error(f"API 통신 오류: {str(e)}")
+        return jsonify({"error": f"API 서버 연결 오류: {str(e)}"}), 503
+        
+    except Exception as e:
+        logger.error(f"문서 업로드 중 오류: {str(e)}", exc_info=True)
+        return jsonify({"error": f"처리 중 오류 발생: {str(e)}"}), 500
+
+# 정적 파일 ���빙
+@app.route('/static/<path:path>')
+def send_static(path):
+    return send_from_directory('static', path)
+
+if __name__ == '__main__':
+    port = int(os.getenv('PORT', 5000))
+    app.run(debug=True, host='0.0.0.0', port=port)