Spaces:
Sleeping
Sleeping
| import gradio as gr | |
| import google.generativeai as genai | |
| GOOGLE_API_KEY = "AIzaSyBrEIzU4mH733NMtkVs-axMHhRb0B-1h70" | |
| genai.configure(api_key=GOOGLE_API_KEY) | |
| def chat_with_gemini(message, persona): | |
| try: | |
| # 列出可用模型 | |
| models = genai.list_models() | |
| # 选择支持 generate_content 的模型 | |
| model_name = None | |
| for m in models: | |
| if "generateContent" in m.supported_generation_methods: | |
| model_name = m.name | |
| break | |
| if not model_name: | |
| return "没有可用的 Gemini 聊天模型，请检查 API Key 权限或 Google Cloud 配置。" | |
| model = genai.GenerativeModel(model_name) | |
| response = model.generate_content([f"你是一个{persona}。", message]) | |
| return response.text | |
| except Exception as e: | |
| return f"请求或解析出错: {str(e)}" | |
| with gr.Blocks() as demo: | |
| persona = gr.Textbox(label="AI Persona 个性描述") | |
| message = gr.Textbox(label="你的问题") | |
| output = gr.Textbox(label="AI 回复") | |
| btn = gr.Button("发送") | |
| btn.click(chat_with_gemini, inputs=[message, persona], outputs=output) | |
| demo.launch() |