Update app.py

app.py

@@ -4,6 +4,8 @@ from bs4 import BeautifulSoup
 import pandas as pd
 import re
 import gradio as gr
+from openpyxl import load_workbook
+from openpyxl.styles import Font
 
 # 네이버 증권 URL
 url = "https://finance.naver.com/sise/sise_rise.naver?sosok=1"
@@ -39,9 +41,10 @@ def scrape_naver_finance():
 
             # '상한가', '상승' 등의 글씨 제거
             change_cleaned = re.sub(r'[^\d-]', '', change).strip()
+            rate_cleaned = re.sub(r'[^\d.-]', '', rate).strip()
 
             # 각 열의 데이터를 리스트로 저장
-            data.append([rank, name, price, change_cleaned, rate, volume])
+            data.append([rank, name, price, change_cleaned, rate_cleaned, volume])
 
     # DataFrame으로 변환
     df = pd.DataFrame(data, columns=["순위", "종목명", "현재가", "전일비", "등락률", "거래량"])
@@ -55,13 +58,25 @@ def scrape_naver_finance():
     save_path = os.path.join(save_dir, "naver_top_stocks.xlsx")
     df.to_excel(save_path, index=False)
 
-    return df
+    # 엑셀 파일에 스타일 추가 (상승률이 20% 이상이면 종목명을 빨간색으로 표시)
+    workbook = load_workbook(save_path)
+    sheet = workbook.active
+    red_font = Font(color="FF0000")  # 빨간색 글씨체 설정
 
-# 그라디오 인터페이스 정의
+    for row in range(2, sheet.max_row + 1):
+        rate_value = sheet.cell(row=row, column=5).value
+        if rate_value and float(rate_value) >= 20.0:
+            sheet.cell(row=row, column=2).font = red_font  # 종목명 컬럼을 빨간색으로 설정
+
+    workbook.save(save_path)
+
+    return save_path
+
+# 엑셀 파일을 다운로드할 수 있도록 링크 제공
 def get_top_stocks():
-    df = scrape_naver_finance()
-    return df
+    file_path = scrape_naver_finance()
+    return file_path
 
-# Gradio 인터페이스 생성
-iface = gr.Interface(fn=get_top_stocks, inputs=None, outputs="dataframe", title="네이버 증권 상승 TOP 종목")
+# Gradio 인터페이스 생성 (다운로드 링크 제공)
+iface = gr.Interface(fn=get_top_stocks, inputs=None, outputs="file", title="네이버 증권 상승 TOP 종목")
 iface.launch()