Update ui/ui_core.py

ui/ui_core.py

@@ -108,10 +108,10 @@ def create_ui(agent: TxAgent):
             )
 
             try:
-                # Show loading status
-                yield gr.update(visible=True), history
+                # Show loading centered
+                yield [{"role": "assistant", "content": "⏳ Processing... Please wait while I analyze the files."}]
 
-                # Extract file text
+                # Step 1: Extract files
                 extracted_text = ""
                 if uploaded_files and isinstance(uploaded_files, list):
                     total_files = len(uploaded_files)
@@ -127,22 +127,20 @@ def create_ui(agent: TxAgent):
                             else:
                                 extracted_text += f"(Uploaded file: {os.path.basename(path)})\n"
                         except Exception as file_error:
-                            extracted_text += f"[Error processing file: {os.path.basename(path)}] — {str(file_error)}\n"
-                            continue
+                            extracted_text += f"[Error processing {os.path.basename(path)}]: {str(file_error)}\n"
 
-                # Sanitize and chunk
+                # Step 2: Chunk and sanitize
                 sanitized = sanitize_utf8(extracted_text.strip())
                 chunks = chunk_text(sanitized, max_tokens=8192)
 
                 all_responses = ""
-
                 for i, chunk in enumerate(chunks):
                     full_message = (
                         f"{context}\n\n--- Uploaded File Chunk {i+1}/{len(chunks)} ---\n\n{chunk}\n\n--- End of Chunk ---\n\nNow begin your reasoning:"
                     )
                     generator = agent.run_gradio_chat(
                         message=full_message,
-                        history=history,
+                        history=[],
                         temperature=0.3,
                         max_new_tokens=1024,
                         max_token=8192,
@@ -153,20 +151,24 @@ def create_ui(agent: TxAgent):
                     )
                     for update in generator:
                         if isinstance(update, str):
-                            all_responses += update  # collect only final string output
+                            all_responses += update
 
                 all_responses = sanitize_utf8(all_responses.strip())
 
-                # Convert to gr.Chatbot message format
-                history.append({"role": "user", "content": message})
-                history.append({"role": "assistant", "content": all_responses})
-                yield gr.update(visible=False), history
-
-            except Exception as chat_error:
-                print(f"Chat handling error: {chat_error}")
-                history.append({"role": "user", "content": message})
-                history.append({"role": "assistant", "content": "❌ An error occurred while processing your request. Please try again."})
-                yield gr.update(visible=False), history
+                # Final messages
+                final_history = history + [
+                    {"role": "user", "content": message},
+                    {"role": "assistant", "content": all_responses}
+                ]
+                yield final_history
+
+            except Exception as e:
+                print(f"[TxAgent Error] {e}")
+                final_history = history + [
+                    {"role": "user", "content": message},
+                    {"role": "assistant", "content": "⚠️ An error occurred while processing the files."}
+                ]
+                yield final_history
 
         inputs = [message_input, chatbot, conversation_state, file_upload]
         send_button.click(fn=handle_chat, inputs=inputs, outputs=[status_box, chatbot])