Update app.py

app.py

@@ -26,7 +26,7 @@ def fetch_naver_news(query, display=5):
         return response.json()
     except requests.exceptions.RequestException as e:
         st.error(f"API 호출 중 오류 발생: {e}")
-        return None
+        return None  # 뉴스 데이터가 없으면 None 반환
 
 # 정치 성향 분류
 def classify_sentiment(text, classifier):
@@ -77,33 +77,34 @@ if st.button("분석 시작"):
             # 네이버 뉴스 데이터 수집
             news_data = fetch_naver_news(query, display=10)
             if news_data is None:
-                return
-            news_items = news_data["items"]
-
-            # 모델 로드
-            classifier = load_model()
-
-            # 뉴스 데이터 분석
-            results, detailed_results = analyze_news(news_items, classifier)
-
-            # 분석 결과 시각화
-            st.subheader("분석 결과 요약")
-            st.write(f"진보: {results['진보']}건")
-            st.write(f"보수: {results['보수']}건")
-            st.write(f"중립: {results['중립']}건")
-
-            # 파이 차트
-            st.subheader("성향 분포 차트")
-            st.bar_chart(pd.DataFrame.from_dict(results, orient='index', columns=["건수"]))
-
-            # 세부 결과 출력
-            st.subheader("세부 결과")
-            df = pd.DataFrame(detailed_results)
-            st.dataframe(df)
-
-            # 링크 포함한 뉴스 출력
-            st.subheader("뉴스 링크")
-            for index, row in df.iterrows():
-                st.write(f"- [{row['제목']}]({row['링크']}) (성향: {row['성향']}, 점수: {row['점수']:.2f})")
+                st.error("뉴스 데이터가 없습니다.")
+            else:
+                news_items = news_data["items"]
+
+                # 모델 로드
+                classifier = load_model()
+
+                # 뉴스 데이터 분석
+                results, detailed_results = analyze_news(news_items, classifier)
+
+                # 분석 결과 시각화
+                st.subheader("분석 결과 요약")
+                st.write(f"진보: {results['진보']}건")
+                st.write(f"보수: {results['보수']}건")
+                st.write(f"중립: {results['중립']}건")
+
+                # 파이 차트
+                st.subheader("성향 분포 차트")
+                st.bar_chart(pd.DataFrame.from_dict(results, orient='index', columns=["건수"]))
+
+                # 세부 결과 출력
+                st.subheader("세부 결과")
+                df = pd.DataFrame(detailed_results)
+                st.dataframe(df)
+
+                # 링크 포함한 뉴스 출력
+                st.subheader("뉴스 링크")
+                for index, row in df.iterrows():
+                    st.write(f"- [{row['제목']}]({row['링크']}) (성향: {row['성향']}, 점수: {row['점수']:.2f})")
         except Exception as e:
             st.error(f"오류 발생: {e}")