File size: 2,306 Bytes
e38d45e c3cefac e38d45e |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 |
import gradio as gr
import pandas as pd
import os
import zipfile
def process_csv(uploaded_file):
# Load the data from the uploaded file's byte stream
data = pd.read_csv(uploaded_file.name)
# Dictionary to store column name and its mapping of original values to codes
legend_dict = {}
# List to store the details of columns where data was added
data_added_details = []
# Loop through each column in the DataFrame
for col in data.columns:
# Check if the column is of type object (text-based) or if it's numerical with less than six unique options
if data[col].dtype == 'object' or (data[col].nunique() < 6 and pd.api.types.is_numeric_dtype(data[col])):
# Create a mapping of original values to codes, including NaN or blank values mapped to -9999
mapping = {value: code if pd.notna(value) else -9999 for code, value in enumerate(data[col].unique())}
legend_dict[col] = mapping
# Replace the values in the column with their respective codes
data[col] = data[col].map(mapping)
elif pd.api.types.is_numeric_dtype(data[col]) and any(pd.isna(data[col])):
# Replace with median
median_value = data[col].median()
data[col].fillna(median_value, inplace=True)
data_added_details.append([col, "Median", median_value])
# Name of the zip file based on uploaded file name
zip_name = "processed_files.zip"
# Save CSV files and add them to the zip file
with zipfile.ZipFile(zip_name, 'w') as zipf:
data.to_csv("modified_data.csv", index=False)
zipf.write("modified_data.csv")
legend_df = pd.DataFrame(list(legend_dict.items()), columns=['Column', 'Mapping'])
legend_df.to_csv("legend.csv", index=False)
zipf.write("legend.csv")
data_added_df = pd.DataFrame(data_added_details, columns=['Column', 'Method', 'Value Added'])
data_added_df.to_csv("data_added_details.csv", index=False)
zipf.write("data_added_details.csv")
return zip_name
# Gradio Interface
iface = gr.Interface(
fn=process_csv,
inputs=gr.inputs.File(type="file", label="Upload CSV File"),
outputs=gr.outputs.File(label="Download Processed Files"),
live=False
)
iface.launch() |