Update app.py

app.py

@@ -85,9 +85,11 @@ def process_text(text):
         start_time = normalize_data_time(start_str)
         end_time = normalize_data_time(end_str)
     else:
-        # If no range, use the first TIME entity as start time
-        if entities["TIME"]:
+         # If no range, use the first and second TIME entities
+        if len(entities["TIME"]) >= 1:
             start_time = normalize_data_time(entities["TIME"][0])
+        if len(entities["TIME"]) >= 2:
+            end_time = normalize_data_time(entities["TIME"][1])
 
     # Calculate duration if both start and end times are available
     if start_time and end_time: