Create app.py

app.py

@@ -0,0 +1,36 @@
+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}")