Update app.py

app.py

@@ -20,26 +20,30 @@ def fetch_kosdaq_data():
         data = []
         for row in rows:
             columns = row.find_all("td")
-            if len(columns) > 0:
-                # 데이터 파싱
-                rank = columns[0].get_text(strip=True)
-                name = columns[1].get_text(strip=True)
-                current_price = columns[2].get_text(strip=True)
-                diff = columns[3].get_text(strip=True)
-                change_rate = columns[4].get_text(strip=True)
-                volume = columns[5].get_text(strip=True)
-                buy_price = columns[6].get_text(strip=True)
-                sell_price = columns[7].get_text(strip=True)
-                buy_total = columns[8].get_text(strip=True)
-                sell_total = columns[9].get_text(strip=True)
-                per = columns[10].get_text(strip=True)
-                roe = columns[11].get_text(strip=True)
+            if len(columns) >= 12:  # 12개 열이 있는 행만 처리
+                try:
+                    # 데이터 파싱
+                    rank = columns[0].get_text(strip=True)
+                    name = columns[1].get_text(strip=True)
+                    current_price = columns[2].get_text(strip=True)
+                    diff = columns[3].get_text(strip=True)
+                    change_rate = columns[4].get_text(strip=True)
+                    volume = columns[5].get_text(strip=True)
+                    buy_price = columns[6].get_text(strip=True)
+                    sell_price = columns[7].get_text(strip=True)
+                    buy_total = columns[8].get_text(strip=True)
+                    sell_total = columns[9].get_text(strip=True)
+                    per = columns[10].get_text(strip=True)
+                    roe = columns[11].get_text(strip=True)
 
-                data.append([
-                    rank, name, current_price, diff, change_rate, 
-                    volume, buy_price, sell_price, buy_total, 
-                    sell_total, per, roe
-                ])
+                    data.append([
+                        rank, name, current_price, diff, change_rate, 
+                        volume, buy_price, sell_price, buy_total, 
+                        sell_total, per, roe
+                    ])
+                except Exception as e:
+                    print(f"Error parsing row: {e}")
+                    continue
 
         # DataFrame 생성
         columns = ["Rank", "Name", "Current Price", "Difference", "Change Rate", 
@@ -54,10 +58,10 @@ def fetch_kosdaq_data():
 
 def display_data():
     df = fetch_kosdaq_data()
-    if df is not None:
+    if df is not None and not df.empty:
         return df
     else:
-        return "Failed to fetch data. Please check the logs."
+        return "Failed to fetch data or no data available. Please check the logs."
 
 # Gradio 인터페이스 설정
 def gradio_interface():