Update app.py

app.py

@@ -186,10 +186,19 @@ class PDF(FPDF, HTMLMixin):
     pass
 
 def build_pdf(md: str, charts: Dict[str, str]) -> bytes:
+    # For PDF generation, we still need to convert file paths to something FPDF can use.
+    # Reading bytes and embedding as a data URI is a robust way to handle this.
+    def embed_chart_for_pdf(match):
+        desc = match.group("d").strip()
+        path = charts.get(desc)
+        if path and Path(path).exists():
+            with open(path, "rb") as f:
+                b64 = base64.b64encode(f.read()).decode()
+            return f'<img src="data:image/png;base64,{b64}" width="600">'
+        return ""
+
     html = MarkdownIt("commonmark", {"breaks": True}).enable("table").render(
-        TAG_RE.sub(
-            lambda m: f'<img src="{charts.get(m.group("d").strip(), "")}">', md
-        )
+        TAG_RE.sub(embed_chart_for_pdf, md)
     )
     pdf = PDF()
     pdf.set_auto_page_break(True, margin=15)
@@ -201,6 +210,7 @@ def build_pdf(md: str, charts: Dict[str, str]) -> bytes:
     pdf.write_html(html)
     return pdf.output(dest="S").encode("latin-1")
 
+
 # ─── QUICK STATIC CHART (fallback if LLM code fails) ───────────────────────
 def quick_chart(desc: str, df: pd.DataFrame, out: Path):
     ctype, *rest = [s.strip().lower() for s in desc.split("|", 1)]
@@ -697,11 +707,22 @@ if st.button("🚀 Generate", type="primary", disabled=not upl):
 # 1) live PDF reports (may be multiple)
 for rep_key, rep in st.session_state.lazy_reports.items():
     st.subheader("📄 Generated Report")
-    md_with_imgs = TAG_RE.sub(
-        lambda m: _inline_image_or_placeholder(rep, m.group("d").strip()), rep["md"]
-    )
     with st.expander("View Report", expanded=True):
-        st.markdown(md_with_imgs, unsafe_allow_html=True)
+        # Split markdown by chart tags to render images natively using st.image,
+        # which is more robust than embedding base64 in markdown.
+        parts = TAG_RE.split(rep["md"])
+        for i, part in enumerate(parts):
+            if i % 2 == 0:  # This is a text part
+                if part.strip():
+                    st.markdown(part, unsafe_allow_html=True)
+            else:  # This is a chart description captured from the tag
+                desc = part.strip()
+                chart_path = rep["charts"].get(desc)
+                if chart_path and Path(chart_path).exists():
+                    st.image(chart_path)
+                else:
+                    # Show a native Streamlit placeholder while the chart renders
+                    st.info(f"⏳ Rendering chart: '{desc}'...")
 
     if rep["finished"]:
         c1, c2 = st.columns(2)