Update ui/ui_core.py

ui/ui_core.py

@@ -21,18 +21,19 @@ def extract_all_text_from_csv_or_excel(file_path, progress=None, index=0, total=
         if progress:
             progress((index + 1) / total, desc=f"Processed table: {os.path.basename(file_path)}")
 
-        # Group by "Booking Number" or "Form Name" if available
         if "Booking Number" in df.columns:
             groups = df.groupby("Booking Number")
         elif "Form Name" in df.columns:
             groups = df.groupby("Form Name")
         else:
-            return tabulate(df, headers="keys", tablefmt="github", showindex=False)
+            return tabulate(df, headers="keys", tablefmt="github", showindex=False).encode("utf-8", errors="replace").decode("utf-8")
 
         result = []
         for group_name, group_df in groups:
             result.append(f"\n### Group: {group_name}\n")
-            result.append(tabulate(group_df, headers="keys", tablefmt="github", showindex=False))
+            formatted = tabulate(group_df, headers="keys", tablefmt="github", showindex=False)
+            result.append(formatted.encode("utf-8", errors="replace").decode("utf-8"))
+
         return "\n".join(result)
 
     except Exception as e:
@@ -51,7 +52,7 @@ def extract_all_text_from_pdf(file_path, progress=None, index=0, total=1):
                             extracted.append("\t".join([cell or "" for cell in row]))
                 if progress:
                     progress((index + i / num_pages) / total, desc=f"Parsing PDF: {os.path.basename(file_path)} ({i+1}/{num_pages})")
-        return "\n".join(extracted)
+        return "\n".join(extracted).encode("utf-8", errors="replace").decode("utf-8")
     except Exception as e:
         return f"Error parsing PDF: {e}"
 
@@ -120,4 +121,4 @@ def create_ui(agent: TxAgent):
             ["Is there anything abnormal in the attached blood work report?"]
         ], inputs=message_input)
 
-    return demo
+    return demo