show graph

app.py

@@ -238,34 +238,30 @@ def safe_get_column(df, column):
     # If still not found, raise a helpful error
     raise KeyError(f"Column '{column}' not found in DataFrame. Available columns: {list(df.columns)}")
 
-def plot_and_save(plt, filename=None, **savefig_kwargs):
-    import os
-    import matplotlib
-    
-    # Make sure we're using the non-interactive backend
-    matplotlib.use('Agg')
-    
-    if filename is None:
-        filename = PLOT_FILENAME
-    
-    # Convert to absolute path
-    filename = os.path.abspath(str(filename))
-    
-    # Ensure the directory exists
-    os.makedirs(os.path.dirname(filename) or '.', exist_ok=True)
-    
+def show_plot(plt):
     try:
-        # Set default savefig parameters if not provided
-        save_params = {'bbox_inches': 'tight', 'dpi': 100}
-        save_params.update(savefig_kwargs)
+        # Use a non-interactive backend to avoid GUI issues
+        import matplotlib
+        matplotlib.use('Agg')
+        
+        # Create a temporary file to store the plot
+        import io
+        import base64
         
-        # Save the plot using the matplotlib savefig function directly
-        plt.savefig(filename, **save_params)
+        # Save plot to a bytes buffer
+        buf = io.BytesIO()
+        plt.savefig(buf, format='png', bbox_inches='tight', dpi=100)
         plt.close()
-        print(f"[DEBUG] Plot saved to: {filename}")
-        return filename
+        
+        # Convert to base64 for display
+        buf.seek(0)
+        img_str = base64.b64encode(buf.read()).decode('utf-8')
+        buf.close()
+        
+        # Return HTML to display the image
+        return f'<img src="data:image/png;base64,{img_str}" />'
     except Exception as e:
-        print(f"[ERROR] Failed to save plot: {str(e)}")
+        print(f"[ERROR] Failed to display plot: {str(e)}")
         return None
 
 # Monkey patch DataFrame to add safe column access
@@ -287,16 +283,16 @@ pd.DataFrame.safe_get = safe_get_column
                 script_content = script_content.replace("['Close']", ".safe_get('close')")
                 script_content = script_content.replace("['close']", ".safe_get('close')")
                 
-                # Replace plt.savefig() calls with our helper
+                # Replace plt.show() calls with our helper
                 script_content = re.sub(
-                    r'plt\.savefig\(([^)]+)\)',
-                    r'plot_and_save(plt, \1)',
+                    r'plt\\.show\\(\s*\\)',
+                    r'print(show_plot(plt))',
                     script_content
                 )
                 
-                # If no savefig call is found, add one at the end of the script
-                if 'plt.savefig(' not in script_content and 'plot_and_save(' not in script_content:
-                    script_content += "\n# Ensure plot is saved\nif 'plt' in locals() and len(plt.get_fignums()) > 0:\n    plot_and_save(plt, PLOT_FILENAME)\n"
+                # If no show() call is found, add one at the end of the script
+                if 'plt.show()' not in script_content:
+                    script_content += "\n# Display the plot if any figures exist\nif 'plt' in locals() and len(plt.get_fignums()) > 0:\n    print(show_plot(plt))\n"
                 
                 # Write the updated script back
                 with open(temp_script_path, 'w') as f: