Create app.py

app.py

@@ -0,0 +1,77 @@
+import requests
+from bs4 import BeautifulSoup
+import pandas as pd
+import gradio as gr
+import logging
+
+# Set up logging
+logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')
+logger = logging.getLogger()
+
+def scrape_kosdaq():
+    url = "https://finance.naver.com/sise/sise_quant.naver?sosok=1"
+    headers = {
+        "User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/91.0.4472.124 Safari/537.36"
+    }
+
+    logger.info("Sending request to URL: %s", url)
+    response = requests.get(url, headers=headers)
+    
+    if response.status_code != 200:
+        logger.error("Failed to fetch data. Status code: %d", response.status_code)
+        return f"Error: Unable to fetch data (status code: {response.status_code})"
+
+    soup = BeautifulSoup(response.content, 'html.parser')
+
+    table = soup.find("table", {"class": "type_2"})
+    if not table:
+        logger.error("Failed to find the data table in the HTML.")
+        return "Error: Unable to find the data table."
+
+    rows = table.find_all("tr")
+    data = []
+    
+    for row in rows:
+        cols = row.find_all("td")
+        cols = [col.text.strip() for col in cols]
+        if len(cols) > 1:
+            data.append(cols)
+
+    if not data:
+        logger.error("No data found in the table.")
+        return "Error: No data found in the table."
+
+    # Define column names based on the structure
+    columns = [
+        "N", "종목명", "현재가", "전일비", "등락률", "거래량", "거래대금", "매수호가",
+        "매도호가", "시가총액", "PER", "ROE"
+    ]
+
+    df = pd.DataFrame(data, columns=columns[:len(data[0])])
+    
+    logger.info("Successfully scraped data with %d rows.", len(df))
+    return df
+
+def interface_function():
+    try:
+        df = scrape_kosdaq()
+        if isinstance(df, str):  # If error message
+            return df
+
+        logger.info("Displaying scraped data to user.")
+        return df.to_csv(index=False)
+    except Exception as e:
+        logger.exception("An error occurred during scraping.")
+        return f"An error occurred: {str(e)}"
+
+def display_interface():
+    gr.Interface(
+        fn=interface_function,
+        inputs=[],
+        outputs="text",
+        title="네이버 증권 코스닥 스크래퍼",
+        description="네이버 증권에서 코스닥 데이터를 스크래핑하여 CSV 형식으로 반환합니다."
+    ).launch()
+
+if __name__ == "__main__":
+    display_interface()