Update app.py

app.py

@@ -97,17 +97,22 @@ else:
 
                 # 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:
+                    # Convert cell values from strings to floats for plotting
+                    cell_values = [float(cell) for cell in cells if cell.replace('.', '', 1).isdigit()]
+
+                    # Plot the data if we have valid numeric values
+                    if len(cell_values) > 0:
+                        # Assuming that the coordinates or answer provides context on column names
+                        # You can adjust the labels or data based on the actual output
+                        column_names = [f"Row {i+1}" for i in range(len(cell_values))]
+
                         fig, ax = plt.subplots()
                         ax.bar(column_names, cell_values)
-                        ax.set_xlabel('Columns')
+                        ax.set_xlabel('Rows')
                         ax.set_ylabel('Values')
                         ax.set_title('Bar Plot of TAPAS Answer')
+
+                        # Display the plot in the Streamlit app
                         st.pyplot(fig)
 
             except Exception as e: