Update app.py

app.py

@@ -1,6 +1,5 @@
 import base64
 import io
-import os
 import ast
 import traceback
 from threading import Thread
@@ -20,6 +19,7 @@ app.layout = dbc.Container([
     html.H1("Data Analysis Dashboard", className="my-4"),
     dbc.Card([
         dbc.CardBody([
+            dbc.Input(id="api-key", placeholder="Enter your Gemini API key", type="password", className="mb-3"),
             dcc.Upload(
                 id='upload-data',
                 children=html.Div([
@@ -66,10 +66,9 @@ def parse_contents(contents, filename):
         print(e)
         return None
 
-def process_data(df, instructions):
+def process_data(df, instructions, api_key):
     try:
-        # Initialize Gemini
-        api_key = os.environ.get('GEMINI_API_KEY')
+        # Initialize Gemini with provided API key
         genai.configure(api_key=api_key)
         model = genai.GenerativeModel('gemini-2.5-pro-preview-03-25')
         
@@ -146,17 +145,18 @@ def generate_plot(df, plot_info):
     [Input('submit-button', 'n_clicks')],
     [State('upload-data', 'contents'),
      State('upload-data', 'filename'),
-     State('instructions', 'value')]
+     State('instructions', 'value'),
+     State('api-key', 'value')]
 )
-def update_output(n_clicks, contents, filename, instructions):
-    if n_clicks is None or contents is None:
+def update_output(n_clicks, contents, filename, instructions, api_key):
+    if n_clicks is None or contents is None or not api_key:
         return dash.no_update, dash.no_update, dash.no_update
 
     df = parse_contents(contents, filename)
     if df is None:
         return dash.no_update, dash.no_update, dash.no_update
 
-    plots = process_data(df, instructions)
+    plots = process_data(df, instructions, api_key)
     if plots is None or len(plots) < 3:
         return dash.no_update, dash.no_update, dash.no_update