Update app.py

app.py

@@ -22,7 +22,8 @@ def scrape_naver_stock():
             rank = cols[0].text.strip()
             name = cols[1].text.strip()
             price = cols[2].text.strip()
-            diff = cols[3].text.strip()
+            # 전일비에서 상한가, 상승 등의 텍스트를 제거하고 숫자만 추출
+            diff = cols[3].text.strip().replace("상한가", "▲").replace("상승", "▲")
             change_rate = cols[4].text.strip()
             volume = cols[5].text.strip()
             buy_price = cols[6].text.strip()
@@ -38,10 +39,31 @@ def scrape_naver_stock():
     
     return df
 
+# 엑셀 파일 저장 함수
+def save_to_excel():
+    df = scrape_naver_stock()
+    file_path = "/mnt/data/naver_stock_data.xlsx"  # Gradio에서 사용할 수 있는 파일 경로
+    df.to_excel(file_path, index=False)
+    return file_path
+
 # 그라디오 UI
 def display_stocks():
     df = scrape_naver_stock()
     return df
 
-iface = gr.Interface(fn=display_stocks, inputs=[], outputs="dataframe")
+# 그라디오 인터페이스
+with gr.Blocks() as iface:
+    # 스크래핑된 데이터 테이블 출력
+    stock_table = gr.DataFrame(label="상승 TOP 종목")
+    download_button = gr.Button("엑셀 파일 다운로드")
+
+    # 버튼 클릭 시 엑셀 파일 저장 및 다운로드 제공
+    download_output = gr.File()
+
+    # 버튼 클릭 시 데이터 업데이트
+    download_button.click(save_to_excel, outputs=download_output)
+
+    # 스크래핑된 데이터를 UI에 로드
+    iface.load(fn=display_stocks, inputs=[], outputs=stock_table)
+
 iface.launch()