add date picker using dropdown

app.py

@@ -2,6 +2,8 @@ import pandas as pd
 import gradio as gr
 import os
 from gradio_rangeslider import RangeSlider
+import calendar
+import datetime
 
 from src.filter_utils import filter, filter_cols
 from src.process_data import merge_data
@@ -12,7 +14,8 @@ text_leaderboard = pd.read_csv(os.path.join('assets', 'merged_data.csv'))
 text_leaderboard = text_leaderboard.sort_values(by=tc.CLEMSCORE, ascending=False)  
 
 # When displaying latency values
-text_leaderboard[tc.LATENCY] = text_leaderboard[tc.LATENCY].round(3)
+text_leaderboard[tc.LATENCY] = text_leaderboard[tc.LATENCY].round(1)
+text_leaderboard[tc.CLEMSCORE] = text_leaderboard[tc.CLEMSCORE].round(1)
 
 open_weight_df = text_leaderboard[text_leaderboard[tc.OPEN_WEIGHT] == True]
 if not open_weight_df.empty:  # Check if filtered df is non-empty
@@ -67,8 +70,16 @@ context_step = 8
 min_date = min(dates)
 max_date = max(dates)
 
-TITLE = tc.TITLE
+# Date settings
+today = datetime.date.today()
+end_year = today.year
+start_year = tc.START_YEAR
+
+YEARS = list(range(int(start_year), int(end_year)+1))
+YEARS = [str(y) for y in YEARS] 
+MONTHS = list(calendar.month_name[1:])
 
+TITLE = tc.TITLE
 
 llm_calc_app = gr.Blocks()
 with llm_calc_app:
@@ -92,17 +103,30 @@ with llm_calc_app:
                 )
 
             with gr.Row():
-                start_date = gr.DateTime(
-                    value=min_date,
-                    type="string",
-                    label="Release Date 📅 | Start Date"
-                )
+                with gr.Column():
+                    start_year = gr.Dropdown(
+                        choices=YEARS,
+                        value=[YEARS[0]],
+                        allow_custom_value=True
+                    )
+                    start_month = gr.Dropdown(
+                        choices=MONTHS,
+                        value=[MONTHS[0]],
+                        allow_custom_value=True
+                    )
+
+                with gr.Column():
+                    end_year = gr.Dropdown(
+                        choices=YEARS,
+                        value=[YEARS[-1]],
+                        allow_custom_value=True
+                    )
+                    end_month = gr.Dropdown(
+                        choices=MONTHS,
+                        value=[MONTHS[-1]],
+                        allow_custom_value=True
+                    )
 
-                end_date = gr.DateTime(
-                    value=max_date,
-                    type="string",
-                    label="End Date"
-                )
 
             # Multiodality Select
             with gr.Row():
@@ -201,7 +225,7 @@ with llm_calc_app:
             filter,
             [dummy_leaderboard_table, lang_dropdown, parameter_slider,
              input_pricing_slider, output_pricing_slider, multimodal_checkbox,
-             context_slider, open_weight_checkbox, start_date, end_date, license_checkbox],
+             context_slider, open_weight_checkbox, start_year, start_month, end_year, end_month, license_checkbox],
             [leaderboard_table],   
             queue=True
         )
@@ -210,7 +234,7 @@ with llm_calc_app:
             filter,
             [dummy_leaderboard_table, lang_dropdown, parameter_slider,
              input_pricing_slider, output_pricing_slider, multimodal_checkbox,
-             context_slider, open_weight_checkbox, start_date, end_date, license_checkbox],
+             context_slider, open_weight_checkbox, start_year, start_month, end_year, end_month, license_checkbox],
             [leaderboard_table],
             queue=True
         )
@@ -219,7 +243,7 @@ with llm_calc_app:
             filter,
             [dummy_leaderboard_table, lang_dropdown, parameter_slider,
              input_pricing_slider, output_pricing_slider, multimodal_checkbox,
-             context_slider, open_weight_checkbox, start_date, end_date, license_checkbox],
+             context_slider, open_weight_checkbox, start_year, start_month, end_year, end_month, license_checkbox],
             [leaderboard_table],
             queue=True
         )
