import gradio as gr
from transformers import pipeline
import pandas as pd
import os

# Carregando o modelo pré-treinado
qa_pipeline = pipeline("question-answering", model="pierreguillou/bert-base-cased-squad-v1.1-portuguese")

# Função para ler o arquivo XLSX
def ler_ficha_tecnica():
    caminho_arquivo = "FTMO-AV21 - 6' MM BMD (04 FIGURAS).xlsx"
    if os.path.exists(caminho_arquivo):
        try:
            df = pd.read_excel(caminho_arquivo)
            info = "FICHA TÉCNICA:\n"
            for col in df.columns:
                info += f"{col}: {df[col].iloc[0]}\n"
            return info
        except Exception as e:
            return f"Erro ao ler o arquivo: {str(e)}"
    else:
        return "Arquivo da ficha técnica não encontrado."

ficha_tecnica = ler_ficha_tecnica()

contexto = """
SOBRE A EMPRESA:
A INTERCAST S.A. é uma empresa renomada no setor de fundição e metalurgia, fundada em 1970. Especializada na produção de peças fundidas em ferro e aço, atende principalmente os setores automotivo, agrícola e de construção. A empresa está localizada em Itaúna, cidade no estado de Minas Gerais, Brasil. Itaúna é conhecida como um importante centro industrial na região central de Minas Gerais.

A INTERCAST faz parte do grupo industrial norte-americano American, fundado em 1905, conhecido por sua tradição e pioneirismo. Este grupo tem mais de um século de experiência e reconhecimento global no setor industrial.

SEGMENTOS DA EMPRESA:
A INTERCAST está presente em diversos segmentos, incluindo:
- Caminhões
- Equipamentos agrícolas
- Máquinas para construção
- Componentes para saneamento

CLIENTES:
A empresa atende clientes no Brasil, América do Norte e Europa, demonstrando sua capacidade de atender mercados diversos e internacionais.

LINHA DE PEÇAS:
A INTERCAST desenvolve peças fundidas técnicas, conhecidas como Midcastings, que variam de 20 a 250 kg, feitas de ferro cinzento ou nodular. A empresa oferece soluções completas, incluindo usinagem, pré-montagem e pintura especial. Alguns dos produtos incluem:
- Trombeta
- Carcaça Central de Transmissão
- Suporte
- Cárter de Óleo
- Quinta Roda
- Hub Case
- Caixa do Diferencial
- Bottom Hidrante
- Válvulas

CONTATO:
Para obter informações da empresa, recomenda-se:
- E-mail: intercast@intercast.com.br
- Telefone: +55 37 3249 7000
- Site oficial: www.intercast.com.br
- Instagram: @intercastsa

""" + ficha_tecnica

def responder_pergunta(message, history):
    try:
        resposta = qa_pipeline(question=message, context=contexto)
        if resposta['score'] < 0.1:  # Se a confiança for baixa
            return "Desculpe, não tenho informações suficientes para responder essa pergunta com precisão. Posso ajudar com algo mais sobre a INTERCAST ou suas peças?"
        return resposta['answer']
    except Exception as e:
        return f"Desculpe, ocorreu um erro ao processar sua pergunta: {str(e)}"

interface = gr.ChatInterface(
    responder_pergunta,
    title="IA INTERCAST",
    description="Faça perguntas sobre a INTERCAST, seus produtos, história, localização e informações gerais."
)

interface.launch()