Update app.py

app.py

@@ -6,18 +6,20 @@ from PIL import Image
 import io
 import base64
 
-def image_to_parquet(images):
+def image_to_parquet(files):
     # 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")
+    for file_info in files:
+        # Read image
+        with open(file_info.name, "rb") as image_file:
+            img = Image.open(image_file)
+            buffered = io.BytesIO()
+            img.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})
+        image_data.append({"name": file_info.orig_name, "data": img_str})
     
     # Create DataFrame
     df = pd.DataFrame(image_data)
@@ -39,8 +41,8 @@ def download_parquet(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")
+        image_input = gr.File(label="Upload Images", type="filepath", file_count="multiple", file_types=["image"])
+        download_button = gr.File(label="Download Parquet File", interactive=False)
         
     convert_button = gr.Button("Convert to Parquet")