import streamlit as st
import pandas as pd
import joblib
from predict_module import extract_features_from_video_id, predict_view_count, visualize_result, predict_views

# joblib에서 load를 가져옵니다
from joblib import load

# 모델 로드 (파일 경로는 Hugging Face Spaces 기준 상대경로로)
model = load("view_predictor.joblib")

# YouTube API 키 설정
api_key = "AIzaSyAgkZQp9EqA6N49J7TCh6Q40mWyVIGBit8"

st.title("🎬 YouTube 조회수 예측기")

# 사용자 입력
video_id = st.text_input("YouTube 영상 ID를 입력하세요:")

if st.button("예측 시작"):
    try:
        # 전체 정보
        info = predict_views(video_id, api_key)

        # 특징 추출
        features = extract_features_from_video_id(video_id, api_key)

        # 예측
        predicted = predict_view_count(model, features)

        # 시각화 출력
        html = visualize_result(video_id, features, predicted, info)
        st.components.v1.html(html, height=1000)

    except Exception as e:
        st.error(f"❌ 오류 발생: {e}")