excel_kana / app.py
taskforce-dev's picture
Update app.py
a0e1209 verified
raw
history blame
1.46 kB
import gradio as gr
import pandas as pd
import json
import openai
# OpenAI APIキーを設定(あなたのAPIキーをここに入力してください)
openai.api_key = 'sk-proj-O9Qlw9lhCqMrWx4TNtLCT3BlbkFJDqrx6zKPfAv3ljoutKhi'
def excel_to_json_and_convert(file):
# エクセルファイルを読み込み
df = pd.read_excel(file)
# C列の2行目以降の値を取得
column_values = df.iloc[1:, 2].tolist()
# JSON形式に変換
json_result = json.dumps(column_values, ensure_ascii=False, indent=2)
# OpenAIのAPIに送信するプロンプトを作成
prompt = "次の内容を半角カタカタにへんかんしてください:\n" + json_result
# OpenAI GPT-3にリクエストを送信
response = openai.Completion.create(
engine="gpt-4o",
prompt=prompt,
max_tokens=500
)
# レスポンスからテキストを抽出
converted_text = response.choices[0].text.strip()
return converted_text
# Gradioインターフェースの設定
interface = gr.Interface(
fn=excel_to_json_and_convert,
inputs=gr.File(label="Upload Excel File"),
outputs=gr.Textbox(label="Converted Output"),
title="Excel to JSON Converter and OpenAI GPT-3 Converter",
description="Upload an Excel file and get JSON output of column C values converted to half-width Katakana by OpenAI GPT-3."
)
# Gradioアプリの実行
interface.launch()