@@ -228,7 +252,7 @@ with llm_calc_app:
             filter,
             [dummy_leaderboard_table, lang_dropdown, parameter_slider,
              input_pricing_slider, output_pricing_slider, multimodal_checkbox,
-             context_slider, open_weight_checkbox, start_date, end_date, license_checkbox],
+             context_slider, open_weight_checkbox, start_year, start_month, end_year, end_month, license_checkbox],
             [leaderboard_table],
             queue=True
         )
@@ -237,7 +261,7 @@ with llm_calc_app:
             filter,
             [dummy_leaderboard_table, lang_dropdown, parameter_slider,
              input_pricing_slider, output_pricing_slider, multimodal_checkbox,
-             context_slider, open_weight_checkbox, start_date, end_date, license_checkbox],
+             context_slider, open_weight_checkbox, start_year, start_month, end_year, end_month, license_checkbox],
             [leaderboard_table],
             queue=True
         )
@@ -246,7 +270,7 @@ with llm_calc_app:
             filter,
             [dummy_leaderboard_table, lang_dropdown, parameter_slider,
              input_pricing_slider, output_pricing_slider, multimodal_checkbox,
-             context_slider, open_weight_checkbox, start_date, end_date, license_checkbox],
+             context_slider, open_weight_checkbox, start_year, start_month, end_year, end_month, license_checkbox],
             [leaderboard_table],
             queue=True
         )
@@ -255,25 +279,43 @@ with llm_calc_app:
             filter,
             [dummy_leaderboard_table, lang_dropdown, parameter_slider,
              input_pricing_slider, output_pricing_slider, multimodal_checkbox,
-             context_slider, open_weight_checkbox, start_date, end_date, license_checkbox],
+             context_slider, open_weight_checkbox, start_year, start_month, end_year, end_month, license_checkbox],
+            [leaderboard_table],
+            queue=True
+        )
+
+        start_year.change(
+            filter,
+            [dummy_leaderboard_table, lang_dropdown, parameter_slider,
+             input_pricing_slider, output_pricing_slider, multimodal_checkbox,
+             context_slider, open_weight_checkbox, start_year, start_month, end_year, end_month, license_checkbox],
+            [leaderboard_table],
+            queue=True
+        )
+
+        start_month.change(
+            filter,
+            [dummy_leaderboard_table, lang_dropdown, parameter_slider,
+             input_pricing_slider, output_pricing_slider, multimodal_checkbox,
+             context_slider, open_weight_checkbox, start_year, start_month, end_year, end_month, license_checkbox],
             [leaderboard_table],
             queue=True
         )
 
-        start_date.change(
+        end_year.change(
             filter,
             [dummy_leaderboard_table, lang_dropdown, parameter_slider,
              input_pricing_slider, output_pricing_slider, multimodal_checkbox,
-             context_slider, open_weight_checkbox, start_date, end_date, license_checkbox],
+             context_slider, open_weight_checkbox, start_year, start_month, end_year, end_month, license_checkbox],
             [leaderboard_table],
             queue=True
         )
 
-        end_date.change(
+        end_month.change(
             filter,
             [dummy_leaderboard_table, lang_dropdown, parameter_slider,
              input_pricing_slider, output_pricing_slider, multimodal_checkbox,
-             context_slider, open_weight_checkbox, start_date, end_date, license_checkbox],
+             context_slider, open_weight_checkbox, start_year, start_month, end_year, end_month, license_checkbox],
             [leaderboard_table],
             queue=True
         )
@@ -282,7 +324,7 @@ with llm_calc_app:
             filter,
             [dummy_leaderboard_table, lang_dropdown, parameter_slider,
              input_pricing_slider, output_pricing_slider, multimodal_checkbox,
-             context_slider, open_weight_checkbox, start_date, end_date, license_checkbox],
+             context_slider, open_weight_checkbox, start_year, start_month, end_year, end_month, license_checkbox],
             [leaderboard_table],
             queue=True
         )

assets/text_content.py

@@ -36,4 +36,7 @@ TEMP_DATE = "Temp Date"
 OPEN = "Open-Weight"
 COMM = "Commercial"
 
