Update app.py

app.py

@@ -155,12 +155,10 @@ def update_table(df: pd.DataFrame, query: str, param_ranges: list, is_foundation
     columns = columns + [col for col in additional_cols if col in ADDITIONAL_COLS]
     
     # Ensure date columns are sorted as dates and then formatted as strings
-    if 'Release Date' in columns:
-        filtered_df.loc[:, 'Release Date'] = pd.to_datetime(filtered_df['Release Date'], errors='coerce')
-        filtered_df.loc[:, 'Release Date'] = filtered_df['Release Date'].dt.strftime('%Y-%m-%d')
-    if 'Date Added' in columns:
-        filtered_df.loc[:, 'Date Added'] = pd.to_datetime(filtered_df['Date Added'], errors='coerce')
-        filtered_df.loc[:, 'Date Added'] = filtered_df['Date Added'].dt.strftime('%Y-%m-%d')
+    for date_col in ['Release Date', 'Date Added']:
+        if date_col in columns:
+            filtered_df[date_col] = pd.to_datetime(filtered_df[date_col], errors='coerce')
+            filtered_df[date_col] = filtered_df[date_col].apply(lambda x: x.strftime('%Y-%m-%d') if pd.notnull(x) else '')
     
     return filtered_df[columns]