modified app.py for data memory issue

app.py

@@ -1387,14 +1387,8 @@ def create_combined_roi_time_series_graph(df):
         graph_file = "optimus_roi_graph.html"
         fig.write_html(graph_file, include_plotlyjs='cdn', full_html=False)
         
-        # Also save as image for compatibility
-        img_file = "optimus_roi_graph.png"
-        try:
-            fig.write_image(img_file)
-            logger.info(f"ROI graph saved to {graph_file} and {img_file}")
-        except Exception as e:
-            logger.error(f"Error saving ROI image: {e}")
-            logger.info(f"ROI graph saved to {graph_file} only")
+        # Skip PNG generation to avoid Kaleido/Chrome dependency issues in HF Spaces
+        logger.info(f"ROI graph saved to {graph_file} (PNG generation skipped for HF Spaces compatibility)")
         
         # Return the figure object for direct use in Gradio
         return fig
@@ -1636,11 +1630,8 @@ def create_time_series_graph_per_agent(df):
     graph_file = "optimus_apr_per_agent_graph.html"
     fig.write_html(graph_file, include_plotlyjs='cdn', full_html=False)
     
-    # Also save as image for compatibility
-    img_file = "optimus_apr_per_agent_graph.png"
-    fig.write_image(img_file)
-    
-    logger.info(f"Per-agent graph saved to {graph_file} and {img_file}")
+    # Skip PNG generation to avoid Kaleido/Chrome dependency issues in HF Spaces
+    logger.info(f"Per-agent graph saved to {graph_file} (PNG generation skipped for HF Spaces compatibility)")
     
     # Return the figure object for direct use in Gradio
     return fig
@@ -2184,14 +2175,8 @@ def create_combined_time_series_graph(df):
         graph_file = "optimus_apr_combined_graph.html"
         fig.write_html(graph_file, include_plotlyjs='cdn', full_html=False)
         
-        # Also save as image for compatibility
-        img_file = "optimus_apr_combined_graph.png"
-        try:
-            fig.write_image(img_file)
-            logger.info(f"Combined graph saved to {graph_file} and {img_file}")
-        except Exception as e:
-            logger.error(f"Error saving image: {e}")
-            logger.info(f"Combined graph saved to {graph_file} only")
+        # Skip PNG generation to avoid Kaleido/Chrome dependency issues in HF Spaces
+        logger.info(f"Combined graph saved to {graph_file} (PNG generation skipped for HF Spaces compatibility)")
         
         # Return the figure object for direct use in Gradio
         return fig