import gradio as gr

import openai
import os

#from google.colab import userdata

import os

# OpenAI APIキーを設定
openai.api_key = os.getenv('OPENAI_API_KEY')

def generate_text(prompt):
    # GPT-3.5-turboを使用してテキスト生成（ChatCompletionを使用）
    response = openai.ChatCompletion.create(
      model="gpt-4", # モデルをgpt-4に変更
      temperature=0, #ここを0.2ずつ上げていく
      #max_tokens=8192,
      messages=[
          {"role": "system", "content": "You are a helpful assistant."},
          {"role": "user", "content": prompt},
      ],
    )
    return response.choices[0].message['content']

# Gradioインターフェースを作成
interface = gr.Interface(
    fn=generate_text,
    inputs=gr.Textbox(lines=100, placeholder="ここにプロンプトを入力してください..."),
    outputs=gr.Textbox(lines=100),
    title="コメントの再生成検証用",
    description="通教のLBFのコメント再生成用です"
).queue()

# Webサーバーを起動
interface.launch(auth=("llmjmam", "llmjmam"),share=True)