Update app.py

app.py

@@ -70,37 +70,36 @@ else:
                 # Ensure the question is a string
                 if not isinstance(question, str):
                     st.error("The question input is not a valid string.")
-                    return
-
-                # Use TAPAS model to process the question
-                result = tqa(table=df, query=question)
-
-                # Check if TAPAS is returning the expected answer
-                answer = result['answer']
-                st.write(f"TAPAS Answer: {answer}")
-
-                # Determine if the question relates to graph generation
-                if 'between' in question.lower() and 'and' in question.lower():
-                    # This is a request for a scatter plot (two columns)
-                    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):
-                        fig = px.scatter(df, x=columns[0], y=columns[1], title=f"Scatter Plot between {columns[0]} and {columns[1]}")
-                        st.plotly_chart(fig, use_container_width=True)
-                        st.success(f"Here is the scatter plot between '{columns[0]}' and '{columns[1]}'.")
-                    else:
-                        st.warning("Columns not found in the dataset.")
-                elif 'column' in question.lower():
-                    # This is a request for a line graph (single column)
-                    column = question.split('of')[-1].strip()  # Handle 'of' keyword
-                    if column in df.columns:
-                        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.")
                 else:
-                    st.warning("Please ask a valid graph-related question (e.g., 'make a graph between column1 and column2').")
+                    # Use TAPAS model to process the question
+                    result = tqa(table=df, query=question)
+
+                    # Check if TAPAS is returning the expected answer
+                    answer = result['answer']
+                    st.write(f"TAPAS Answer: {answer}")
+
+                    # Determine if the question relates to graph generation
+                    if 'between' in question.lower() and 'and' in question.lower():
+                        # This is a request for a scatter plot (two columns)
+                        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):
+                            fig = px.scatter(df, x=columns[0], y=columns[1], title=f"Scatter Plot between {columns[0]} and {columns[1]}")
+                            st.plotly_chart(fig, use_container_width=True)
+                            st.success(f"Here is the scatter plot between '{columns[0]}' and '{columns[1]}'.")
+                        else:
+                            st.warning("Columns not found in the dataset.")
+                    elif 'column' in question.lower():
+                        # This is a request for a line graph (single column)
+                        column = question.split('of')[-1].strip()  # Handle 'of' keyword
+                        if column in df.columns:
+                            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.")
+                    else:
+                        st.warning("Please ask a valid graph-related question (e.g., 'make a graph between column1 and column2').")
 
             except Exception as e:
                 st.warning(f"Error processing question or generating graph: {str(e)}")