Update emotion_predictor.py

emotion_predictor.py

@@ -10,6 +10,7 @@ from sklearn.linear_model import LinearRegression
 from collections import defaultdict
 import base64
 from io import BytesIO
+import streamlit as st
 
 # 폰트 설정
 font_path = './NanumGothic.ttf'
@@ -136,4 +137,22 @@ def predict_and_plot(raw_text):
             html_output += "<p>⚠️ 시각화할 수 있는 감정이 없습니다.</p><hr/>"
 
     return html_output
-#22
+
+# 파일 업로드 처리 함수
+def upload_and_analyze():
+    uploaded_file = st.file_uploader("텍스트 파일 업로드", type=["txt"])
+    if uploaded_file is not None:
+        # 파일 내용 읽기
+        raw_text = uploaded_file.getvalue().decode("utf-8")
+        
+        # 예측 및 시각화
+        html_output = predict_and_plot(raw_text)
+        
+        # 출력 결과를 HTML로 보여주기
+        st.markdown(html_output, unsafe_allow_html=True)
+
+# Streamlit 앱 실행
+if __name__ == "__main__":
+    st.title("감정 분석기")
+    st.write("텍스트 파일을 업로드하여 감정 분석을 해보세요.")
+    upload_and_analyze()