import gradio as gr
from openai import OpenAI
import os

client = OpenAI(api_key=os.getenv("OPENAI_API_KEY"))

def chat(message, history):
    try:
        messages = [{"role": "system", "content": "You are Jarvis, a smart AI assistant."}]

        for user, bot in history:
            messages.append({"role": "user", "content": user})
            messages.append({"role": "assistant", "content": bot})

        messages.append({"role": "user", "content": message})

        response = client.responses.create(
            model="gpt-4o-mini",
            input=messages
        )

        reply = response.output_text
        return reply

    except Exception as e:
        return f"Error: {str(e)}"

iface = gr.ChatInterface(fn=chat)
iface.launch()