Spaces:
Sleeping
Sleeping
File size: 535 Bytes
ae6ef27 1e67714 4d178a1 ae6ef27 4d178a1 25c59ac 4d178a1 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 |
import os
import google.generativeai as genai
# Configure Gemini client
genai.configure(api_key=os.environ["GEMINI_API_KEY"])
# ✅ Use your custom Gemini model
model = genai.GenerativeModel("gemini-2.0-flash-thinking-exp")
def ask_gpt(messages):
try:
prompt = "\n".join(
[f"{m['role'].capitalize()}: {m['content']}" for m in messages]
)
response = model.generate_content(prompt)
return response.text.strip()
except Exception as e:
return f"⚠️ Gemini Error: {str(e)}"
|