Consolidated instructions for saving figures

Author: gekinci

src/climateclaw/prompt_library/baseline/examples.jsonl

@@ -131,7 +131,7 @@
 // Handlind daily data
 {"variant":"User","content":"Make a map plot of surface wind on 25.04.2010."}
 {"variant":"Assistant","content":"Here's how I will proceed:\n\n1. Query ERA5 surface wind (sfcWind) data for 25 April 2010 using freva-client.\n2. Load the data for that day using xarray.\n3. Create a global map plot (PlateCarree) of daily mean surface wind using matplotlib and cartopy.\n4. Add coastlines, borders, and a colorbar to the plot.\n\nLet's perform these steps now."}
-{"variant":"Code", "content":["{\"code\":\"import freva_client\\nimport xarray as xr\\nimport matplotlib.pyplot as plt\\nimport cartopy.crs as ccrs\\nimport cartopy.feature as cfeature\\nimport numpy as np\\n\\n# 1. Query ERA5 surface wind (sfcWind) data for 25 April 2010\\nerawind_files = freva_client.databrowser(\\n    project='reanalysis',\\n    experiment='era5',\\n    variable='sfcwind',\\n    time_frequency='day',\\n    time='2010-04-25',\\n    host='nextgems.dkrz.de')\\n\\n# 2. Load the data\\nds = xr.open_mfdataset(list(erawind_files), combine='by_coords')\\nwind = ds['sfcWind'].sel(time='2010-04-25').squeeze()\\n\\n# 3. Prepare lon/lat grid\\nd_lon = ds['lon']\\nd_lat = ds['lat']\\nlon2d, lat2d = np.meshgrid(d_lon, d_lat)\\n\\n# 4. Plotting\\nfig, ax = plt.subplots(figsize=(12,6), subplot_kw={'projection': ccrs.PlateCarree()})\\nlevels = np.linspace(np.nanmin(wind), np.nanmax(wind), 60)\\ncf = ax.contourf(lon2d, lat2d, wind, levels=levels, cmap='viridis', extend='both', transform=ccrs.PlateCarree())\\nax.add_feature(cfeature.COASTLINE)\\nax.add_feature(cfeature.BORDERS, linestyle=':')\\nplt.colorbar(cf, orientation='horizontal', pad=0.05, aspect=50, label='Surface Wind (m/s)')\\nplt.title('ERA5 Surface Wind on 2010-04-25')\\nplt.tight_layout()\\nplt.show()\"}","daily_wind_plot"]}
+{"variant":"Code", "content":["{\"code\":\"import freva_client\\nimport xarray as xr\\nimport matplotlib.pyplot as plt\\nimport cartopy.crs as ccrs\\nimport cartopy.feature as cfeature\\nimport numpy as np\\n\\n# 1. Query ERA5 surface wind (sfcWind) data for 25 April 2010\\nerawind_files = freva_client.databrowser(\\n    project='reanalysis',\\n    experiment='era5',\\n    variable='sfcwind',\\n    time_frequency='day',\\n    time='2010-04-25',\\n    host='nextgems.dkrz.de')\\n\\n# 2. Load the data\\nds = xr.open_mfdataset(list(erawind_files), combine='by_coords')\\nwind = ds['sfcWind'].sel(time='2010-04-25').squeeze()\\n\\n# 3. Prepare lon/lat grid\\nd_lon = ds['lon']\\nd_lat = ds['lat']\\nlon2d, lat2d = np.meshgrid(d_lon, d_lat)\\n\\n# 4. Plotting\\nfig, ax = plt.subplots(figsize=(12,6), subplot_kw={'projection': ccrs.PlateCarree()})\\nlevels = np.linspace(np.nanmin(wind), np.nanmax(wind), 60)\\ncf = ax.contourf(lon2d, lat2d, wind, levels=levels, cmap='viridis', extend='both', transform=ccrs.PlateCarree())\\nax.add_feature(cfeature.COASTLINE)\\nax.add_feature(cfeature.BORDERS, linestyle=':')\\nplt.colorbar(cf, orientation='horizontal', pad=0.05, aspect=50, label='Surface Wind (m/s)')\\nplt.title('ERA5 Surface Wind on 2010-04-25')\\nplt.tight_layout()\\nplt.savefig('era5_surface_wind_20100425.png')\"}","daily_wind_plot"]}
 {"variant":"CodeOutput","content":["","daily_wind_plot"]}
 {"variant":"Assistant","content":"Here is the global map plot of ERA5 surface wind (sfcWind, m/s) for 25 April 2010.\nThe map uses a PlateCarree projection, includes coastlines and country borders, and the colorbar shows the wind speed in meters per second. If you need any specific adjustments or a different region/projection, just let me know!"}
 // Weighting by area

src/climateclaw/prompt_library/baseline/starting_prompt.txt

@@ -112,8 +112,9 @@ G. FAILURE & TIMEOUT HANDLING
 H. FILE SAVING
 
 1. When using code_interpreter to create a plot, chart, figure, image, map, or visualization for the user, always save it to a file in the current working directory.
-2. Use a descriptive filename with an appropriate extension, such as `.png`, `.jpg`, `.svg`, or `.pdf`.
-2. Only use `open` and do NOT `import os`.
+2. Instead of displaying figures inline, save every generated figure to disk, and close each figure after saving to avoid unnecessary inline rendering or memory use.
+3. Use a descriptive filename with an appropriate extension, such as `.png`, `.jpg`, `.svg`, or `.pdf`.
+4. Only use `open` and do NOT `import os`.
 
 I. FORMATTING