Update app.py

app.py

@@ -37,7 +37,7 @@ from txagent.txagent import TxAgent
 MAX_MODEL_TOKENS = 131072
 MAX_NEW_TOKENS = 4096
 MAX_CHUNK_TOKENS = 8192
-BATCH_SIZE = 2
+BATCH_SIZE = 1
 PROMPT_OVERHEAD = 300
 SAFE_SLEEP = 0.5
 
@@ -263,8 +263,8 @@ def generate_pdf_report_with_charts(summary: str, report_path: str, detailed_bat
         pdf.ln(5)
         for idx, detail in enumerate(detailed_batches):
             pdf.multi_cell(0, 10, txt=f"Tool Result #{idx + 1}", align="L")
-            for line in remove_non_ascii(detail).split("\n"):
-
+            for line in remove_non_ascii(detail).split("
+"):
                 pdf.multi_cell(0, 10, txt=line)
             pdf.ln(5)
     pdf.output(pdf_path)
@@ -289,6 +289,7 @@ def process_report(agent, file, messages: List[Dict[str, str]]) -> Tuple[List[Di
         messages.append({"role": "assistant", "content": f"🔍 Split into {len(batches)} batches. Analyzing..."})
 
         batch_results = analyze_batches(agent, batches)
+        all_tool_outputs = batch_results.copy()
         valid = [res for res in batch_results if not res.startswith("❌")]
 
         if not valid:
@@ -301,7 +302,7 @@ def process_report(agent, file, messages: List[Dict[str, str]]) -> Tuple[List[Di
         with open(report_path, 'w', encoding='utf-8') as f:
             f.write(f"# Final Medical Report\n\n{summary}")
 
-        pdf_path = generate_pdf_report_with_charts(summary, report_path, detailed_batches=valid)
+        pdf_path = generate_pdf_report_with_charts(summary, report_path, detailed_batches=all_tool_outputs)
 
         end_time = time.time()
         elapsed_time = end_time - start_time