import os
import gradio as gr
from huggingface_hub import InferenceClient

# Acessar a chave de API do secret
API_TOKEN = os.environ.get("HF_TOKEN")

# Inicialize o InferenceClient
client = InferenceClient(model="defog/sqlcoder", token=API_TOKEN)

# Função para gerar SQL a partir de linguagem natural
def gerar_sql(consulta_natural):
    # Envia a consulta para a API
    resposta = client.text_generation(prompt=consulta_natural, max_new_tokens=100)
    return resposta

# Interface Gradio
interface = gr.Interface(
    fn=gerar_sql,
    inputs="text",
    outputs="text",
    title="Gerador de SQL",
    description="Digite uma consulta em linguagem natural e gere a consulta SQL correspondente."
)

# Iniciar a interface
interface.launch()