Update app.py

app.py

@@ -27,11 +27,11 @@ def load_sp500_tickers():
 
 
 
-def load_data(ticker):
-    """Load stock data using yfinance."""
+def load_data(ticker, interval="1d"):
+    """Load stock data using yfinance with a specified interval."""
     end_date = datetime.today()
     start_date = end_date - timedelta(days=365)  # Get 1 year of data
-    data = yf.download(ticker, start=start_date, end=end_date)
+    data = yf.download(ticker, start=start_date, end=end_date, interval=interval)
     return data
 
 
@@ -148,16 +148,16 @@ def next_business_day(date):
 
 
 
-def gradio_scan_sp500(top_n, progress=gr.Progress()):
+def gradio_scan_sp500(top_n, interval, progress=gr.Progress()):
     progress(0, desc="Downloading Data")
     tickers = load_sp500_tickers()
     tickers.append("QQQ")
     
     progress(0.3, desc="Running Scanner")
-    results = scan_sp500(top_n, progress)
+    results = scan_sp500(top_n, interval, progress)
     
     # Get the last date of the data and find the next business day
-    last_data = load_data(results[0][0])  # Load data for the first ticker in results
+    last_data = load_data(results[0][0], interval)  # Load data for the first ticker in results
     last_date = last_data.index[-1].date()
     next_market_day = next_business_day(last_date)
     date_created = next_market_day.strftime("%Y-%m-%d")
@@ -170,12 +170,16 @@ def gradio_scan_sp500(top_n, progress=gr.Progress()):
 
 iface = gr.Interface(
     fn=gradio_scan_sp500,
-    inputs=gr.Slider(minimum=1, maximum=100, step=1, label="Number of top stocks to display", value=25),
+    inputs=[
+        gr.Slider(minimum=1, maximum=100, step=1, label="Number of top stocks to display", value=25),
+        gr.Dropdown(choices=["1d", "1wk", "1mo"], label="Data Interval", value="1d"),
+    ],
     outputs="text",
     title="S&P 500 Stock Scanner",
     description="Scan S&P 500 stocks and display top N stocks based on today's candle score.",
     allow_flagging="never",
 )
 
+
 if __name__ == "__main__":
     iface.launch(server_name="0.0.0.0", server_port=7860)