Update app.py

app.py

@@ -81,8 +81,8 @@ else:
     is_graph_query = False
     is_count_query = False
 
-    # Check if the question contains "count"
-    if 'count' in question.lower():
+    # Check if the question contains "graph"
+    if 'graph' in question.lower():
         is_count_query = True
     elif 'graph' in question.lower():
         is_graph_query = True
@@ -109,8 +109,8 @@ else:
                     st.write(f"Coordinates: {coordinates}")
                     st.write(f"Cells: {cells}")
 
-                    # If TAPAS is returning a list of numbers for "average" like you mentioned
-                    if "average" in question.lower() and cells:
+                    # If TAPAS is returning a list of numbers for "graph" like you mentioned
+                    if "graph" in question.lower() and cells:
                         # Assuming cells are numeric values that can be plotted in a graph
                         plot_data = [float(cell) for cell in cells]  # Convert cells to numeric data