Update app.py

app.py

@@ -107,8 +107,8 @@ else:
                             st.success(f"Here is the graph of column '{column}'.")
                         else:
                             st.warning(f"Column '{column}' not found in the data.")
-                    # Skip the TAPAS processing if it's a graph query
-                    return  # Simply return to exit this block for graph queries
+                    # Skip TAPAS processing if it's a graph query
+                    st.stop()  # Use `st.stop()` to exit the flow for graph queries and prevent TAPAS processing.
 
                 # Process TAPAS-related questions if it's not a graph query
                 raw_answer = tqa(table=df, query=question, truncation=True)