Update app.py

app.py

@@ -1,6 +1,7 @@
 # streamlit 라이브러리가 import 되기 전에,
 # 설정 파일 경로를 앱 내부의 쓰기 가능한 경로로 강제 지정합니다.
 import os, pathlib
+from huggingface_hub import hf_hub_download
 
 APP_DIR = pathlib.Path(__file__).parent.resolve()
 os.environ.setdefault("HOME", str(APP_DIR))  # '~'가 /가 아니라 /app으로 가도록
@@ -11,7 +12,43 @@ os.environ["STREAMLIT_HOME"] = str(CONFIG_DIR)
 os.environ["STREAMLIT_SERVER_HEADLESS"] = "true"
 os.environ["STREAMLIT_BROWSER_GATHER_USAGE_STATS"] = "false"  # 선택: metrics 파일 생성 줄이기
 
+HF_DATASET_REPO = os.getenv("HF_DATASET_REPO", "yourname/your-dataset")  # ← 본인 리포
+HF_DATASET_REV  = os.getenv("HF_DATASET_REV", "main")
+
+def _is_pointer_bytes(b: bytes) -> bool:
+    head = b[:2048].decode(errors="ignore").lower()
+    # git-lfs / xet 포인터 텍스트 패턴 모두 감지
+    return (
+        "version https://git-lfs.github.com/spec/v1" in head or
+        "git-lfs" in head or
+        "xet" in head or            # e.g. "Xet backed hash"
+        "pointer size" in head
+    )
 
+def _read_csv_bytes(b: bytes) -> pd.DataFrame:
+    try:
+        return pd.read_csv(io.BytesIO(b), encoding="utf-8")
+    except UnicodeDecodeError:
+        return pd.read_csv(io.BytesIO(b), encoding="cp949")
+
+def load_csv_smart(local_path: str, hub_filename: str,
+                   repo_id: str = HF_DATASET_REPO, repo_type: str = "dataset",
+                   revision: str = HF_DATASET_REV) -> pd.DataFrame:
+    # 1) 로컬 우선
+    if os.path.exists(local_path):
+        with open(local_path, "rb") as f:
+            data = f.read()
+        if not _is_pointer_bytes(data):
+            return _read_csv_bytes(data)
+        # 포인터면 허브로 폴백
+    # 2) 허브 다운로드
+    cached = hf_hub_download(repo_id=repo_id, filename=hub_filename,
+                             repo_type=repo_type, revision=revision)
+    try:
+        return pd.read_csv(cached, encoding="utf-8")
+    except UnicodeDecodeError:
+        return pd.read_csv(cached, encoding="cp949")
+        
 import pandas as pd
 import json
 import os
@@ -65,12 +102,12 @@ def load_data(path):
 #     st.error("TRIPDATA_URL 미설정: Streamlit Secrets에 URL을 넣어주세요.")
 #     st.stop()
 
-travel_df = load_data("트립닷컴_감정_테마_한줄설명_통합_07_08.csv")
-external_score_df = load_data("클러스터_포함_외부요인_종합점수_결과_최종.csv")
-festival_df = load_data("전처리_통합지역축제.csv")
-weather_df = load_data("전처리_날씨_통합_07_08.csv")
-package_df = load_data("모두투어_컬럼별_개수_07_08.csv")
-master_df = load_data("나라_도시_리스트.csv")
+travel_df = load_csv_smart("trip_emotions.csv")
+external_score_df = load_csv_smart("external_scores.csv")
+festival_df = load_csv_smart("festivals.csv")
+weather_df = load_csv_smart("weather.csv")
+package_df = load_csv_smart("packages.csv")
+master_df = load_csv_smart("countries_cities.csv")
 theme_title_phrases = load_json_data("theme_title_phrases.json")
 
 # travel_df가 성공적으로 로드되었는지 최종 확인