import os
import gradio as gr
from openai import OpenAI
from openai.types.chat import ChatCompletionMessageParam

# 建立 OpenAI client（新版寫法）
client = OpenAI(api_key=os.getenv("OPENAI_API_KEY"))

# 使用微調模型名稱
MODEL = "ft:gpt-3.5-turbo-1106:atuan:kaxabu-v1:BubyDmJD"

def translate(input_text):
    if not input_text.strip():
        return "請輸入內容。"

    try:
        messages: list[ChatCompletionMessageParam] = [
            {"role": "system", "content": "你是一個能把華語翻譯成噶哈巫語的翻譯系統"},
            {"role": "user", "content": input_text}
        ]
        response = client.chat.completions.create(
            model=MODEL,
            messages=messages
        )
        return response.choices[0].message.content.strip()
    except Exception as e:
        return f"錯誤：{str(e)}"

iface = gr.Interface(
    fn=translate,
    inputs=gr.Textbox(lines=2, placeholder="請輸入華語句子"),
    outputs="text",
    title="噶哈巫語翻譯系統",
    description="請輸入中文，系統會自動翻譯為噶哈巫語（使用 AI 微調模型）"
)

iface.launch()