Add application file

app.py

@@ -1,188 +1,75 @@
 import streamlit as st
 import requests
-import tempfile
 import os
-from dotenv import load_dotenv
 import json
+import sys
 
-# VITO STT 클래스 임포트
-from utils.vito_stt import VitoSTT
+# 시스템 정보 확인
+st.write("Python version:", sys.version)
+st.write("현재 작업 디렉토리:", os.getcwd())
 
-# 환경 변수 로드
-load_dotenv()
+# 환경 변수 확인 (보안상 민감한 정보는 가려서 표시)
+env_vars = {}
+for key in os.environ:
+    value = os.environ[key]
+    if key in ['API_BASE_URL']:
+        env_vars[key] = value
+    elif 'SECRET' in key or 'KEY' in key or 'TOKEN' in key or 'PASSWORD' in key:
+        env_vars[key] = value[:4] + '****' if len(value) > 4 else '****'
 
-# 페이지 구성
-st.set_page_config(
-    page_title="RAG 챗봇",
-    page_icon="🤖",
-    layout="wide"
-)
+st.subheader("환경 변수")
+st.json(env_vars)
 
-# RAG API 설정
-API_BASE_URL = os.getenv('API_BASE_URL', 'https://your-deployed-api-url.com')
+# API 연결 테스트
+st.subheader("API 연결 테스트")
 
-# VITO STT 클라이언트 초기화
-stt_client = VitoSTT()
+api_url = os.environ.get('API_BASE_URL', 'https://enter-your-api-url.com')
+st.write(f"API URL: {api_url}")
 
-# 세션 상태 초기화
-if 'messages' not in st.session_state:
-    st.session_state.messages = []
-if 'sources' not in st.session_state:
-    st.session_state.sources = []
+test_url = st.text_input("테스트할 API URL", api_url)
 
-# 사이드바 설정
-with st.sidebar:
-    st.title("🤖 RAG 챗봇")
-    st.write("Retrieval-Augmented Generation 기반 챗봇 서비스")
-    
-    st.subheader("⚙️ 설정")
-    retriever_type = st.selectbox(
-        "검색 엔진",
-        ["reranker", "vector"],
-        index=0
-    )
-    
-    top_k = st.slider(
-        "참고 문서 수 (Top-K)",
-        min_value=1,
-        max_value=10,
-        value=3
-    )
-    
-    temperature = st.slider(
-        "다양성 (Temperature)",
-        min_value=0.0,
-        max_value=1.0,
-        value=0.7,
-        step=0.1
-    )
-    
-    st.divider()
-    
-    st.subheader("📚 참고 문서")
-    if st.session_state.sources:
-        for i, source in enumerate(st.session_state.sources):
-            with st.expander(f"{i+1}. {source.get('metadata', {}).get('category', '일반')}"):
-                st.write(source.get('content', ''))
-    else:
-        st.info("참고 문서가 여기에 표시됩니다")
-
-# 메인 영역
-st.title("💬 RAG 챗봇")
+if st.button("API 연결 테스트"):
+    try:
+        with st.spinner("API 연결 테스트 중..."):
+            response = requests.get(f"{test_url}/docs", timeout=10)
+            st.success(f"API 연결 성공! 상태 코드: {response.status_code}")
+            st.text(f"응답 내용 (처음 500자): {response.text[:500]}")
+    except Exception as e:
+        st.error(f"API 연결 실패: {str(e)}")
+        st.info("API 서버가 실행 중인지, URL이 올바른지 확인하세요.")
 
-# 저장된 대화 표시
-for message in st.session_state.messages:
-    with st.chat_message(message["role"]):
-        st.write(message["content"])
+# 파일 시스템 확인
+st.subheader("파일 시스템 확인")
+files = os.listdir('.')
+st.write("현재 디렉토리 파일:", files)
 
-# 사용자 입력
-query = st.chat_input("질문을 입력하세요...")
+# 간단한 채팅 UI
+st.subheader("간단한 채팅 테스트")
+user_input = st.text_input("질문 입력")
 
-# 음성 입력 옵션
-audio_input = st.file_uploader("또는 음성으로 질문하기", type=["wav", "mp3", "ogg", "m4a"])
-
-if query:
-    # 사용자 메시지 추가
-    st.session_state.messages.append({"role": "user", "content": query})
-    with st.chat_message("user"):
-        st.write(query)
+if st.button("전송") and user_input:
+    st.write(f"사용자: {user_input}")
     
