Update app.py

app.py

@@ -3,6 +3,7 @@ import streamlit as st
 from st_aggrid import AgGrid
 import pandas as pd
 from transformers import pipeline, T5ForConditionalGeneration, T5Tokenizer
+import matplotlib.pyplot as plt
 
 # Set the page layout for Streamlit
 st.set_page_config(layout="wide")
@@ -94,6 +95,21 @@ else:
                 st.write(f"Coordinates: {coordinates}")
                 st.write(f"Cells: {cells}")
 
+                # If cells are returned, we will extract the corresponding values for plotting
+                if cells:
+                    # For simplicity, let's assume cells contain data we want to plot (e.g., numeric data)
+                    cell_values = [float(cell['value']) for cell in cells if cell['value'].isnumeric()]
+                    column_names = [cell['column'] for cell in cells]
+                    
+                    # Plot the data
+                    if len(cell_values) > 0 and len(column_names) > 0:
+                        fig, ax = plt.subplots()
+                        ax.bar(column_names, cell_values)
+                        ax.set_xlabel('Columns')
+                        ax.set_ylabel('Values')
+                        ax.set_title('Bar Plot of TAPAS Answer')
+                        st.pyplot(fig)
+
             except Exception as e:
                 st.warning(f"Error processing question or generating answer: {str(e)}")
                 st.warning("Please retype your question and make sure to use the column name and cell value correctly.")