Update app.py

app.py

@@ -42,13 +42,20 @@ app.layout = dbc.Container([
             dbc.Button("Generate Insights", id="submit-button", color="primary", className="mt-3"),
         ])
     ], className="mb-4"),
-    dbc.Card([
-        dbc.CardBody([
-            dcc.Graph(id='visualization-1'),
-            dcc.Graph(id='visualization-2'),
-            dcc.Graph(id='visualization-3'),
-        ])
-    ])
+    html.Div(id="error-message", className="text-danger mb-3"),
+    dcc.Loading(
+        id="loading-visualizations",
+        type="default",
+        children=[
+            dbc.Card([
+                dbc.CardBody([
+                    dcc.Graph(id='visualization-1'),
+                    dcc.Graph(id='visualization-2'),
+                    dcc.Graph(id='visualization-3'),
+                ])
+            ])
+        ]
+    )
 ], fluid=True)
 
 def parse_contents(contents, filename):
@@ -141,7 +148,8 @@ def generate_plot(df, plot_info):
 @app.callback(
     [Output('visualization-1', 'figure'),
      Output('visualization-2', 'figure'),
-     Output('visualization-3', 'figure')],
+     Output('visualization-3', 'figure'),
+     Output('error-message', 'children')],
     [Input('submit-button', 'n_clicks')],
     [State('upload-data', 'contents'),
      State('upload-data', 'filename'),
@@ -149,19 +157,26 @@ def generate_plot(df, plot_info):
      State('api-key', 'value')]
 )
 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
+    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."
 
-    df = parse_contents(contents, filename)
-    if df is None:
-        return dash.no_update, dash.no_update, dash.no_update
+    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)
-    if plots is None or len(plots) < 3:
-        return dash.no_update, dash.no_update, dash.no_update
+        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, "Unable to generate visualizations. Please check your instructions and try again."
 
-    figures = [generate_plot(df, plot_info) for plot_info in plots[:3]]
-    return figures
+        figures = [generate_plot(df, plot_info) for plot_info in plots[:3]]
+        return figures[0], figures[1], figures[2], ""
+    except Exception as e:
+        error_message = f"An error occurred: {str(e)}"
+        return dash.no_update, dash.no_update, dash.no_update, error_message
 
 if __name__ == '__main__':
     app.run(debug=True, host='0.0.0.0', port=7860, threaded=True)