Update ui/ui_core.py

ui/ui_core.py

@@ -134,6 +134,9 @@ def create_ui(agent: TxAgent):
                 chunks = chunk_text(extracted_text.strip())
 
                 def process_chunk(i, chunk):
+                    if len(chunk.split()) > 8192:
+                        return f"[Chunk {i+1}] ⚠️ Skipped: input exceeds model limit."
+
                     chunked_prompt = (
                         f"{context}\n\n--- Uploaded File Content (Chunk {i+1}/{len(chunks)}) ---\n\n{chunk}\n\n"
                         f"--- End of Chunk ---\n\nNow begin your analysis:"
@@ -153,7 +156,6 @@ def create_ui(agent: TxAgent):
                         result = ""
                         for update in generator:
                             if update is None:
-                                print(f"[Warning] Empty response in chunk {i+1}")
                                 continue
                             if isinstance(update, str):
                                 result += update
@@ -166,10 +168,7 @@ def create_ui(agent: TxAgent):
                         print(f"[Error in chunk {i+1}] {err}")
                         return f"[Chunk {i+1}] ❌ Failed to process due to error."
 
-                with ThreadPoolExecutor(max_workers=min(8, len(chunks))) as executor:
-                    futures = [executor.submit(process_chunk, i, chunk) for i, chunk in enumerate(chunks)]
-                    results = [f.result() for f in as_completed(futures)]
-
+                results = [process_chunk(i, chunk) for i, chunk in enumerate(chunks)]
                 full_response = "\n\n".join(results)
                 full_response = clean_final_response(full_response.strip())
                 history[-1] = {"role": "assistant", "content": full_response}
@@ -190,4 +189,4 @@ def create_ui(agent: TxAgent):
             ["Is there anything abnormal in the attached blood work report?"]
         ], inputs=message_input)
 
-    return demo
+    return demo