initial project

app.py

@@ -0,0 +1,156 @@
+import boto3
+import gradio as gr
+import os
+from dotenv import load_dotenv
+load_dotenv()
+
+ACCOUNT_ID = os.getenv('AWS_ACCOUNT_ID', '')
+AWS_ACCESS_KEY_ID = os.getenv('AWS_ACCESS_KEY_ID', '')
+AWS_SECRET_ACCESS_KEY = os.getenv('AWS_SECRET_ACCESS_KEY', '')
+AWS_DEFAULT_REGION = os.getenv('AWS_DEFAULT_REGION', '')
+AWS_S3_BUCKET_NAME = os.getenv('AWS_S3_BUCKET_NAME', '')
+
+is_dev = os.environ.get('ENV', 'DEV') != 'PROD'
+
+s3_client = boto3.client(
+    service_name='s3',
+    region_name=AWS_DEFAULT_REGION,
+    aws_access_key_id=AWS_ACCESS_KEY_ID,
+    aws_secret_access_key=AWS_SECRET_ACCESS_KEY
+)
+
+def auth(username, password):
+    u = os.environ.get('USERNAME')
+    p = os.environ.get('PASSWORD')
+    return (username == u and password == p)
+
+# Function to list images in a specific S3 bucket
+def list_images_in_bucket(bucket_name):
+    response = s3_client.list_objects_v2(Bucket=bucket_name)
+    return [obj['Key'] for obj in response.get('Contents', []) if obj['Key'].lower().endswith(('.png', '.jpg', '.jpeg'))]
+
+# Function to get image URLs
+def get_image_urls(bucket_name, image_keys):
+    return [s3_client.generate_presigned_url('get_object', Params={'Bucket': bucket_name, 'Key': key}) for key in image_keys]
+
+# Function to upload files to S3
+def upload_files(files, progress=gr.Progress()):
+    for file in progress.tqdm(files, desc="Uploading..."):
+        file_name = os.path.basename(file.name)
+        with open(file.name, 'rb') as f:
+            print(f'Uploading "{file_name}"...')
+            s3_client.upload_fileobj(f, AWS_S3_BUCKET_NAME, file_name)
+
+    return display_images()
+
+# Function to fetch and display images
+def display_images():
+    image_keys = list_images_in_bucket(AWS_S3_BUCKET_NAME)
+    image_urls = get_image_urls(AWS_S3_BUCKET_NAME, image_keys)
+    # print(image_keys)
+    # print(image_urls)
+    return image_urls, image_keys
+
+# Function to delete a file from S3
+def delete_image(file_key):
+    if isinstance(file_key, list):
+        for k in file_key:
+            print(f'Removing "{k}"...')
+            s3_client.delete_object(Bucket=AWS_S3_BUCKET_NAME, Key=k)
+    else:
+        s3_client.delete_object(Bucket=AWS_S3_BUCKET_NAME, Key=file_key)
+    return display_images()
+
+def update_dropdown_image(evt: gr.SelectData):
+    # print(f"You selected {evt.value} at {evt.index} from {evt.target}")
+    # print(f"You selected {evt.value['image']['orig_name']}")
+    return evt.value['image']['orig_name']
+
+# ---------------------------------------------------------------------------
+
+# Function to list images in a specific S3 bucket
+def list_files_in_bucket(bucket_name):
+    response = s3_client.list_objects_v2(Bucket=bucket_name)
+    return [obj['Key'] for obj in response.get('Contents', [])]
+
+# Function to get image URLs
+def get_files_urls(bucket_name, file_keys):
+    return [s3_client.generate_presigned_url('get_object', Params={'Bucket': bucket_name, 'Key': key}) for key in file_keys]
+
+# Function to fetch and display files
+def display_files():
+    file_keys = list_files_in_bucket(AWS_S3_BUCKET_NAME)
+    file_urls = get_files_urls(AWS_S3_BUCKET_NAME, file_keys)
+    # print(file_keys)
+    # print(file_urls)
+    return file_urls, file_keys
+
+# Function to delete a file from S3
+def delete_files(file_key):
+    if isinstance(file_key, list):
+        for k in file_key:
+            print(f'Removing "{k}"...')
+            s3_client.delete_object(Bucket=AWS_S3_BUCKET_NAME, Key=k)
+    else:
+        s3_client.delete_object(Bucket=AWS_S3_BUCKET_NAME, Key=file_key)
+    return display_files()
+
+def update_dropdown_files(evt: gr.SelectData):
+    #print(f"You selected {evt.value} at {evt.index} from {evt.target}")
+    return evt.value.split('AWSAccessKeyId')[0]
+        
+# Create a Gradio interface
+with gr.Blocks() as demo:
+
+    gr.Markdown("# **Data collector**")
+    
+    with gr.Row():
+        with gr.Column(scale=1):
+            gr.Markdown("""## **1. Upload files**
+
+1. โยนไฟล์ (drag/drop) ไปยัง ***ไฟล์ list ทาง panel ทางด้านซ้าย*** ได้มากกว่า 1 ไฟล์
+2. กด **Upload Files**
+""")
+            file_uploader = gr.File(label="Upload Images", file_count="multiple", height=400)
+            upload_btn = gr.Button("Upload Files")
+
+        with gr.Column(scale=3):
+            with gr.Tabs():
+                with gr.Tab('Images'):
+                    gr.Markdown("""## **2. Load images**
+กดที่ **Refresh Gallery**
+""")
+                    img_gallery = gr.Gallery(label="Images from S3", columns=9, height=600, interactive=True)
+                    img_refresh_btn = gr.Button("Refresh Gallery")
+                    
+                    gr.Markdown("""### ***Delete images***
+กรณีต้องการลบไฟล์ ท่านสามารถเลือกภาพที่สนใจ แล้วกด **Delete Selected Files** แต่ถ้าต้องการลบมากกว่า 1 file ให��ใส่ชื่อไฟล์ คั่นด้วย comma (,)
+""")
+                    img_selected_image = gr.Dropdown(choices=[], label="Select Image to Delete", allow_custom_value=True)
+                    img_delete_btn = gr.Button("Delete Selected Files")
+
+                with gr.Tab('Files'):
+                    gr.Markdown("""## **2. Load files**
+กดที่ **Refresh Gallery**
+""")
+                    file_gallery = gr.File(label="Files from S3", interactive=True)
+                    file_refresh_btn = gr.Button("Refresh Gallery")
+                    
+                    gr.Markdown("""### ***Delete files***
+กรณีต้องการลบไฟล์ ท่านสามารถเลือกภาพที่สนใจ แล้วกด **Delete Selected Files** แต่ถ้าต้องการลบมากกว่า 1 file ให้ใส่ชื่อไฟล์ คั่นด้วย comma (,)
+""")
+                    file_selected = gr.Dropdown(choices=[], label="Select File to Delete", allow_custom_value=True)
+                    file_delete_btn = gr.Button("Delete Selected Files")
+
+        upload_btn.click(fn=upload_files, inputs=[file_uploader], outputs=[img_gallery, img_selected_image])
+
+        img_refresh_btn.click(fn=display_images, outputs=[img_gallery, img_selected_image])
+        img_delete_btn.click(fn=delete_image, inputs=img_selected_image, outputs=[img_gallery, img_selected_image])
+        img_gallery.select(fn=update_dropdown_image, outputs=img_selected_image)
+
+        file_refresh_btn.click(fn=display_files, outputs=[file_gallery, file_selected])
+        file_delete_btn.click(fn=delete_files, inputs=file_selected, outputs=[file_gallery, file_selected])
+        file_gallery.select(fn=update_dropdown_files, outputs=file_selected)
+
+# Launch the Gradio app
+demo.launch(auth=auth if not is_dev else None)

requirements.txt

@@ -0,0 +1,4 @@
+boto3
+gradio
+pillow
+python-dotenv