Update app.py

app.py

@@ -32,24 +32,29 @@ def process_file(api_key, file, instructions):
         # Generate visualizations
         images = []
         for plot in plots[:3]:  # Ensure max 3 plots
-            fig = plt.figure()
+            fig, ax = plt.subplots(figsize=(10, 6))
             title, plot_type, x, y = plot
             
             if plot_type == 'bar':
-                df.plot.bar(x=x, y=y, ax=plt.gca())
+                df.plot(kind='bar', x=x, y=y, ax=ax)
             elif plot_type == 'line':
-                df.plot.line(x=x, y=y, ax=plt.gca())
+                df.plot(kind='line', x=x, y=y, ax=ax)
             elif plot_type == 'scatter':
-                df.plot.scatter(x=x, y=y, ax=plt.gca())
+                df.plot(kind='scatter', x=x, y=y, ax=ax)
             elif plot_type == 'hist':
-                df[y].hist(ax=plt.gca())
+                df[y].hist(ax=ax)
+            
+            ax.set_title(title)
+            ax.set_xlabel(x)
+            ax.set_ylabel(y)
+            plt.tight_layout()
             
-            plt.title(title)
             buf = io.BytesIO()
-            fig.savefig(buf, format='png', bbox_inches='tight')
+            plt.savefig(buf, format='png')
             buf.seek(0)
-            images.append(Image.open(buf))
-            plt.close()
+            img = Image.open(buf)
+            images.append(img)
+            plt.close(fig)
 
         return images if len(images) == 3 else images + [Image.new('RGB', (800, 600), (255,255,255))]*(3-len(images))