Update app.py

app.py

@@ -6,10 +6,12 @@ import ast
 from PIL import Image, ImageDraw
 import google.generativeai as genai
 import traceback
+import os
 
-def process_file(file, instructions, api_key):
+def process_file(file, instructions):
     try:
         # Initialize Gemini
+        api_key = os.environ.get('GEMINI_API_KEY')
         genai.configure(api_key=api_key)
         model = genai.GenerativeModel('gemini-2.5-pro-preview-03-25')
         
@@ -120,14 +122,13 @@ with gr.Blocks(theme=gr.themes.Default()) as demo:
         file = gr.File(label="Upload Dataset", file_types=[".csv", ".xlsx"])
         instructions = gr.Textbox(label="Analysis Instructions", placeholder="Describe the analysis you want...")
     
-    api_key = gr.Textbox(label="Gemini API Key", type="password")
     submit = gr.Button("Generate Insights", variant="primary")
     
     output_images = [gr.Image(label=f"Visualization {i+1}") for i in range(3)]
 
     submit.click(
         process_file,
-        inputs=[file, instructions, api_key],
+        inputs=[file, instructions],
         outputs=output_images
     )