guardrails for time

app.py

@@ -3,6 +3,65 @@ import pandas as pd
 import numpy as np
 from sklearn.metrics import roc_auc_score, precision_score, recall_score
 from pandas.tseries.offsets import BDay
+import streamlit as st
+from datetime import datetime, time, timedelta
+import pytz
+import holidays
+
+# Function to convert UTC to PST
+def convert_to_pst(utc_time):
+    utc_timezone = pytz.timezone('UTC')
+    pst_timezone = pytz.timezone('America/Los_Angeles')
+    utc_time = utc_timezone.localize(utc_time)
+    pst_time = utc_time.astimezone(pst_timezone)
+    return pst_time
+
+# Get the current UTC time
+utc_now = datetime.utcnow()
+
+# Convert UTC to PST
+pst_now = convert_to_pst(utc_now)
+
+# Function to check if a date is a business day (Monday to Friday)
+def is_business_day(date):
+    return date.weekday() < 5  # Monday = 0, Friday = 4
+
+# Function to get the list of holidays in the US for the current year
+def get_us_holidays():
+    return holidays.US(years=datetime.now().year)
+
+# Function to convert the time string to a datetime object
+def convert_to_datetime(time_str, pst_now):
+    time_obj = datetime.strptime(time_str, "%H:%M")
+    # Combine the time with the current date to get the full datetime in PST
+    return datetime.combine(pst_now.date(), time_obj.time())
+
+# Function to determine the emoji for each time
+def get_time_emojis(times, pst_now):
+    pst_now_time = pst_now.time()
+    us_holidays = get_us_holidays()
+    emojis = []
+    
+    for t in times:
+        time_obj = convert_to_datetime(t, pst_now)
+        
+        # Check if the time is less than or equal to the current time in PST
+        if time_obj.time() <= pst_now_time:
+            emoji = "✅"
+        else:
+            # Check if it's a business day or a holiday
+            if is_business_day(time_obj) or time_obj.date() in us_holidays:
+                emoji = "⛔"
+            else:
+                emoji = "✅"
+        
+        emojis.append(emoji)
+    
+    return emojis
+
+# Example usage:
+times_list = ['06:30', '07:00', '07:30', '08:00']
+emojis_list = get_time_emojis(times_list, pst_now)
 
 st.set_page_config(
     page_title="Gameday $SPX",
@@ -21,9 +80,11 @@ with st.form("choose_model"):
         
     with col1:
         option = st.select_slider(
-            'Slide the scale based on PST, then run.',
-            ['06:30', '07:00', '07:30', '08:00']
+            f"""Slide the scale based on PST, then run [{pst_now.strftime('%H:%M')}]""",
+            times_list,
+            format_func=lambda x: f"{emojis_list[times_list.index(x)]} {x}"
         )
+
     with col2:
         submitted = st.form_submit_button('🏃🏽‍♂️ Run',use_container_width=True)
         cleared = st.form_submit_button('🧹 Clear All',use_container_width=True)
@@ -501,6 +562,9 @@ with st.form("choose_model"):
             f"{len(data.query('High > VIX_EM_15_High | Low < VIX_EM_15_Low')) / len(data):.1%}"
             ]
         
+        # Cache all DFs
+        all_dfs = []
+
         st.dataframe(top_of_fold, use_container_width=True)
 
         tab1, tab2, tab3, tab4 = st.tabs(["🤖 Stats", "✨ New Data", "📚 Historical", "📊 Performance"])

requirements.txt

@@ -14,4 +14,6 @@ streamlit==1.24.1
 scikit-learn==1.3.0
 scipy==1.9.3
 datasets
-huggingface_hub
+huggingface_hub
+holidays
+pytz