from AlgorithmImports import *

class RSIStrategyAlgorithm(QCAlgorithm):
    def Initialize(self):
        self.SetStartDate(2024, 1, 1)  # Set Start Date
        self.SetCash(1000)           # Set Strategy Cash
        
        # Add equity with daily resolution
        self.spy = self.AddEquity("SMH", Resolution.Daily)
        
        # Initialize RSI(2) indicator
        self.rsi = self.RSI("SMH", 2, MovingAverageType.Simple, Resolution.Daily)
        
        # Variable to keep track of previous day's high
        self.previousHigh = None

    def OnData(self, data):
        # Skip if RSI data is not yet ready
        if not self.rsi.IsReady:
            return
        
        # Update previousHigh if it's not None
        if self.previousHigh is not None:
            # Exit condition: today's close is higher than yesterday's high
            if self.spy.Close > self.previousHigh and self.Portfolio["SMH"].Invested:
                self.Liquidate("SMH")
                
        # Update previousHigh with today's high at the end of the day
        self.previousHigh = self.spy.High
        
        # Entry condition: RSI(2) < 15 and not already invested
        if self.rsi.Current.Value < 20 and not self.Portfolio["SMH"].Invested:
            self.SetHoldings("SMH", 1)  # Use SetHoldings for simplicity; adjust position size as needed

# Note: Deploy this script within the QuantConnect environment to run the backtest.