import yfinance as yf
import pandas as pd
import streamlit as st

def load_data(ticker, start_date, end_date):
    """
    Loads stock data from Yahoo Finance for a single ticker and date range.

    Args:
        ticker (str): Stock ticker symbol.
        start_date (str): Start date in YYYY-MM-DD format.
        end_date (str): End date in YYYY-MM-DD format.

    Returns:
        pandas.DataFrame: DataFrame containing stock data for the ticker.
    """
    try:
        data = yf.download(ticker, start=start_date, end=end_date)
        # Flatten the MultiIndex columns
        data.columns = data.columns.droplevel(1)  # Drop the second level (ticker name)
        data.columns.name = None  # Remove the column name

        if data is None or data.empty:
            st.error(f"No data found for ticker: {ticker}")
            return None

        return data

    except Exception as e:
        st.error(f"Error loading data: {e}")
        return None