Update app.py

app.py

@@ -3,28 +3,69 @@ import pandas as pd
 from PIL import Image
 import os
 import uuid
+import boto3
 
-# 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"
+aws_access_key = os.environ['AWS_ACCESS_KEY']
+aws_secret_key = os.environ['AWS_SECRET_KEY']
+aws_region = os.environ['AWS_REGION']
+
+s3 = boto3.client('s3', aws_access_key_id=aws_access_key, aws_secret_access_key=aws_secret_key, region_name=aws_region)
+
+s3_bucket_name = os.environ['S3_BUCKET_NAME']
 
-# Initialize an empty list to store the data
 data = []
 
-# Function to handle the user's input and store it in the list
+
+base_csv_file_path = "descriptions"
+csv_file_path = f"{base_csv_file_path}.csv"
+
+
+def file_exists(file_path):
+    return os.path.isfile(file_path)
+
+
+def generate_unique_csv_file():
+    count = 1
+    while file_exists(f"{base_csv_file_path}{count}.csv"):
+        count += 1
+    return f"{base_csv_file_path}{count}.csv"
+
+
 def describe_image(image, filename, description, copyright):
     global data
-    # 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
+
+   
+    image_filename = f"{str(uuid.uuid4())}.jpg"
+
+    
+    local_image_path = os.path.join(image_directory, image_filename)
+    image.save(local_image_path)
+
+   
+    s3.upload_file(local_image_path, s3_bucket_name, f"images/{image_filename}")
+
+   
+    os.remove(local_image_path)
+
+    
+    data.append({"Image": image_filename, "Filename": filename, "Description": description, "Copyright": copyright})        
     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
-    return None  # Return None for no specific output
+
+    
+    if file_exists(csv_file_path):
+        
+        new_csv_file_path = generate_unique_csv_file()
+        data_df.to_csv(new_csv_file_path, index=False)
+        s3.upload_file(new_csv_file_path, s3_bucket_name, new_csv_file_path)
+    else:
+       
+        data_df.to_csv(csv_file_path, index=False)
+        s3.upload_file(csv_file_path, s3_bucket_name, csv_file_path)
+
+    return None 
 
 iface = gr.Interface(
     fn=describe_image,
@@ -34,7 +75,7 @@ iface = gr.Interface(
         "text",
         "text"
     ],
-    outputs=gr.Info("Data saved successfully"),  # Display the success message
+    outputs=gr.Info("Data saved successfully"), 
     live=False,
     title="Image Description App",
     description="Upload an image and provide three descriptions for it.",
@@ -45,4 +86,3 @@ iface = gr.Interface(
     )
 )
 
-iface.launch(share=True)