app.py

# -*- coding: utf-8 -*-
"""app.ipynb

Automatically generated by Colaboratory.

Original file is located at
    https://colab.research.google.com/drive/1D-iexFe9RSsGXN4BhAqS--aCPbsTe6Bj
"""
import csv
import gradio as gr
import os
from datasets import load_dataset
# CSVファイルのパスを設定
dataset_file = "test.csv"

def create_dataset(question, output, input="", username=""):
    # データセットの追加回数をカウントする変数
    count = 0

    # CSVファイルにデータを書き込む
    with open(dataset_file, 'a', newline='') as file:
        writer = csv.writer(file)
        writer.writerow([question, input, output, username])

        # データセットの追加回数をカウント
        with open(dataset_file, 'r') as file:
            reader = csv.reader(file)
            for row in reader:
                if row[3] == username:  # ユーザー名でフィルタリング
                    count += 1

    # データセットの追加回数をメッセージとして返す
    return f"{username}さんは、データセットを合計 {count} 回作成しました！ありがとうございます！"

# 入力インターフェースを作成
inputs = [
    gr.inputs.Textbox(label="質問", lines=2),
    gr.inputs.Textbox(label="回答"),
    gr.inputs.Textbox(label="入力 (オプション)"),
    gr.inputs.Textbox(label="ユーザー名")
]

# 出力インターフェースを作成
output = gr.outputs.Textbox(label="ステータス")

# インターフェースを作成
interface = gr.Interface(fn=create_dataset, inputs=inputs, outputs=output)

# インターフェースを起動
interface.launch()