Update app.py

app.py

@@ -7,42 +7,46 @@ import io
 import base64
 import tempfile
 
+
+css = '''
+.gradio-container{max-width: 600px !important}
+h1{text-align:center}
+'''
+
+DESCRIPTIONz= """## Image to Parquet 📂
+
+"""
+
 def image_to_parquet(files):
-    # List to store image data
+
     image_data = []
 
     for file_info in files:
-        # Read image
         with open(file_info, "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": file_info, "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
+
     with tempfile.NamedTemporaryFile(delete=False, suffix=".parquet") as tmp_file:
         pq.write_table(table, tmp_file)
         parquet_file_path = tmp_file.name
     
-    # Return Parquet file path
     return parquet_file_path
 
-# Gradio interface
-with gr.Blocks() as demo:
+with gr.Blocks(css=css, theme="bethecloud/storj_theme") as demo:
+    gr.Markdown(DESCRIPTIONz)
     with gr.Row():
         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")
+    convert_button = gr.Button("Convert Image to Parquet")
     
     convert_button.click(fn=image_to_parquet, inputs=[image_input], outputs=[download_button])