Update app.py

app.py

@@ -13,6 +13,7 @@ from io import BytesIO
 from simple_salesforce import Salesforce
 import base64
 import logging
+import json
 
 # Configure logging
 logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')
@@ -247,7 +248,17 @@ if submit_button:
             
             # Generate Chart.js heatmap
             chart_config = generate_heatmap(prediction['delay_probability'], f"{phase}: {task}")
-            st.write("```chartjs\n" + str(chart_config) + "\n```")
+            chart_id = f"chart-{hash(str(chart_config))}"
+            chart_html = f"""
+            <canvas id="{chart_id}" style="max-height: 200px; max-width: 600px;"></canvas>
+            <script src="https://cdn.jsdelivr.net/npm/chart.js"></script>
+            <script>
+                const ctx = document.getElementById('{chart_id}').getContext('2d');
+                new Chart(ctx, {json.dumps(chart_config)});
+            </script>
+            """
+            st.markdown(chart_html, unsafe_allow_html=True)
+            logger.info("Chart.js heatmap rendered")
             
             # Generate matplotlib figure for PDF
             fig, ax = plt.subplots(figsize=(8, 2))