Only remove Total flex when it is the last column and a flex column exists

Signed-off-by: krishna2323 <belivethatkg@gmail.com>

Author: Krishna2323

src/libs/SearchUIUtils.ts

@@ -5768,8 +5768,24 @@ const FLEX_COLUMNS = new Set<string>([
     CONST.SEARCH.TABLE_COLUMNS.TAX_RATE,
 ]);
 
+const TOTAL_COLUMNS = new Set<string>([
+    CONST.SEARCH.TABLE_COLUMNS.TOTAL,
+    CONST.SEARCH.TABLE_COLUMNS.TOTAL_AMOUNT,
+    CONST.SEARCH.TABLE_COLUMNS.TOTAL_PER_ATTENDEE,
+    CONST.SEARCH.TABLE_COLUMNS.ORIGINAL_AMOUNT,
+]);
+
+/**
+ * Returns true when it's safe to remove flex from total columns — i.e., another flex column
+ * exists to fill the space, and the total column is the last visible column.
+ */
 function hasFlexColumn(columns?: SearchColumnType[]): boolean {
-    return !!columns?.some((col) => FLEX_COLUMNS.has(col));
+    if (!columns?.length) {
+        return false;
+    }
+    const lastColumn = columns.at(-1);
+    const isTotalLastColumn = !!lastColumn && TOTAL_COLUMNS.has(lastColumn);
+    return isTotalLastColumn && columns.some((col) => FLEX_COLUMNS.has(col));
 }
 
 export {