feat: add update news data (button) (#5)

- feat: add update news data (button) (370b54b6cf0450592b6155cfd3c505d89b412ade)


Co-authored-by: Hyeonseo Yun <[email protected]>

app.py

@@ -78,13 +78,14 @@ class news_collector:
 
 
 
-def collect_news():
-    news = news_collector()
-    news.update_news_examples()
-    return news.examples_text
-
-
+news_data = []
 
+def update_news_data():
+    global news_data
+    news_data = news_collector().update_news_examples()
+    print("뉴스 데이터 업데이트 완료")
+    
+update_news_data()
 
 title = "균형잡힌 뉴스 읽기 (Balanced News Reading)"
 
@@ -115,9 +116,12 @@ with gr.Blocks(theme='pseudolab/huggingface-korea-theme') as demo:
         # gr.load("models/Hyeonseo/ko-finance_news_classifier",
                 inputs = Link_TXT)
         Link_URL = gr.Textbox(label="뉴스 URL")
-
+        
+        update_button = gr.Button(value="뉴스 데이터 업데이트")
+        update_button.click(fn=update_news_data, inputs=None, outputs=None)
+        
         gr.Examples(
-            collect_news(),
+            news_data,
             [Link_TXT, Link_URL],
         )