Update app.py

app.py

@@ -123,12 +123,12 @@ else:
                     else:
                         st.write(f"No data to plot for the question: '{question}'")
 
-                else:
+                 else:
                     # Handle graph-related questions
-                 if 'count' in question.lower():  # Check if the question contains "count"
-        # Extract the column name to count
-        column_name = question.split('count')[-1].strip()
-   
+                    if is_count_query:
+                        # Extract the column name to count
+                        column_name = question.split('count')[-1].strip()
+
                         
 
                         if column_name in df.columns: