Update app.py

app.py

@@ -54,7 +54,10 @@ else:
         if df is not None:
             numeric_columns = df.select_dtypes(include=['object']).columns
             for col in numeric_columns:
-                df[col] = pd.to_numeric(df[col], errors='ignore')
+                try:
+                    df[col] = pd.to_numeric(df[col])
+                except ValueError:
+                    st.warning(f"Column '{col}' contains non-numeric values that could not be converted.")
 
             st.write("Original Data:")
             st.write(df)
@@ -65,10 +68,10 @@ 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',
                 editable=True, 
                 height=300, 
                 width='100%',
+                fit_columns_on_grid_load=True  # Correct option for auto-sizing
             )
             
     except Exception as e:
@@ -94,8 +97,6 @@ else:
                         columns = question.split('between')[-1].split('and')
                         columns = [col.strip() for col in columns]
                         if len(columns) == 2 and all(col in df.columns for col in columns):
-                            # Debugging log
-                            st.write(f"Generating graph for columns: {columns[0]} and {columns[1]}")  # Debug statement
                             fig = px.scatter(df, x=columns[0], y=columns[1], title=f"Graph between {columns[0]} and {columns[1]}")
                             st.plotly_chart(fig, use_container_width=True)
                             st.success(f"Here is the graph between '{columns[0]}' and '{columns[1]}'.")
@@ -104,16 +105,14 @@ else:
                     elif 'column' in question.lower():
                         column = question.split('of')[-1].strip()
                         if column in df.columns:
-                            # Debugging log
-                            st.write(f"Generating graph for column: {column}")  # Debug statement
                             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.")
                     
-                    # Use `st.stop()` to stop processing further (no need for return)
-                    st.stop()  # Ensure this is placed after rendering the graph
+                    # Skip the TAPAS processing if it's a graph query
+                    st.stop()  # This ensures the code halts and avoids further processing
 
                 # Process TAPAS-related questions if it's not a graph query
                 raw_answer = tqa(table=df, query=question, truncation=True)