import gradio as gr
import pandas as pd
from io import BytesIO

def convert_file(input_file, conversion_type):
    # Check if a file was uploaded
    if input_file is None:
        raise ValueError("Please upload a file.")

    # Determine if input_file is a file-like object or a file path string.
    try:
        # Try reading from file-like object
        file_bytes = input_file.read()
        file_name = input_file.name
    except AttributeError:
        # If there's an AttributeError, treat input_file as a file path.
        file_name = input_file
        with open(file_name, "rb") as f:
            file_bytes = f.read()
    
    file_extension = file_name.lower().split('.')[-1]
    df = None
    output_file = None
    converted_format = None

    # Conversion: CSV to Parquet
    if conversion_type == "CSV to Parquet":
        if file_extension != "csv":
            raise ValueError("For CSV to Parquet conversion, please upload a CSV file.")
        df = pd.read_csv(BytesIO(file_bytes))
        output_file = "output.parquet"
        df.to_parquet(output_file, index=False)
        converted_format = "Parquet"
    # Conversion: Parquet to CSV
    elif conversion_type == "Parquet to CSV":
        if file_extension != "parquet":
            raise ValueError("For Parquet to CSV conversion, please upload a Parquet file.")
        df = pd.read_parquet(BytesIO(file_bytes))
        output_file = "output.csv"
        df.to_csv(output_file, index=False)
        converted_format = "CSV"
    else:
        raise ValueError("Invalid conversion type selected.")

    # Generate a preview of the top 10 rows
    preview = df.head(10).to_string(index=False)
    info_message = (
        f"Input file: {file_name}\n"
        f"Converted file format: {converted_format}\n\n"
        f"Preview (Top 10 Rows):\n{preview}"
    )
    return output_file, info_message

# Custom CSS for a modern and sleek look
custom_css = """
body {
    background-color: #f4f4f4;
    font-family: 'Helvetica Neue', Arial, sans-serif;
}
.gradio-container {
    max-width: 900px;
    margin: 40px auto;
    padding: 20px;
    background-color: #ffffff;
    border-radius: 12px;
    box-shadow: 0 8px 16px rgba(0,0,0,0.1);
}
h1, h2 {
    color: #333333;
}
.gradio-input, .gradio-output {
    margin-bottom: 20px;
}
.gradio-button {
    background-color: #4CAF50 !important;
    color: white !important;
    border: none !important;
    padding: 10px 20px !important;
    font-size: 16px !important;
    border-radius: 6px !important;
    cursor: pointer;
}
.gradio-button:hover {
    background-color: #45a049 !important;
}
"""

with gr.Blocks(css=custom_css, title="CSV <-> Parquet Converter") as demo:
    gr.Markdown("# CSV <-> Parquet Converter")
    gr.Markdown("Upload a CSV or Parquet file and select the conversion type. The app converts the file to the opposite format and displays a preview of the top 10 rows.")
    
    with gr.Row():
        with gr.Column(scale=1):
            input_file = gr.File(label="Upload CSV or Parquet File")
        with gr.Column(scale=1):
            conversion_type = gr.Radio(choices=["CSV to Parquet", "Parquet to CSV"], label="Conversion Type")
    
    convert_button = gr.Button("Convert", elem_classes=["gradio-button"])
    
    with gr.Row():
        output_file = gr.File(label="Converted File")
        preview = gr.Textbox(label="Preview (Top 10 Rows)", lines=15)
    
    convert_button.click(fn=convert_file, inputs=[input_file, conversion_type], outputs=[output_file, preview])

demo.launch()