Update app.py

app.py

@@ -305,7 +305,7 @@ Please output a JSON object with exactly two keys (no extra commentary):
         except Exception as e:
             logging.error(f"Error processing chunk {idx}: {e}. Raw result: {chunk_result}")
 
-    # --- Post-process the report to update the reference table ---
+    # --- Post-process the report to update the reference table without appending extra content ---
     references_prompt = f"""Review the following report HTML. 
 If any new inline citations (formatted as [x] where x is a number) have been introduced 
 that are not yet included in the references table at the end of the report, 
@@ -317,8 +317,21 @@ Report HTML:
 """
     updated_references = openai_call(prompt=references_prompt, model="o3-mini", max_tokens_param=1000, temperature=0.5)
     updated_references = updated_references.strip().strip("```")
-    # Append the updated reference table at the very end of the report.
-    updated_report_html += "\n" + updated_references
+    
+    # Instead of appending, check if a references section exists and replace its content.
+    soup_updated = BeautifulSoup(updated_report_html, "html.parser")
+    ref_heading = soup_updated.find(lambda tag: tag.name == "h1" and "Reference Summary Table" in tag.get_text())
+    if ref_heading:
+        # Find the next sibling (which should be the references table or container)
+        next_sibling = ref_heading.find_next_sibling()
+        if next_sibling:
+            new_ref_html = BeautifulSoup(updated_references, "html.parser")
+            next_sibling.replace_with(new_ref_html)
+        # Update the working HTML string
+        updated_report_html = str(soup_updated)
+    else:
+        # If no reference section is found, do nothing.
+        logging.info("No existing reference table found; skipping reference replacement.")
 
     summary_text = "Summary of Fine-Tuning Improvements:\n" + "\n".join(improvements_summary)
     global_summary = "Combined Chunk Improvement Guidelines:\n" + "\n".join(all_guidelines)