데이터 소스 설정을 Lang2SQL 페이지에서 선택해 사용가능하도록 수정

Author: ehddnr301

interface/app_pages/components/data_source_selector.py

@@ -0,0 +1,85 @@
+import streamlit as st
+
+from interface.core.config import (
+    load_config,
+    get_data_sources_registry,
+    update_datahub_server,
+    update_vectordb_settings,
+    update_data_source_mode,
+)
+
+
+def render_sidebar_data_source_selector(config=None) -> None:
+    if config is None:
+        config = load_config()
+
+    registry = get_data_sources_registry()
+
+    st.sidebar.markdown("### 데이터 소스")
+    enable_data_source = st.sidebar.checkbox(
+        "데이터 소스 적용", value=True, key="enable_data_source"
+    )
+    if not enable_data_source:
+        return
+
+    mode_index = 0 if (config.data_source_mode or "datahub").lower() == "datahub" else 1
+    selected_mode = st.sidebar.radio(
+        "소스 종류", options=["DataHub", "VectorDB"], index=mode_index, horizontal=True
+    )
+
+    if selected_mode == "DataHub":
+        datahub_names = [s.name for s in registry.datahub]
+        if not datahub_names:
+            st.sidebar.warning(
+                "등록된 DataHub가 없습니다. 설정 > 데이터 소스에서 추가하세요."
+            )
+            return
+        dh_name = st.sidebar.selectbox(
+            "DataHub 인스턴스", options=datahub_names, key="sidebar_dh_select"
+        )
+        if st.sidebar.button("소스 적용", key="sidebar_apply_dh"):
+            selected = next((s for s in registry.datahub if s.name == dh_name), None)
+            if selected is None:
+                st.sidebar.error("선택한 DataHub를 찾을 수 없습니다.")
+                return
+            try:
+                update_datahub_server(config, selected.url)
+                # DataHub 선택 시, FAISS 경로가 정의되어 있으면 기본 VectorDB 로케이션으로도 반영
+                if selected.faiss_path:
+                    try:
+                        update_vectordb_settings(
+                            config,
+                            vectordb_type="faiss",
+                            vectordb_location=selected.faiss_path,
+                        )
+                    except Exception as e:
+                        st.sidebar.warning(f"FAISS 경로 적용 경고: {e}")
+                update_data_source_mode(config, "datahub")
+                st.sidebar.success(f"DataHub 적용됨: {selected.name}")
+            except Exception as e:
+                st.sidebar.error(f"적용 실패: {e}")
+    else:
+        vdb_names = [s.name for s in registry.vectordb]
+        if not vdb_names:
+            st.sidebar.warning(
+                "등록된 VectorDB가 없습니다. 설정 > 데이터 소스에서 추가하세요."
+            )
+            return
+        vdb_name = st.sidebar.selectbox(
+            "VectorDB 인스턴스", options=vdb_names, key="sidebar_vdb_select"
+        )
+        if st.sidebar.button("소스 적용", key="sidebar_apply_vdb"):
+            selected = next((s for s in registry.vectordb if s.name == vdb_name), None)
+            if selected is None:
+                st.sidebar.error("선택한 VectorDB를 찾을 수 없습니다.")
+                return
+            try:
+                update_vectordb_settings(
+                    config,
+                    vectordb_type=selected.type,
+                    vectordb_location=selected.location,
+                )
+                update_data_source_mode(config, "vectordb")
+                st.sidebar.success(f"VectorDB 적용됨: {selected.name}")
+            except Exception as e:
+                st.sidebar.error(f"적용 실패: {e}")

interface/app_pages/lang2sql.py

@@ -20,6 +20,10 @@
 from interface.core.lang2sql_runner import run_lang2sql
 from interface.core.result_renderer import display_result
 from interface.core.session_utils import init_graph
+from interface.core.config import load_config
+from interface.app_pages.components.data_source_selector import (
+    render_sidebar_data_source_selector,
+)
 
 TITLE = "Lang2SQL"
 DEFAULT_QUERY = "고객 데이터를 기반으로 유니크한 유저 수를 카운트하는 쿼리"
@@ -37,6 +41,10 @@
 
 st.title(TITLE)
 
+config = load_config()
+
+render_sidebar_data_source_selector(config)
+
 st.sidebar.markdown("### 워크플로우 선택")
 use_enriched = st.sidebar.checkbox(
     "프로파일 추출 & 컨텍스트 보강 워크플로우 사용", value=False

interface/app_pages/settings.py

@@ -14,7 +14,7 @@
 
 config = load_config()
 
-tabs = st.tabs(["데이터 소스", "LLM", "DB", "VectorDB", "Device"])
+tabs = st.tabs(["데이터 소스", "LLM", "DB", "Device"])
 
 with tabs[0]:
     render_data_source_section(config)
@@ -26,9 +26,6 @@
     st.info("DB 연결 설정은 곧 제공됩니다.")
 
 with tabs[3]:
-    st.info("VectorDB 설정은 곧 제공됩니다.")
-
-with tabs[4]:
     st.info("디바이스 설정은 곧 제공됩니다.")
 
 st.divider()