Update app.py

app.py

@@ -3,52 +3,66 @@ from bs4 import BeautifulSoup
 import pandas as pd
 import gradio as gr
 
-# URL 설정
+# 네이버 코스닥 URL
 KOSDAQ_URL = "https://finance.naver.com/sise/sise_rise.naver?sosok=1"
 
-def scrape_kosdaq_info():
-    # 요청 및 HTML 파싱
+def scrape_kosdaq_data():
+    print("디버깅: 네이버 증권 페이지 요청 시작...")
+    # 네이버 증권 페이지 요청
     response = requests.get(KOSDAQ_URL)
-    response.raise_for_status()
-    soup = BeautifulSoup(response.text, "html.parser")
+    if response.status_code == 200:
+        print("디버깅: 네이버 증권 페이지 요청 성공")
+    else:
+        print(f"디버깅: 요청 실패, 상태 코드: {response.status_code}")
+        return None
 
-    # 데이터 추출
+    # HTML 파싱
+    soup = BeautifulSoup(response.text, "html.parser")
+    
+    print("디버깅: 표 헤더 추출 시작...")
+    # 표의 헤더 추출
+    headers = []
+    header_tags = soup.select("table.type_2 thead tr th")
+    for tag in header_tags:
+        header_text = tag.get_text(strip=True)
+        if header_text:  # 빈 값 제거
+            headers.append(header_text)
+    print(f"디버깅: 추출된 헤더 - {headers}")
+    
+    print("디버깅: 표 데이터 추출 시작...")
+    # 표의 데이터 추출
     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})
+        cols = row.find_all("td")
+        if cols:  # 빈 행 제외
+            row_data = [col.get_text(strip=True) for col in cols]
+            data.append(row_data)
+    print(f"디버깅: 추출된 데이터 행 수 - {len(data)}")
     
-    # DataFrame으로 변환
-    df = pd.DataFrame(data)
-    return df
+    if data:
+        print(f"디버깅: 첫 번째 데이터 행 - {data[0]}")
+    else:
+        print("디버깅: 데이터 추출 실패")
 
-def display_kosdaq_info():
-    # 데이터 스크래핑 및 출력
-    df = scrape_kosdaq_info()
+    # DataFrame 생성
+    df = pd.DataFrame(data, columns=headers)
+    print("디버깅: DataFrame 생성 완료")
     return df
 
-# 그라디오 UI 정의
-def kosdaq_ui():
-    def get_table():
-        # 테이블 데이터 가져오기
-        df = scrape_kosdaq_info()
+def display_kosdaq_info():
+    # 데이터 스크래핑 및 반환
+    df = scrape_kosdaq_data()
+    if df is not None:
         return df
+    else:
+        return "데이터를 가져오는 데 실패했습니다."
 
-    interface = gr.Interface(
-        fn=get_table,
-        inputs=None,
-        outputs="dataframe",
-        title="코스닥 종목 정보 스크래퍼",
-        description="네이버 증권 사이트에서 코스닥 종목 정보를 스크래핑하여 표시합니다."
-    )
-    interface.launch()
-
-# 실행
-if __name__ == "__main__":
-    kosdaq_ui()
+# Gradio UI 설정
+gr.Interface(
+    fn=display_kosdaq_info,
+    inputs=None,
+    outputs="dataframe",
+    title="코스닥 종목 정보 스크래퍼",
+    description="네이버 증권 사이트에서 코스닥 종목 정보를 스크래핑하여 표시합니다."
+).launch()