Update app.py

app.py

@@ -61,13 +61,18 @@ else:
     with st.spinner():
         if st.button('Generate Graph'):
             try:
+                # 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 column names that could be used for graphing
+                # 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)
@@ -81,7 +86,7 @@ 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()
+                    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)