import os
import openai
import gradio as gr

# Set your API key from Hugging Face secret
openai.api_key = os.environ.get("OPENAI_API_KEY")

def ask_agent(question):
    try:
        response = openai.ChatCompletion.create(
            model="gpt-4o",
            messages=[
                {"role": "system", "content": "You are a helpful, concise AI agent."},
                {"role": "user", "content": question}
            ],
            temperature=0.3,
        )
        return response['choices'][0]['message']['content'].strip()
    except Exception as e:
        return f"[ERROR] {str(e)}"

# Gradio UI
demo = gr.Interface(fn=ask_agent, inputs="text", outputs="text", title="Nargis AI Gaia Agent")
demo.launch()