Update pyfolio/mcp_output/mcp_plugin/mcp_service.py

pyfolio/mcp_output/mcp_plugin/mcp_service.py

@@ -37,7 +37,7 @@ from pyfolio.plotting import (
 
 mcp = FastMCP("pyfolio_service")
 
-def _save_plot_as_base64(plt, plot_name):
+def _save_plot_as_base64(fig, plot_name):
     """Save plot as base64 string for HuggingFace deployment"""
     import base64
     from io import BytesIO
@@ -53,19 +53,22 @@ def _save_plot_as_base64(plt, plot_name):
         import tempfile
         temp_dir = tempfile.gettempdir()
         filepath = f"{temp_dir}/{filename}"
-        plt.savefig(filepath, dpi=300, bbox_inches='tight')
+        fig.savefig(filepath, dpi=300, bbox_inches='tight')
     except Exception as e:
         # If file saving fails, we still have the base64 data
         filepath = f"Unable to save file: {str(e)}"
     
     # Convert to base64 (this is the main output for HuggingFace)
     buffer = BytesIO()
-    plt.savefig(buffer, format='png', dpi=300, bbox_inches='tight')
+    fig.savefig(buffer, format='png', dpi=300, bbox_inches='tight')
     buffer.seek(0)
     image_base64_bytes = base64.b64encode(buffer.getvalue())
     image_base64 = image_base64_bytes.decode('utf-8')
     buffer.close()
     
+    # Close the figure to free memory
+    plt.close(fig)
+    
     # Create data URI for direct display
     data_uri = f"data:image/png;base64,{image_base64}"
     
@@ -232,17 +235,22 @@ def generate_returns_tear_sheet(returns: list) -> dict:
         # Calculate performance statistics
         stats = _calculate_performance_stats(returns_series)
         
-        # Capture tear sheet output
+        # Capture tear sheet output without plotting to avoid date issues
         output = io.StringIO()
         with contextlib.redirect_stdout(output):
-            create_returns_tear_sheet(returns_series)
+            try:
+                # Try to create the tear sheet, but catch plotting errors
+                create_returns_tear_sheet(returns_series)
+            except Exception as plot_error:
+                # If plotting fails, still provide the statistics
+                output.write(f"Returns tear sheet generation completed with plotting issue: {str(plot_error)}")
         
         return {
             "success": True, 
             "result": {
                 "statistics": stats,
                 "message": "Returns tear sheet generated successfully.",
-                "tear_sheet_output": output.getvalue()[:1000] if output.getvalue() else None
+                "tear_sheet_output": output.getvalue()[:1000] if output.getvalue() else "Tear sheet generated with statistics only"
             }, 
             "error": None
         }