Update app.py

app.py

@@ -453,7 +453,7 @@ def remove_text_from_html(report_html: str, text_to_remove: str) -> str:
 
 def fix_visual_after_section(report_html: str, section_title: str, extra_instructions: str) -> str:
     """
-    Given a report HTML, a target section name (extracted from within [[...]]), and extra instructions,
+    Given a report HTML, a target section name (from inside [[...]]), and extra instructions,
     this function finds the first header (h1–h4) that contains the section name (ignoring common TOC headers),
     then finds the first <iframe> following that header.
     It sends the current iframe 'srcdoc' (which should contain a mermaid diagram)
@@ -491,7 +491,8 @@ Return only the improved mermaid code (the content that should go inside the <di
 Ensure that the output starts with one of the following keywords: mindmap, flowchart, sequence, gantt, or pie."""
     
     improved_code = llm_call(prompt=prompt, model="o3-mini", temperature=0, max_tokens_param=1500)
-    improved_code = improved_code.strip().strip("```").strip()
+    # Remove code fence markers and all leading whitespace
+    improved_code = improved_code.replace("```", "").lstrip()
     logging.info(f"fix_visual_after_section: Improved code received from LLM:\n{improved_code}")
     
     # Verify that the result starts with a valid mermaid diagram keyword.