app.py

import gradio as gr
import requests
from bs4 import BeautifulSoup
import pandas as pd
import logging
import io
import tempfile
from openpyxl import load_workbook
from openpyxl.styles import Font

# 디버깅을 위한 로깅 설정
logging.basicConfig(level=logging.DEBUG)

def format_change_web(text):
    """
    전일비 및 등락률 텍스트를 아이콘과 색상으로 포맷팅합니다.
    상승: 붉은색, 하락: 파란색
    웹 인터페이스용 HTML 포맷팅
    """
    if "상한가" in text:
        number = ''.join(filter(lambda x: x.isdigit() or x == '-', text))
        return f'<span style="color:red;">↑{number}</span>'
    elif "상승" in text:
        number = ''.join(filter(lambda x: x.isdigit() or x == '-', text))
        return f'<span style="color:red;">▲{number}</span>'
    elif "하한가" in text:
        number = ''.join(filter(lambda x: x.isdigit() or x == '-', text))
        return f'<span style="color:blue;">↓{number}</span>'
    elif "하락" in text:
        number = ''.join(filter(lambda x: x.isdigit() or x == '-', text))
        return f'<span style="color:blue;">▼{number}</span>'
    elif text.startswith('+'):
        return f'<span style="color:red;">▲{text}</span>'
    elif text.startswith('-'):
        return f'<span style="color:blue;">▼{text}</span>'
    else:
        return text

def format_change_excel(text):
    """
    전일비 및 등락률 텍스트를 아이콘으로 포맷팅합니다.
    상승: ▲, ↑
    하락: ▼, ↓
    엑셀용 텍스트 포맷팅 (HTML 태그 없음)
    """
    if "상한가" in text:
        number = ''.join(filter(lambda x: x.isdigit() or x == '-', text))
        return f'↑{number}'
    elif "상승" in text:
        number = ''.join(filter(lambda x: x.isdigit() or x == '-', text))
        return f'▲{number}'
    elif "하한가" in text:
        number = ''.join(filter(lambda x: x.isdigit() or x == '-', text))
        return f'↓{number}'
    elif "하락" in text:
        number = ''.join(filter(lambda x: x.isdigit() or x == '-', text))
        return f'▼{number}'
    elif text.startswith('+'):
        return f'▲{text}'
    elif text.startswith('-'):
        return f'▼{text}'
    else:
        return text

