Support filtering by date range

src/datasource.py

@@ -23,6 +23,13 @@ if not ALPACA_API_KEY or not ALPACA_SECRET_KEY:
 requests_cache.install_cache("alpaca_api_cache", expire_after=120)
 
 
+def get_dataset_path(dataset_name="alpaca"):
+    if os.getenv("APP_ENV") == "development":
+        return f"{dataset_name}.parquet"
+    else:
+        return f"hf://datasets/Arrechenash/stocks/{dataset_name}.parquet"
+
+
 @st.cache_data
 def load_symbols(dataset):
     return (

src/main.py

@@ -1,16 +1,9 @@
 import os
+from datetime import date
 
 import streamlit as st
 
-from datasource import get_data, load_symbols
-
-
-def get_dataset_path(dataset_name):
-    if os.getenv("APP_ENV") == "development":
-        return f"{dataset_name}.parquet"
-    else:
-        return f"hf://datasets/Arrechenash/stocks/{dataset_name}.parquet"
-
+from datasource import get_data, get_dataset_path, load_symbols
 
 st.set_page_config(layout="wide")
 
@@ -40,19 +33,26 @@ with st.sidebar:
             get_dataset_path("yahoo"),
         ),
     )
-    st.session_state.date = st.date_input("Date", value=None)
+
+    use_date_filter = st.checkbox("Filter by date range", value=False)
+    if use_date_filter:
+        start_default = date.today().replace(day=1)
+        end_default = date.today()
+        st.session_state.date = st.date_input(
+            "Date range",
+            value=(start_default, end_default),
+        )
+    else:
+        st.session_state.date = None
+
     st.session_state.symbols = st.multiselect(
         "Symbols", load_symbols(st.session_state.dataset)
     )
     st.session_state.min_open = st.number_input(
         "Min open", value=st.session_state.min_open
     )
-    st.session_state.min_gap = st.number_input(
-        "Min gap %", value=st.session_state.min_gap
-    )
-    st.session_state.min_run = st.number_input(
-        "Min run %", value=st.session_state.min_run
-    )
+    st.session_state.min_gap = st.number_input("Min gap %", value=None)
+    st.session_state.min_run = st.number_input("Min run %", value=None)
     st.session_state.min_vol = st.number_input(
         "Min volume", value=st.session_state.min_vol
     )
@@ -60,12 +60,16 @@ with st.sidebar:
         "Min rel volume", value=st.session_state.min_relvol
     )
 
-    results_placeholder = st.empty()
+results_placeholder = st.empty()
 
 f = st.session_state
 filters = []
-if f.date:
-    filters.append(f"date = '{f.date.strftime('%Y-%m-%d')}'")
+if f.date and isinstance(f.date, tuple) and len(f.date) == 2:
+    start_date, end_date = f.date
+    filters.append(
+        f"date >= '{start_date.strftime('%Y-%m-%d')}' AND date <= '{end_date.strftime('%Y-%m-%d')}'"
+    )
+
 if f.symbols:
     filters.append(f"symbol IN {tuple(f.symbols)}")
 if f.min_open:
@@ -86,7 +90,7 @@ with results_placeholder:
     st.markdown(f"**Results: {len(df)}**")
 
 if df.empty:
-    st.info("No data found with current filters")
+    st.info("No data found with current filters.")
 else:
     numeric_cols = df.select_dtypes("number").columns.tolist()
 

src/pages/chart.py

@@ -1,4 +1,4 @@
-from datetime import timedelta
+from datetime import datetime, timedelta
 
 import pandas as pd
 import plotly.graph_objs as go
@@ -10,8 +10,12 @@ st.set_page_config(layout="wide")
 st.title("Candlestick Chart")
 st.sidebar.title("Filters")
 
-symbol = st.sidebar.text_input("Ticker symbol", value="HTOO").upper()
-date_start = st.sidebar.date_input("Start date", pd.Timestamp("2025-07-22"))
+symbol = st.sidebar.text_input("Ticker symbol", value="TSLA").upper()
+date_start = st.sidebar.date_input(
+    "Start date",
+    datetime.today() - timedelta(days=1),
+    max_value=datetime.today() - timedelta(days=1),
+)
 timeframe = st.sidebar.selectbox("Timeframe", options=["1", "5", "15"], index=2)
 
 try: