Update app.py

app.py

@@ -94,6 +94,8 @@ else:
                         columns = question.split('between')[-1].split('and')
                         columns = [col.strip() for col in columns]
                         if len(columns) == 2 and all(col in df.columns for col in columns):
+                            # Debugging log
+                            st.write(f"Generating graph for columns: {columns[0]} and {columns[1]}")  # Debug statement
                             fig = px.scatter(df, x=columns[0], y=columns[1], title=f"Graph between {columns[0]} and {columns[1]}")
                             st.plotly_chart(fig, use_container_width=True)
                             st.success(f"Here is the graph between '{columns[0]}' and '{columns[1]}'.")
@@ -102,13 +104,16 @@ else:
                     elif 'column' in question.lower():
                         column = question.split('of')[-1].strip()
                         if column in df.columns:
+                            # Debugging log
+                            st.write(f"Generating graph for column: {column}")  # Debug statement
                             fig = px.line(df, x=df.index, y=column, title=f"Graph of column '{column}'")
                             st.plotly_chart(fig, use_container_width=True)
                             st.success(f"Here is the graph of column '{column}'.")
                         else:
                             st.warning(f"Column '{column}' not found in the data.")
-                    # 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.
+                    
+                    # Use `st.stop()` to stop processing further (no need for return)
+                    st.stop()  # Ensure this is placed after rendering the graph
 
                 # Process TAPAS-related questions if it's not a graph query
                 raw_answer = tqa(table=df, query=question, truncation=True)