Update ui/ui_core.py

ui/ui_core.py

@@ -21,19 +21,18 @@ 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).encode("utf-8", errors="replace").decode("utf-8")
+            return tabulate(df, headers="keys", tablefmt="github", showindex=False)
 
         result = []
         for group_name, group_df in groups:
             result.append(f"\n### Group: {group_name}\n")
-            formatted = tabulate(group_df, headers="keys", tablefmt="github", showindex=False)
-            result.append(formatted.encode("utf-8", errors="replace").decode("utf-8"))
-
+            result.append(tabulate(group_df, headers="keys", tablefmt="github", showindex=False))
         return "\n".join(result)
 
     except Exception as e:
@@ -52,7 +51,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).encode("utf-8", errors="replace").decode("utf-8")
+        return "\n".join(extracted)
     except Exception as e:
         return f"Error parsing PDF: {e}"
 
@@ -97,6 +96,7 @@ def create_ui(agent: TxAgent):
 
                 message = f"{context}\n\n---\n{extracted_text.strip()}\n---\n\nBegin your reasoning."
 
+            final_response = None
             generator = agent.run_gradio_chat(
                 message=message,
                 history=history,
@@ -109,7 +109,14 @@ def create_ui(agent: TxAgent):
                 max_round=30
             )
             for update in generator:
-                yield update
+                if isinstance(update, list):
+                    cleaned = [msg for msg in update if not (msg.role == "assistant" and msg.content.strip().startswith("🧰"))]
+                    if cleaned:
+                        final_response = cleaned
+                        yield cleaned
+                else:
+                    if isinstance(update, str) and not update.strip().startswith("🧰"):
+                        yield update
 
         inputs = [message_input, chatbot, conversation_state, file_upload]
         send_button.click(fn=handle_chat, inputs=inputs, outputs=chatbot)