-    # 챗봇 응답 영역
-    with st.chat_message("assistant"):
-        with st.spinner("답변 생성 중..."):
-            try:
-                # RAG API 요청
-                response = requests.post(
-                    f"{API_BASE_URL}/rag",
-                    json={
-                        "query": query,
-                        "retriever_type": retriever_type,
-                        "top_k": top_k,
-                        "temperature": temperature
-                    },
-                    timeout=30
-                )
-                
-                if response.status_code == 200:
-                    result = response.json()
-                    answer = result.get("answer", "")
-                    context_docs = result.get("context_docs", [])
-                    
-                    # 응답 표시
-                    st.markdown(answer)
-                    
-                    # 세션 상태 업데이트
-                    st.session_state.messages.append({"role": "assistant", "content": answer})
-                    st.session_state.sources = context_docs
-                else:
-                    st.error(f"API 오류: {response.status_code} - {response.text}")
-            except Exception as e:
-                st.error(f"오류 발생: {str(e)}")
-
-# 음성 입력 처리
-if audio_input:
-    with st.spinner("음성 처리 중..."):
-        try:
-            # 임시 파일로 저장
-            with tempfile.NamedTemporaryFile(delete=False, suffix='.wav') as tmp:
-                tmp.write(audio_input.getvalue())
-                tmp_path = tmp.name
-            
-            # 파일 읽기
-            with open(tmp_path, 'rb') as f:
-                audio_bytes = f.read()
-            
-            # 임시 파일 삭제
-            os.unlink(tmp_path)
-            
-            # VITO STT 처리
-            stt_result = stt_client.transcribe_audio(audio_bytes)
+    try:
+        with st.spinner("응답 생성 중..."):
+            response = requests.post(
+                f"{api_url}/rag",
+                json={
+                    "query": user_input,
+                    "retriever_type": "reranker",
+                    "top_k": 3
+                },
+                timeout=30
+            )
             
-            if stt_result["success"]:
-                transcription = stt_result["text"]
-                if transcription:
-                    st.success(f"인식된 텍스트: {transcription}")
-                    
-                    # 사용자 메시지 추가
-                    st.session_state.messages.append({"role": "user", "content": transcription})
-                    with st.chat_message("user"):
-                        st.write(transcription)
-                    
-                    # 챗봇 응답 영역
-                    with st.chat_message("assistant"):
-                        with st.spinner("답변 생성 중..."):
-                            try:
-                                # RAG API 요청
-                                response = requests.post(
-                                    f"{API_BASE_URL}/rag",
-                                    json={
-                                        "query": transcription,
-                                        "retriever_type": retriever_type,
-                                        "top_k": top_k,
-                                        "temperature": temperature
-                                    },
-                                    timeout=30
-                                )
-                                
-                                if response.status_code == 200:
-                                    result = response.json()
-                                    answer = result.get("answer", "")
-                                    context_docs = result.get("context_docs", [])
-                                    
-                                    # 응답 표시
-                                    st.markdown(answer)
-                                    
-                                    # 세션 상태 업데이트
-                                    st.session_state.messages.append({"role": "assistant", "content": answer})
-                                    st.session_state.sources = context_docs
-                                else:
-                                    st.error(f"API 오류: {response.status_code} - {response.text}")
-                            except Exception as e:
-                                st.error(f"오류 발생: {str(e)}")
-                else:
-                    st.warning("음성에서 텍스트를 인식하지 못했습니다.")
+            if response.status_code == 200:
+                result = response.json()
+                st.write(f"챗봇: {result.get('answer', '응답을 받지 못했습니다.')}")
+                st.write("응답 세부 정보:")
+                st.json(result)
             else:
-                st.error(f"음성 인식 오류: {stt_result.get('error', '알 수 없는 오류')}")
-                
-        except Exception as e:
-            st.error(f"음성 처리 중 오류: {str(e)}")
+                st.error(f"API 오류: {response.status_code} - {response.text}")
+    except Exception as e:
+        st.error(f"오류 발생: {str(e)}")
+        st.info("자세한 오류 정보:")
+        st.exception(e)