Update app.py

app.py

@@ -25,7 +25,8 @@ def fetch_naver_news(query, display=10, start=1, sort="date"):
         news_data = response.json()
         return news_data
     else:
-        raise Exception(f"Error: {response.status_code}, {response.text}")
+        st.error(f"Error: {response.status_code}, {response.text}")
+        return None
 
 # Step 2: Hugging Face 번역 모델 로드 (한국어 -> 영어)
 def load_translation_model():
@@ -99,39 +100,37 @@ if st.button("분석 시작"):
             # 네이버 뉴스 데이터 수집
             news_data = fetch_naver_news(query, display=10)
             
-            # 데이터 수집된 상태 확인
-            if not news_data or not news_data.get("items"):
+            if news_data is None:
                 st.error("뉴스 데이터를 불러오는 데 실패했습니다.")
-                return  # 오류 발생 시, return을 대신하여 오류를 보여줍니다.
-
-            news_items = news_data["items"]
-
-            # Hugging Face 모델 로드
-            classifier = load_huggingface_model()
-            translator = load_translation_model()
-
-            # 뉴스 데이터 분석
-            results, detailed_results = analyze_news_political_orientation(news_items, classifier, translator)
-
-            # 분석 결과 시각화
-            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})")
+            else:
+                news_items = news_data["items"]
+
+                # Hugging Face 모델 로드
+                classifier = load_huggingface_model()
+                translator = load_translation_model()
+
+                # 뉴스 데이터 분석
+                results, detailed_results = analyze_news_political_orientation(news_items, classifier, translator)
+
+                # 분석 결과 시각화
+                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}")