Update app.py

app.py

@@ -2,6 +2,7 @@ import base64
 import io
 import ast
 import traceback
+import os
 from threading import Thread
 
 import dash
@@ -19,7 +20,6 @@ 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([
@@ -73,8 +73,13 @@ def parse_contents(contents, filename):
         print(e)
         return None
 
-def process_data(df, instructions, api_key):
+def process_data(df, instructions):
     try:
+        # Get API key from environment variable
+        api_key = os.getenv('GEMINI_API_KEY')
+        if not api_key:
+            raise ValueError("Gemini API key not found in environment variables")
+
         # Initialize Gemini with provided API key
         genai.configure(api_key=api_key)
         model = genai.GenerativeModel('gemini-2.5-pro-preview-03-25')
@@ -153,22 +158,18 @@ def generate_plot(df, plot_info):
     [Input('submit-button', 'n_clicks')],
     [State('upload-data', 'contents'),
      State('upload-data', 'filename'),
-     State('instructions', 'value'),
-     State('api-key', 'value')]
+     State('instructions', 'value')]
 )
-def update_output(n_clicks, contents, filename, instructions, api_key):
+def update_output(n_clicks, contents, filename, instructions):
     if n_clicks is None or contents is None:
         return dash.no_update, dash.no_update, dash.no_update, ""
-    
-    if not api_key:
-        return dash.no_update, dash.no_update, dash.no_update, "Please enter a valid API key."
 
     try:
         df = parse_contents(contents, filename)
         if df is None:
             return dash.no_update, dash.no_update, dash.no_update, "Unable to parse the uploaded file."
 
-        plots = process_data(df, instructions, api_key)
+        plots = process_data(df, instructions)
         if plots is None or len(plots) < 3:
             return dash.no_update, dash.no_update, dash.no_update, "Unable to generate visualizations. Please check your instructions and try again."
 
@@ -179,4 +180,6 @@ def update_output(n_clicks, contents, filename, instructions, api_key):
         return dash.no_update, dash.no_update, dash.no_update, error_message
 
 if __name__ == '__main__':
-    app.run(debug=False, host='0.0.0.0', port=7860, threaded=True)
+    print("Starting the Dash application...")
+    app.run(debug=True, host='0.0.0.0', port=7860)
+    print("Dash application has finished running.")