Improve plot resolution and add CSV download functionality

- Increased plot DPI from 300 to 600 for ultra-high resolution
- Enhanced matplotlib style with larger figure size (14x8)
- Added explicit high-DPI settings in code generation template
- Fixed CSV download for dataframes in Streamlit interface
- Updated system prompt with high-resolution plotting instructions

🤖 Generated with [Claude Code](https://claude.ai/code)

Co-Authored-By: Claude <[email protected]>

app.py

@@ -481,7 +481,22 @@ def show_custom_response(response):
             </div>
             """, unsafe_allow_html=True)
             
+            # Display dataframe with download option
             st.dataframe(content, use_container_width=True)
+            
+            # Add download button for CSV
+            import io
+            csv_buffer = io.StringIO()
+            content.to_csv(csv_buffer, index=False)
+            csv_data = csv_buffer.getvalue()
+            
+            st.download_button(
+                label="📥 Download as CSV",
+                data=csv_data,
+                file_name=f"vayuchat_data_{datetime.now().strftime('%Y%m%d_%H%M%S')}.csv",
+                mime="text/csv",
+                use_container_width=True
+            )
         
         # Show generated code with Streamlit expander
         if response.get("gen_code"):

new_system_prompt.txt

@@ -33,8 +33,8 @@ DATA SAFETY:
 - Use .dropna() to remove missing values before analysis
 
 PLOTTING REQUIREMENTS:
-- Create plots for visualization requests: plt.figure(figsize=(12, 8))
-- Save plots: filename = f"plot_{uuid.uuid4().hex[:8]}.png"; plt.savefig(filename, dpi=300, bbox_inches='tight')
+- Create plots for visualization requests: plt.figure(figsize=(14, 8))
+- Save plots with ultra-high resolution: filename = f"plot_{uuid.uuid4().hex[:8]}.png"; plt.savefig(filename, dpi=600, bbox_inches='tight', facecolor='white', edgecolor='none')
 - Close plots: plt.close()
 - Store filename: answer = filename
 - For non-plots: answer = "text result"

src.py

@@ -254,8 +254,17 @@ import seaborn as sns
 import uuid
 import calendar
 import numpy as np
-# Set professional matplotlib styling
+# Set professional matplotlib styling with high DPI
 plt.style.use('vayuchat.mplstyle')
+# Additional high-resolution settings
+plt.rcParams['figure.dpi'] = 600
+plt.rcParams['savefig.dpi'] = 600
+plt.rcParams['font.size'] = 14
+plt.rcParams['axes.titlesize'] = 16
+plt.rcParams['axes.labelsize'] = 14
+plt.rcParams['xtick.labelsize'] = 12
+plt.rcParams['ytick.labelsize'] = 12
+plt.rcParams['legend.fontsize'] = 12
 df = pd.read_csv("AQ_met_data.csv")
 df["Timestamp"] = pd.to_datetime(df["Timestamp"])
 states_df = pd.read_csv("states_data.csv")

vayuchat.mplstyle

@@ -12,11 +12,11 @@ xtick.labelsize: 11
 ytick.labelsize: 11
 legend.fontsize: 11
 
-# Figure & DPI
-figure.dpi: 300
+# Figure & DPI - Ultra High Resolution
+figure.dpi: 600
 figure.facecolor: white
 figure.edgecolor: none
-figure.figsize: 12, 7
+figure.figsize: 14, 8
 figure.autolayout: True
 
 # Modern Color Palette (inspired by Tailwind/GitHub)
@@ -84,9 +84,10 @@ patch.force_edgecolor: False
 text.color: 1f2937
 text.antialiased: True
 
-# Savefig
-savefig.dpi: 300
+# Savefig - Ultra High Resolution
+savefig.dpi: 600
 savefig.facecolor: white
 savefig.edgecolor: none
 savefig.bbox: tight
-savefig.pad_inches: 0.2
+savefig.pad_inches: 0.2
+savefig.format: png