-TITLE = """<h1 align="center" id="space-title"> LLM Calculator ⚖️⚡ 📏💰</h1>"""
+TITLE = """<h1 align="center" id="space-title"> LLM Calculator ⚖️⚡ 📏💰</h1>"""
+
+# Date Picker (set as Dropdown until datetime object is fixed)
+START_YEAR = "2020"

src/filter_utils.py

@@ -19,9 +19,53 @@ def filter_cols(df):
     
     return df
 
+def convert_date_components_to_timestamp(year: str, month: str) -> int:
+    """Convert year and month strings to timestamp."""
+    # Create a datetime object for the first day of the month
+    date_str = f"{year}-{month:02d}-01"
+    return int(pd.to_datetime(date_str).timestamp())
+
+def filter_by_date(df: pd.DataFrame, 
+                  start_year: str, 
+                  start_month: str,
+                  end_year: str,
+                  end_month: str,
+                  date_column: str) -> pd.DataFrame:
+    """
+    Filter DataFrame by date range using separate year and month components.
+    
+    Args:
+        df: DataFrame to filter
+        start_year: Starting year (e.g., "2023")
+        start_month: Starting month (e.g., "1" for January)
+        end_year: Ending year (e.g., "2024")
+        end_month: Ending month (e.g., "12" for December)
+        date_column: Name of the date column to filter on
+    """
+    # Convert string inputs to integers for date creation
+    start_timestamp = convert_date_components_to_timestamp(
+        int(start_year), 
+        int(start_month)
+    )
+    
+    end_timestamp = convert_date_components_to_timestamp(
+        int(end_year), 
+        int(end_month)
+    )
+    
+    # Convert the DataFrame's date column to timestamps for comparison
+    date_timestamps = pd.to_datetime(df[date_column]).apply(lambda x: int(x.timestamp()))
+    
+    # Filter the DataFrame
+    return df[
+        (date_timestamps >= start_timestamp) & 
+        (date_timestamps <= end_timestamp)
+    ]
+    
+
 
 def filter(df, language_list, parameters, input_price, output_price, multimodal,
-           context, open_weight, start, end, license ):
+           context, open_weight, start_year, start_month, end_year, end_month, license ):
     
 
     if not df.empty:  # Check if df is non-empty
@@ -81,20 +125,23 @@ def filter(df, language_list, parameters, input_price, output_price, multimodal,
     if not df.empty:  # Check if df is non-empty
         df = df[df[tc.LICENSE_NAME].apply(lambda x: any(lic in x for lic in license))]
 
-    # Convert 'Release Date' to int temporarily
-    if not df.empty:  # Check if df is non-empty
-        df[tc.TEMP_DATE] = pd.to_datetime(df[tc.TEMP_DATE]).astype(int) // 10**9  # Convert to seconds since epoch
+    # # Convert 'Release Date' to int temporarily
+    # if not df.empty:  # Check if df is non-empty
+    #     df[tc.TEMP_DATE] = pd.to_datetime(df[tc.TEMP_DATE]).astype(int) // 10**9  # Convert to seconds since epoch
 
-    # Convert start and end to int (seconds since epoch)
-    start = int(pd.to_datetime(start).timestamp())  
-    end = int(pd.to_datetime(end).timestamp())    
+    # # Convert start and end to int (seconds since epoch)
+    # start = int(pd.to_datetime(start).timestamp())  
+    # end = int(pd.to_datetime(end).timestamp())    
 
-    # Filter based on the converted 'Release Date'
-    if not df.empty:  # Check if df is non-empty
-        df = df[(df[tc.TEMP_DATE] >= start) & (df[tc.TEMP_DATE] <= end)]
+    # # Filter based on the converted 'Release Date'
+    # if not df.empty:  # Check if df is non-empty
+    #     df = df[(df[tc.TEMP_DATE] >= start) & (df[tc.TEMP_DATE] <= end)]
+
+    df = filter_by_date(df, start_year, start_month, end_year, end_month, tc.TEMP_DATE)
 
     df = filter_cols(df)
     df = df.sort_values(by=tc.CLEMSCORE, ascending=False)
 
     return df  # Return the filtered dataframe
     
+

test.py

@@ -0,0 +1,10 @@
+import calendar
+import datetime
+
+today = datetime.date.today()
+
+year = today.year
+
+print(year)
+
+print(list(calendar.month_name[1:]))