Merge pull request #8 from alexoh2bd/Natalie

src/streamlit_app.py

@@ -161,7 +161,7 @@ def main():
     # Use custom query if provided
     final_query = custom_query if custom_query else selected_query
     
-    # Time range
+    # Time range (days)
     days = st.sidebar.slider(
         "📅 Days to analyze:",
         min_value=1,
@@ -169,6 +169,11 @@ def main():
         value=7,
         help="How many days back to search for news"
     )
+
+    # Date range filter (optional, after data is loaded)
+    st.sidebar.markdown("---")
+    st.sidebar.markdown("#### Optional: Filter by Date Range")
+    date_range = None
     
     # News sources from config
     news_sources = config["news_sources"]
@@ -212,6 +217,23 @@ def main():
     if 'df' in st.session_state and not st.session_state.df.empty:
         df = st.session_state.df
 
+        # Date range filter UI (if date column exists)
+        if 'published_at' in df.columns:
+            min_date = pd.to_datetime(df['published_at']).min().date()
+            max_date = pd.to_datetime(df['published_at']).max().date()
+            start_date, end_date = st.sidebar.date_input(
+                "Select date range:",
+                value=(min_date, max_date),
+                min_value=min_date,
+                max_value=max_date
+            )
+            # Filter DataFrame by date range
+            mask = (pd.to_datetime(df['published_at']).dt.date >= start_date) & (pd.to_datetime(df['published_at']).dt.date <= end_date)
+            df = df.loc[mask]
+            if df.empty:
+                st.warning("No articles found in the selected date range.")
+                st.stop()
+
         # ===== Summary Metrics =====
         st.markdown("### 📊 Analysis Summary")
         col1, col2, col3, col4 = st.columns(4)