Update app.py

app.py

@@ -35,7 +35,8 @@ def process_file(api_key, file, instructions):
         buf = io.BytesIO()
         plt.savefig(buf, format='png')
         buf.seek(0)
-        visualizations.append(buf)
+        img_str = base64.b64encode(buf.getvalue()).decode()
+        visualizations.append(f"data:image/png;base64,{img_str}")
         plt.close()
 
     return visualizations
@@ -43,7 +44,7 @@ def process_file(api_key, file, instructions):
 # Gradio interface
 with gr.Blocks() as demo:
     gr.Markdown("Data Visualization with Gemini")
-    api_key = gr.Textbox(label="Enter Gemini API Key")
+    api_key = gr.Textbox(label="Enter Gemini API Key", type="password")
     file = gr.File(label="Upload Excel or CSV file")
     instructions = gr.Textbox(label="Optional visualization instructions")
     submit = gr.Button("Generate Visualizations")