Update app.py

app.py

@@ -1,8 +1,8 @@
 ##############################################################################
 # Sozo Business Studio · 10-Jul-2025
-# • REFACTORED: Removed lazy-loading to ensure stability on Streamlit.
-# • Report generation is now a single, synchronous process.
-# • Unified output under a single `st.session_state.bundle` for both modes.
+# • REFACTORED: Implemented a robust rendering loop to fix image display issues.
+# • Uses st.image() for charts instead of embedding HTML in Markdown.
+# • This guarantees rendering on platforms like Hugging Face Spaces.
 # • This is the complete, unabridged code with no functions skipped.
 ##############################################################################
 
@@ -290,19 +290,9 @@ def generate_report_bundle(buf: bytes, name: str, ctx: str, key: str):
     # 3. Assemble the final report bundle
     pdf_bytes = build_pdf(md, chart_paths)
 
-    def _substitute_tags_for_preview(match):
-        desc = match.group("d").strip()
-        path = chart_paths.get(desc)
-        if path:
-            b64 = base64.b64encode(Path(path).read_bytes()).decode()
-            return f'<img src="data:image/png;base64,{b64}" style="max-width:100%;">'
-        return f"*Chart '{desc}' could not be generated.*"
-
-    preview_md = TAG_RE.sub(_substitute_tags_for_preview, md)
-
     return {
-        "type": "report", "key": key, "preview_md": preview_md,
-        "pdf": pdf_bytes, "raw_md": md
+        "type": "report", "key": key, "raw_md": md,
+        "charts": chart_paths, "pdf": pdf_bytes
     }
 
 # ─── ANIMATION HELPERS ────────────────────────────────────────
@@ -546,7 +536,27 @@ if (bundle := st.session_state.get("bundle")):
     if bundle.get("type") == "report":
         st.subheader("📄 Generated Report")
         with st.expander("View Report", expanded=True):
-            st.markdown(bundle["preview_md"], unsafe_allow_html=True)
+            # This robust rendering loop iterates through the report text and
+            # uses native st.image() for charts, guaranteeing correct display.
+            report_md = bundle["raw_md"]
+            charts = bundle["charts"]
+            last_end = 0
+            for match in TAG_RE.finditer(report_md):
+                # Render the text that comes before the chart tag
+                st.markdown(report_md[last_end:match.start()])
+                
+                # Render the chart using st.image
+                desc = match.group("d").strip()
+                chart_path = charts.get(desc)
+                if chart_path and Path(chart_path).exists():
+                    st.image(chart_path)
+                else:
+                    st.warning(f"Could not render chart: '{desc}'")
+                
+                last_end = match.end()
+            
+            # Render any remaining text after the last chart
+            st.markdown(report_md[last_end:])
 
         c1, c2 = st.columns(2)
         with c1: