import gradio as gr
from google import genai
import os

API_KEY = os.getenv("GOOGLE_API_KEY")

if not API_KEY:
    raise Exception("❌ API key missing")

client = genai.Client(api_key=API_KEY)

def chat(message, history):
    try:
        response = client.models.generate_content(
            model="gemini-1.5-flash-latest",  # ✅ ONLY THIS
            contents=message
        )
        return response.text
    except Exception as e:
        return f"❌ Error: {str(e)}"

gr.ChatInterface(fn=chat, title="Jarvis AI 🤖").launch()