adding metadata feature

app.py

@@ -1,5 +1,6 @@
 import gradio as gr
 import random
+import csv
 
 class_names = ['cat', 'dog']
 
@@ -94,5 +95,23 @@ with demo as app:
 
         b3.click(change_labels, [imgs, selected, images_label], [imgs, gallery])
 
+        b4 = gr.Button("Upload to metadata")
+
+        def upload_metadata(imgs):
+            with open('metadata.csv', mode='w', newline='') as csv_file:
+                # Create a CSV writer
+                csv_writer = csv.writer(csv_file)
+
+                # Write the header row
+                csv_writer.writerow(['File_Path', 'Label'])
+
+                # Write the data rows
+                csv_writer.writerows(imgs)
+
+            print(f"Metadata CSV file has been created.")
+            return imgs
+
+        b4.click(upload_metadata, imgs, imgs)
+
 
 demo.launch(debug=True)