4

src/main.py

@@ -1,7 +1,7 @@
 from datetime import datetime
 import json
 from data_classes import Event, TimeWindow
-from utils import filter_event_by_date_facility_id, find_max_continuous_sequence, loadDataFromJson, twenty_four_hours_bucket_time_series_by_minute
+from utils import filter_event_by_date_facility_id, find_max_continuous_sequence, get_all_facility_ids, loadDataFromJson, twenty_four_hours_bucket_time_series_by_minute
 import gradio as gr
 from gradio_calendar import Calendar
 
@@ -28,10 +28,11 @@ def process1(target_date, target_facility_id):
     result = find_most_occupied_window(target_date, target_facility_id) 
     return "Most occupied window is between {} and {} with {} occupied spots".format(result[0][0],result[0][1],result[1])
 
+all_facility_ids = get_all_facility_ids()
 with gr.Blocks() as app:
     with gr.Group():
         target_date = Calendar(label="Target Date",value=target_date)
-        target_facility_id = gr.Number(label="Facility ID", value=target_facility_id)
+        target_facility_id = gr.Dropdown(label="Facility ID",choices=all_facility_ids,value=all_facility_ids[0])
         btn_1 = gr.Button(value="Find Most Occupied Window")
     output1 = gr.Text(label="Most Occupied Window",show_copy_button = True)
     btn_1.click(process1, [target_date, target_facility_id], outputs=output1)

src/utils.py

@@ -1,7 +1,8 @@
 import time
 from data_classes import Event, TimeWindow
-import datetime
 import json
+from datetime import datetime
+data = None
 
 def twenty_four_hours_bucket_time_series_by_minute(valid_events:list[Event],target_date:datetime)->(list[int],list[list[Event]]): # type: ignore
     #create a pre defined time window for 24 hours by minute precision
@@ -56,13 +57,24 @@ def find_max_continuous_sequence(x):
     return (max_element, start, end)
 
 def loadDataFromJson():
-    with open('data/events.json') as f:
-        data = json.load(f)
+    global data
+    if data is None:
+        with open('data/events.json') as f:
+            data = json.load(f)
     return data
 
+
 def filter_event_by_date_facility_id(event:Event,target_date:datetime, facility_id:int):
     if event.facility_id == facility_id: 
         if event.entrance_time.date() <= target_date.date() and event.exit_time.date() >= target_date.date():
             return True
     else:
-        return False
+        return False
+
+def get_all_entrance_dates()->list[datetime]:
+    data = loadDataFromJson()
+    return [datetime.fromisoformat(event['entrance_time']) for event in data]
+
+def get_all_facility_ids()->list[int]:
+    data = loadDataFromJson()
+    return list(set([event['facility_id'] for event in data]))