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() | |