import streamlit as st
import yfinance as yf
import pandas as pd
from Pattern import doji, gravestone_doji, dragonfly_doji, longleg_doji, Hammer_Hanging_Man, Inv_Hammer, \
    Spinning_Top, Marubozu, Engulf, Engulfing, Harami, DarkCloudCover, PiercingPattern

# Streamlit app
def main():
    st.title('Live Candlestick Pattern Detection App')
    
    ticker = st.text_input('Enter Stock Ticker:', 'AAPL')
    interval = st.selectbox('Select Interval:', ['5s','1m', '5m', '15m', '1h'])  # You can adjust intervals as needed
    
    # Fetch real-time data
    stock_data = yf.download(ticker, period='1h', interval=interval)
    
    # Detect candlestick patterns
    doji(stock_data)
    gravestone_doji(stock_data)
    dragonfly_doji(stock_data)
    longleg_doji(stock_data)
    Hammer_Hanging_Man(stock_data)
    Inv_Hammer(stock_data)
    Spinning_Top(stock_data)
    Marubozu(stock_data)
    Engulf(stock_data)
    Engulfing(stock_data)
    Harami(stock_data)
    DarkCloudCover(stock_data)
    PiercingPattern(stock_data)
    
    # Display detected patterns
    st.write(stock_data)

if __name__ == "__main__":
    main()