Update app.py

app.py

@@ -142,7 +142,9 @@ else:
                             st.success(f"Here is the graph of column '{column}'.")
                         else:
                             st.warning(f"Column '{column}' not found in the data.")
-                    return  # Skip TAPAS processing for graph-related queries
+                    
+                    # Stop further execution after generating the graph
+                    st.stop()  # This halts further execution
 
             except Exception as e:
                 st.warning(f"Error processing question or generating answer: {str(e)}")