aliceblue11 commited on
Commit
eec2f50
·
verified ·
1 Parent(s): 6e7a40c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +9 -3
app.py CHANGED
@@ -1,3 +1,4 @@
 
1
  import requests
2
  from bs4 import BeautifulSoup
3
  import pandas as pd
@@ -22,7 +23,6 @@ def scrape_naver_stock():
22
  rank = cols[0].text.strip()
23
  name = cols[1].text.strip()
24
  price = cols[2].text.strip()
25
- # 전일비에서 상한가, 상승 등의 텍스트를 제거하고 숫자만 추출
26
  diff = cols[3].text.strip().replace("상한가", "▲").replace("상승", "▲")
27
  change_rate = cols[4].text.strip()
28
  volume = cols[5].text.strip()
@@ -42,7 +42,13 @@ def scrape_naver_stock():
42
  # 엑셀 파일 저장 함수
43
  def save_to_excel():
44
  df = scrape_naver_stock()
45
- file_path = "/mnt/data/naver_stock_data.xlsx" # Gradio에서 사용할 수 있는 파일 경로
 
 
 
 
 
 
46
  df.to_excel(file_path, index=False)
47
  return file_path
48
 
@@ -66,4 +72,4 @@ with gr.Blocks() as iface:
66
  # 스크래핑된 데이터를 UI에 로드
67
  iface.load(fn=display_stocks, inputs=[], outputs=stock_table)
68
 
69
- iface.launch()
 
1
+ import os
2
  import requests
3
  from bs4 import BeautifulSoup
4
  import pandas as pd
 
23
  rank = cols[0].text.strip()
24
  name = cols[1].text.strip()
25
  price = cols[2].text.strip()
 
26
  diff = cols[3].text.strip().replace("상한가", "▲").replace("상승", "▲")
27
  change_rate = cols[4].text.strip()
28
  volume = cols[5].text.strip()
 
42
  # 엑셀 파일 저장 함수
43
  def save_to_excel():
44
  df = scrape_naver_stock()
45
+ directory = "/mnt/data"
46
+
47
+ # 경로가 존재하지 않으면 생성
48
+ if not os.path.exists(directory):
49
+ os.makedirs(directory)
50
+
51
+ file_path = os.path.join(directory, "naver_stock_data.xlsx")
52
  df.to_excel(file_path, index=False)
53
  return file_path
54
 
 
72
  # 스크래핑된 데이터를 UI에 로드
73
  iface.load(fn=display_stocks, inputs=[], outputs=stock_table)
74
 
75
+ iface.launch(share=True)