Update app.py

app.py

@@ -2,9 +2,7 @@ import gradio as gr
 import pandas as pd
 import matplotlib.pyplot as plt
 import io
-import base64
 import google.generativeai as genai
-import textwrap
 
 def process_file(api_key, file, instructions):
     # Set up Gemini API
@@ -65,10 +63,9 @@ def process_file(api_key, file, instructions):
             buf = io.BytesIO()
             plt.savefig(buf, format='png')
             buf.seek(0)
-            img_str = base64.b64encode(buf.getvalue()).decode()
             plt.close()
             
-            visualizations.append(f"data:image/png;base64,{img_str}")
+            visualizations.append(buf)
         except Exception as e:
             print(f"Error in visualization {i}: {str(e)}")
             visualizations.append(None)