Update app.py

app.py

@@ -62,53 +62,67 @@ else:
     # User input for the question
     question = st.text_input(f'Ask your graph-related question in {language}')
 
-    with st.spinner():
-        if st.button('Generate Graph'):
-            try:
-                # Check if the question is a valid string (not empty or None)
-                if not question or not isinstance(question, str):
-                    st.error("Please enter a valid question in the form of text.")
+   with st.spinner():
+    if st.button('Generate Graph'):
+        try:
+            # Ensure the question is a valid string
+            if not question or not isinstance(question, str):
+                st.error("Please enter a valid question in the form of text.")
+            else:
+                # Use TAPAS model to process the question
+                result = tqa(table=df, query=question)
+                
+                # Display the raw output from TAPAS
+                st.write("TAPAS Raw Output (Response):")
+                st.write(result)  # This will display the raw output from TAPAS
+
+                # Optionally, you can output the raw output as plain text:
+                st.text("Raw TAPAS Output (Plain Text):")
+                st.text(str(result))  # This will display raw output as plain text
+
+                # Check if TAPAS is returning the expected answer
+                answer = result.get('answer', None)
+                if answer:
+                    st.write(f"TAPAS Answer: {answer}")
                 else:
-                    # Use TAPAS model to process the question
-                    result = tqa(table=df, query=question)
-
-                    # Debugging: Output TAPAS raw response
-                    st.write("TAPAS Raw Output (Response):")
-                    st.write(result)  # Show the raw output from TAPAS
-                    
-                    # Optionally, you can output the raw output as plain text:
-                    st.text("Raw TAPAS Output (Plain Text):")
-                    st.text(str(result))  # Show raw output as plain text
-
-                    # Check if TAPAS is returning the expected answer
-                    answer = result.get('answer', None)
-                    if answer:
-                        st.write(f"TAPAS Answer: {answer}")
+                    st.warning("TAPAS did not return a valid 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):
+                        # Prepare the data for Plotly (scatter plot)
+                        x_data = df[columns[0]].dropna()  # Extract x column, drop NaN values
+                        y_data = df[columns[1]].dropna()  # Extract y column, drop NaN values
+
+                        # Ensure x_data and y_data have the same length
+                        min_length = min(len(x_data), len(y_data))
+                        x_data = x_data[:min_length]
+                        y_data = y_data[:min_length]
+
+                        # Create the scatter plot
+                        fig = px.scatter(x=x_data, y=y_data, 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("TAPAS did not return a valid 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.")
+                        st.warning("Columns not found in the dataset or the question format is incorrect.")
+                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:
+                        # Prepare the data for Plotly (line graph)
+                        column_data = df[column].dropna()  # Drop NaN values
+
+                        # Create the line plot
+                        fig = px.line(x=column_data.index, y=column_data, 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("Please ask a valid graph-related question (e.g., 'make a graph between column1 and column2').")
+                        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)}")
+        except Exception as e:
+            st.warning(f"Error processing question or generating graph: {str(e)}")