import os
import gradio as gr
from google import genai
from google.genai import types

# 初始化 Gemini client
api_key = os.getenv("GOOGLE_API_KEY")
client = genai.Client(api_key=api_key)
chat = client.chats.create(
    model="gemini-2.0-flash",
    config=types.GenerateContentConfig(
        system_instruction="You are a helpful assistant and always respond in Traditional Chinese.",
        temperature=0.2,
        max_output_tokens=256,
        top_p=0.8,
        top_k=40,
    ),
)
# 回應函數（符合 type="messages" 的格式）
def respond(message,history):
    response = chat.send_message(message)
    return response.text

# Gradio Chat Interface
demo = gr.ChatInterface(
    fn=respond,
    type="messages",
    title="<你的學號>",
)

if __name__ == "__main__":
    demo.launch()