app.py

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

def process_csv(file_obj):
    # アップロードされたCSVファイルをPandas DataFrameとして読み込む
    df = pd.read_csv(file_obj)
    
    # 'age'列の値を2倍にする
    df['age'] = df['age'].apply(lambda x: x * 2)
    
    # 結果を文字列として出力するため、StringIOを使用してCSV形式に変換
    output = StringIO()
    df.to_csv(output, index=False)
    output.seek(0)
    
    # StringIOオブジェクトの内容をバイト列として返す
    return output.getvalue().encode('utf-8')

# Gradioインターフェースの定義
iface = gr.Interface(fn=process_csv, 
                     inputs=gr.File(label="Upload CSV"), 
                     outputs=gr.File(label="Download Processed CSV", type="file"), 
                     description="Upload a CSV with 'name' and 'age' columns. This app will double the 'age' and return the processed CSV.")

# アプリの起動
iface.launch()