Update app.py

app.py

@@ -125,7 +125,10 @@ 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 column_name in df.columns: