app.py

import requests
from bs4 import BeautifulSoup
import pandas as pd
import gradio as gr

# URL 설정
KOSDAQ_URL = "https://finance.naver.com/sise/sise_rise.naver?sosok=1"

def scrape_kosdaq_info():
    # 요청 및 HTML 파싱
    response = requests.get(KOSDAQ_URL)
    response.raise_for_status()
    soup = BeautifulSoup(response.text, "html.parser")

    # 데이터 추출
    rows = soup.select("table.type_2 tbody tr")
    data = []
    for row in rows:
        # 종목명 추출
        name_tag = row.select_one("a.tltle")
        if name_tag:
            name = name_tag.text.strip()
            # 종목 코드 추출
            code = name_tag["href"].split("code=")[-1]
            data.append({"종목명": name, "종목코드": code})
    
    # DataFrame으로 변환
    df = pd.DataFrame(data)
    return df

def display_kosdaq_info():
    # 데이터 스크래핑 및 출력
    df = scrape_kosdaq_info()
    return df

# 그라디오 UI 정의
def kosdaq_ui():
    def get_table():
        # 테이블 데이터 가져오기
        df = scrape_kosdaq_info()
        return df

    interface = gr.Interface(
        fn=get_table,
        inputs=None,
        outputs="dataframe",
        title="코스닥 종목 정보 스크래퍼",
        description="네이버 증권 사이트에서 코스닥 종목 정보를 스크래핑하여 표시합니다."
    )
    interface.launch()

# 실행
if __name__ == "__main__":
    kosdaq_ui()