Update app.py

app.py

@@ -83,13 +83,14 @@ def init_agent() -> TxAgent:
     agent.init_model()
     return agent
 
-def analyze_serial(agent, chunks: List[str]) -> List[str]:
-    results = []
-    for i, chunk in enumerate(chunks):
+# ⚡ Optimized parallel analysis
+def analyze_parallel(agent, chunks: List[str], max_workers: int = 6) -> List[str]:
+    results = [None] * len(chunks)
+
+    def worker(index, chunk):
         prompt = build_prompt(chunk)
         if estimate_tokens(prompt) > MAX_MODEL_TOKENS:
-            results.append(f"❌ Chunk {i+1} too long. Skipped.")
-            continue
+            return f"❌ Chunk {index+1} too long. Skipped."
         response = ""
         try:
             for r in agent.run_gradio_chat(
@@ -109,10 +110,20 @@ def analyze_serial(agent, chunks: List[str]) -> List[str]:
                             response += m.content
                 elif hasattr(r, "content"):
                     response += r.content
-            gc.collect()
-            results.append(clean_response(response))
+            return clean_response(response)
         except Exception as e:
-            results.append(f"❌ Error in chunk {i+1}: {str(e)}")
+            return f"❌ Error in chunk {index+1}: {str(e)}"
+
+    with ThreadPoolExecutor(max_workers=max_workers) as executor:
+        futures = {executor.submit(worker, idx, chunk): idx for idx, chunk in enumerate(chunks)}
+        for future in futures:
+            idx = futures[future]
+            try:
+                results[idx] = future.result()
+            except Exception as e:
+                results[idx] = f"❌ Error in chunk {idx+1}: {str(e)}"
+
+    gc.collect()
     return results
 
 def generate_final_summary(agent, combined: str) -> str:
@@ -148,7 +159,7 @@ def process_report(agent, file, messages: List[Dict[str, str]]) -> Tuple[List[Di
         chunks = split_text(extracted)
         messages.append({"role": "assistant", "content": f"🔍 Split into {len(chunks)} chunks. Analyzing..."})
 
-        chunk_results = analyze_serial(agent, chunks)
+        chunk_results = analyze_parallel(agent, chunks, max_workers=6)
         valid = [res for res in chunk_results if not res.startswith("❌")]
 
         if not valid: