Update load_files.py

load_files.py

@@ -93,6 +93,28 @@ class ImageGalleryApp:
                     print(f"Error in parallel loading: {e}")
         return images
 
+    def parse_date_input(self, date_input) -> Optional[date]:
+        """Convert string date input to date object"""
+        if not date_input or date_input == "":
+            return None
+        
+        if isinstance(date_input, date):
+            return date_input
+        
+        if isinstance(date_input, str):
+            try:
+                # Handle YYYY-MM-DD format (most common from Calendar component)
+                if date_input.count('-') == 2:
+                    return datetime.strptime(date_input, '%Y-%m-%d').date()
+                # Handle MM/DD/YYYY format
+                elif date_input.count('/') == 2:
+                    return datetime.strptime(date_input, '%m/%d/%Y').date()
+            except ValueError as e:
+                print(f"Error parsing date string '{date_input}': {e}")
+                return None
+        
+        return None
+
     def search_images(self,
                       category: str = "All",
                       file_name: str = "All",
@@ -101,6 +123,10 @@ class ImageGalleryApp:
                       lob: str = "test") -> Tuple[List[Image.Image], str]:
         """Search images based on filters with optimizations"""
 
+        # Convert string inputs to date objects
+        start_date_obj = self.parse_date_input(start_date)
+        end_date_obj = self.parse_date_input(end_date)
+
         # Build query with default status filter
         query = {
             "lob": lob,
@@ -117,18 +143,18 @@ class ImageGalleryApp:
         # Handle date filtering with proper date objects
         if start_date or end_date:
             date_query = {}
-            if start_date:
+            if start_date_obj:
                 try:
                     # Convert date to datetime at start of day
-                    start_dt = datetime.combine(start_date, datetime.min.time())
+                    start_dt = datetime.combine(start_date_obj, datetime.min.time())
                     date_query["$gte"] = start_dt
                 except Exception as e:
                     print(f"Error parsing start date: {e}")
 
-            if end_date:
+            if end_date_obj:
                 try:
                     # Convert date to datetime at end of day
-                    end_dt = datetime.combine(end_date, datetime.max.time())
+                    end_dt = datetime.combine(end_date_obj, datetime.max.time())
                     date_query["$lte"] = end_dt
                 except Exception as e:
                     print(f"Error parsing end date: {e}")
@@ -172,7 +198,12 @@ class ImageGalleryApp:
 
             # Build info text
             info_text = f"Found {len(documents)} records (showing {len(images)} images)\n"
-            info_text += f"Filter: Status = completed, LOB = {lob}\n\n"
+            info_text += f"Filter: Status = completed, LOB = {lob}\n"
+            if start_date_obj:
+                info_text += f"Start Date: {start_date_obj}\n"
+            if end_date_obj:
+                info_text += f"End Date: {end_date_obj}\n"
+            info_text += "\n"
 
             for i, doc in enumerate(documents[:len(images)]):
                 info_text += f"#{i + 1}\n"