def scrape_market(market):
    if market == "코스닥":
        url = "https://finance.naver.com/sise/sise_rise.naver?sosok=1"
    elif market == "코스피":
        url = "https://finance.naver.com/sise/sise_rise.naver?sosok=0"
    else:
        logging.error("잘못된 시장 선택")
        return "잘못된 시장을 선택하셨습니다.", None

    logging.debug(f"요청할 URL: {url}")

    try:
        response = requests.get(url)
        response.raise_for_status()
        logging.debug("웹 페이지 요청 성공")
    except requests.exceptions.RequestException as e:
        logging.error(f"웹 페이지 요청 실패: {e}")
        return f"데이터를 가져오는 중 오류가 발생했습니다: {e}", None

    soup = BeautifulSoup(response.text, "html.parser")
    table = soup.find("table", {"class": "type_2"})

    if not table:
        logging.error("테이블을 찾을 수 없습니다.")
        return "테이블을 찾을 수 없습니다.", None

    rows = table.find_all("tr")
    logging.debug(f"총 행 수: {len(rows)}")

    data_display = []  # 웹용 데이터
    data_excel = []    # 엑셀용 데이터
    headers = ["N", "종목명", "현재가", "전일비", "등락률", "거래량", 
               "매수호가", "매도호가", "매수총잔량", "매도총잔량", "PER", "ROE"]

    for row in rows[2:]:  # 헤더를 건너뛰기 위해 인덱스 2부터 시작
        cols = row.find_all("td")
        if len(cols) < 12:
            logging.debug("데이터가 부족한 행을 건너뜁니다.")
            continue
        try:
            n = cols[0].get_text(strip=True)
            종목명 = cols[1].find("a").get_text(strip=True)
            현재가 = cols[2].get_text(strip=True).replace(',', '')
            전일비_raw = cols[3].get_text(strip=True)
            등락률_raw = cols[4].get_text(strip=True)
            거래량 = cols[5].get_text(strip=True).replace(',', '')
            매수호가 = cols[6].get_text(strip=True).replace(',', '')
            매도호가 = cols[7].get_text(strip=True).replace(',', '')
            매수총잔량 = cols[8].get_text(strip=True).replace(',', '')
            매도총잔량 = cols[9].get_text(strip=True).replace(',', '')
            PER = cols[10].get_text(strip=True)
            ROE = cols[11].get_text(strip=True)
            
            # 전일비와 등락률 포맷팅
            전일비_web = format_change_web(전일비_raw)
            등락률_web = format_change_web(등락률_raw)
            전일비_excel = format_change_excel(전일비_raw)
            등락률_excel = format_change_excel(등락률_raw)
            
            logging.debug(f"추출 데이터 - N: {n}, 종목명: {종목명}, 현재가: {현재가}, 전일비: {전일비_raw} -> {전일비_excel}, 등락률: {등락률_raw} -> {등락률_excel}, 거래량: {거래량}, 매수호가: {매수호가}, 매도호가: {매도호가}, 매수총잔량: {매수총잔량}, 매도총잔량: {매도총잔량}, PER: {PER}, ROE: {ROE}")
            
            # 웹용 데이터에 HTML 포맷팅 적용
            data_display.append([n, 종목명, 현재가, 전일비_web, 등락률_web, 거래량, 
                                 매수호가, 매도호가, 매수총잔량, 매도총잔량, PER, ROE])
            
            # 엑셀용 데이터에 아이콘 추가 (HTML 태그 없음)
            data_excel.append([n, 종목명, 현재가, 전일비_excel, 등락률_excel, 거래량, 
                               매수호가, 매도호가, 매수총잔량, 매도총잔량, PER, ROE])
        except Exception as e:
            logging.error(f"데이터 추출 중 오류 발생: {e}")
            continue

    if not data_display:
        logging.error("추출된 데이터가 없습니다.")
        return "추출된 데이터가 없습니다.", None

    # 웹용 데이터프레임 생성
    df_display = pd.DataFrame(data_display, columns=headers)
    logging.debug("웹용 데이터프레임 생성 완료")
    
    # 엑셀용 데이터프레임 생성
    df_excel = pd.DataFrame(data_excel, columns=headers)
    logging.debug("엑셀용 데이터프레임 생성 완료")

    # HTML 테이블 생성 (웹용)
    html = """
    <style>
        table {
            width: 100%;
            border-collapse: collapse;
        }
        th, td {
            border: 1px solid #dddddd;
            text-align: center;
            padding: 8px;
        }
        th {
            background-color: #f2f2f2;
        }
    </style>
    <table>
        <tr>
            <th>N</th>
            <th>종목명</th>
            <th>현재가</th>
            <th>전일비</th>
            <th>등락률</th>
            <th>거래량</th>
            <th>매수호가</th>
            <th>매도호가</th>
            <th>매수총잔량</th>
            <th>매도총잔량</th>
            <th>PER</th>
            <th>ROE</th>
        </tr>
    """

    for index, row in df_display.iterrows():
        html += f"""
        <tr>
            <td>{row['N']}</td>
            <td>{row['종목명']}</td>
            <td>{row['현재가']}</td>
            <td>{row['전일비']}</td>
            <td>{row['등락률']}</td>
            <td>{row['거래량']}</td>
            <td>{row['매수호가']}</td>
            <td>{row['매도호가']}</td>
            <td>{row['매수총잔량']}</td>
            <td>{row['매도총잔량']}</td>
            <td>{row['PER']}</td>
            <td>{row['ROE']}</td>
        </tr>
        """
    
    html += "</table>"
    logging.debug("HTML 테이블 생성 완료")

    # 엑셀 파일 생성 및 셀 색상 적용
    try:
        excel_buffer = io.BytesIO()
        df_excel.to_excel(excel_buffer, index=False, engine='openpyxl')
        excel_buffer.seek(0)

        # 엑셀 파일 로드
        wb = load_workbook(excel_buffer)
        ws = wb.active

        # "전일비"는 D열, "등락률"은 E열
        for row in ws.iter_rows(min_row=2, min_col=4, max_col=5, max_row=ws.max_row):
            전일비_cell, 등락률_cell = row
            # 전일비 색상 적용
            if 전일비_cell.value.startswith('▲') or 전일비_cell.value.startswith('↑'):
                전일비_cell.font = Font(color="FF0000")  # 빨간색
            elif 전일비_cell.value.startswith('▼') or 전일비_cell.value.startswith('↓'):
                전일비_cell.font = Font(color="0000FF")  # 파란색
            
            # 등락률 색상 적용
            if 등락률_cell.value.startswith('▲') or 등락률_cell.value.startswith('↑'):
                등락률_cell.font = Font(color="FF0000")  # 빨간색
            elif 등락률_cell.value.startswith('▼') or 등락률_cell.value.startswith('↓'):
                등락률_cell.font = Font(color="0000FF")  # 파란색

        # 수정된 엑셀 파일을 다시 BytesIO로 저장
        final_excel = io.BytesIO()
        wb.save(final_excel)
        final_excel.seek(0)

        # 임시 파일 생성
        temp = tempfile.NamedTemporaryFile(delete=False, suffix=".xlsx")
        temp.write(final_excel.read())
        temp.close()
        logging.debug("엑셀 파일 생성 및 저장 완료")
    except Exception as e:
        logging.error(f"엑셀 파일 생성 중 오류 발생: {e}")
        temp = None

    return html, temp.name if temp else None

with gr.Blocks() as demo:
    gr.Markdown("### 네이버 증권 코스닥/코스피 종목 정보 스크래핑")
    
    with gr.Row():
        market_choice = gr.Radio(
            choices=["코스닥", "코스피"],
            label="시장 선택",
            value="코스닥"
        )
    
    btn = gr.Button("데이터 가져오기")
    
    with gr.Row():
        output_html = gr.HTML()
        output_file = gr.File(label="엑셀 파일 다운로드")
    
    btn.click(scrape_market, inputs=market_choice, outputs=[output_html, output_file])

if __name__ == "__main__":
    demo.launch()