Jeremy Live commited on
Commit
bf2b46a
·
1 Parent(s): d21581b
Files changed (1) hide show
  1. app.py +6 -2
app.py CHANGED
@@ -224,7 +224,7 @@ def safe_get_column(df, column):
224
  # If still not found, raise a helpful error
225
  raise KeyError(f"Column '{column}' not found in DataFrame. Available columns: {list(df.columns)}")
226
 
227
- def plot_and_save(plt, filename=None):
228
  import os
229
  if filename is None:
230
  filename = PLOT_FILENAME
@@ -235,8 +235,12 @@ def plot_and_save(plt, filename=None):
235
  # Ensure the directory exists
236
  os.makedirs(os.path.dirname(filename) or '.', exist_ok=True)
237
 
 
 
 
 
238
  # Save the plot
239
- plt.savefig(filename, bbox_inches='tight', dpi=100)
240
  plt.close()
241
  print(f"[DEBUG] Plot saved to: {filename}")
242
  return filename
 
224
  # If still not found, raise a helpful error
225
  raise KeyError(f"Column '{column}' not found in DataFrame. Available columns: {list(df.columns)}")
226
 
227
+ def plot_and_save(plt, filename=None, **savefig_kwargs):
228
  import os
229
  if filename is None:
230
  filename = PLOT_FILENAME
 
235
  # Ensure the directory exists
236
  os.makedirs(os.path.dirname(filename) or '.', exist_ok=True)
237
 
238
+ # Set default savefig parameters if not provided
239
+ save_params = {'bbox_inches': 'tight', 'dpi': 100}
240
+ save_params.update(savefig_kwargs)
241
+
242
  # Save the plot
243
+ plt.savefig(filename, **save_params)
244
  plt.close()
245
  print(f"[DEBUG] Plot saved to: {filename}")
246
  return filename