make feedback into a table for better vis + add btn

app.py

@@ -41,6 +41,12 @@ uploaded_file = st.file_uploader(
     "Upload Document", type=["pdf", "jpg", "png"], accept_multiple_files=False
 )
 
+
+feedback_table = wandb.Table(columns=[
+    'image', 'filetype', 'predicted_label', 'predicted_label_id',
+    'correct_label', 'correct_label_id'
+])
+
 if uploaded_file:
     run = wandb.init(project='hydra-classifier', name='feedback-loop')
 
@@ -77,13 +83,17 @@ if uploaded_file:
             )
             print(f'Correct label for image {i}: {correct_label}')
 
-            run.log({
-                'image': wandb.Image(image),
-                'filetype': uploaded_file.type,
-                'predicted_label': id2label[prediction],
-                'predicted_label_id': prediction,
-                'correct_label': correct_label,
-                'correct_label_id': label2id[correct_label]
-            })
+            # Add a button to confirm feedback and log it
+            if st.button(f"Submit Feedback for Image {i}", key=f'submit-{i}'):
+                feedback_table.add({
+                    'image': wandb.Image(image),
+                    'filetype': uploaded_file.type,
+                    'predicted_label': id2label[prediction],
+                    'predicted_label_id': prediction,
+                    'correct_label': correct_label,
+                    'correct_label_id': label2id[correct_label]
+                })
+                st.success(f"Feedback for Image {i} submitted!")
 
+    run.log({'feedback_table': feedback_table})
     run.finish()