mychatbot / geminiai_with_history.py
huchiahsi's picture
Update geminiai_with_history.py
6e23e10 verified
raw
history blame contribute delete
804 Bytes
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()