Create app.py

app.py

@@ -0,0 +1,49 @@
+import gradio as gr
+import pandas as pd
+import pyarrow as pa
+import pyarrow.parquet as pq
+from PIL import Image
+import io
+import base64
+
+def image_to_parquet(images):
+    # List to store image data
+    image_data = []
+
+    for image in images:
+        # Convert image to bytes
+        buffered = io.BytesIO()
+        image.save(buffered, format="PNG")
+        img_str = base64.b64encode(buffered.getvalue()).decode("utf-8")
+        
+        # Store image data and name
+        image_data.append({"name": image.name, "data": img_str})
+    
+    # Create DataFrame
+    df = pd.DataFrame(image_data)
+    
+    # Convert DataFrame to PyArrow Table
+    table = pa.Table.from_pandas(df)
+    
+    # Save table as Parquet file
+    parquet_buffer = io.BytesIO()
+    pq.write_table(table, parquet_buffer)
+    
+    # Return Parquet file
+    parquet_buffer.seek(0)
+    return parquet_buffer
+
+def download_parquet(file):
+    return file
+
+# Gradio interface
+with gr.Blocks() as demo:
+    with gr.Row():
+        image_input = gr.File(label="Upload Images", type="file", file_count="multiple", file_types=["image"])
+        download_button = gr.File(label="Download Parquet File")
+        
+    convert_button = gr.Button("Convert to Parquet")
+    
+    convert_button.click(fn=image_to_parquet, inputs=[image_input], outputs=[download_button])
+
+demo.launch()