Update app.py

app.py

@@ -65,7 +65,7 @@ else:
             # Display the first 5 rows of the dataframe in an editable grid
             grid_response = AgGrid(
                 df.head(5),
-                columns_auto_size_mode='FIT_CONTENTS',
+                fit_columns_on_grid_load=True,  # Correct parameter to fit columns on grid load
                 editable=True, 
                 height=300, 
                 width='100%',
@@ -140,8 +140,6 @@ else:
                             fig = px.line(df, x=df.index, y=column, title=f"Graph of column '{column}'")
                             st.plotly_chart(fig, use_container_width=True)
                             st.success(f"Here is the graph of column '{column}'.")
-                        else:
-                            st.warning(f"Column '{column}' not found in the data.")
                     
                     # Stop further execution after generating the graph
                     st.stop()  # This halts further execution