Update app.py

app.py

@@ -2,6 +2,11 @@ import gradio as gr
 import pandas as pd
 from PIL import Image
 import os
+import uuid
+
+# Create a directory to store the images
+image_directory = "images"
+os.makedirs(image_directory, exist_ok=True)
 
 # Define the path to the CSV file in the current directory
 csv_file_path = "descriptions.csv"
@@ -10,9 +15,12 @@ csv_file_path = "descriptions.csv"
 data = []
 
 # Function to handle the user's input and store it in the list
-def describe_image(image, filename, copyright, description):
+def describe_image(image, filename, description, copyright):
     global data
-    data.append({"Image": image, "XML File Name": filename, "Copyright Holder": copyright, "Physichal Description": description})
+    # Generate a unique filename for each image
+    image_filename = os.path.join(image_directory, f"{str(uuid.uuid4())}.jpg")
+    image.save(image_filename)
+    data.append({"Image": image_filename, "Filename": filename, "Description": description, "Copyright": copyright})
     # Convert the list to a DataFrame
     data_df = pd.DataFrame(data)
     data_df.to_csv(csv_file_path, index=False)  # Save the data to a CSV file in the current directory