app.py

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):
        df = pd.read_excel(caminho_arquivo)
        info = df.to_string(index=False)
        return info
    else:
        return "Arquivo da ficha técnica não encontrado."

ficha_tecnica = ler_ficha_tecnica()

# Contexto completo da INTERCAST
contexto = """
SOBRE A EMPRESA:
A INTERCAST S.A. é uma empresa renomada no setor de fundição e metalurgia, fundada em 1970. Desde o início, seu objetivo tem sido fornecer peças fundidas de alta qualidade para diversos setores industriais. Ao longo dos anos, a INTERCAST expandiu suas operações, incorporando novas tecnologias e processos para melhorar a eficiência e a qualidade de seus produtos.
Especializada na produção de peças fundidas em ferro e aço, a INTERCAST atende principalmente os setores automotivo, agrícola e de construção. A empresa é reconhecida por sua inovação, investindo continuamente em pesquisa e desenvolvimento para aprimorar seus processos de fundição. Com diversas certificações de qualidade, a INTERCAST demonstra seu compromisso com a excelência e a satisfação dos clientes.
Além de consolidar sua presença no mercado nacional, a INTERCAST exporta seus produtos para diversos países, fortalecendo sua posição no cenário global. A empresa mantém parcerias estratégicas com grandes empresas dos setores que atende, garantindo a continuidade e o crescimento sustentado. 
Em termos de responsabilidade social e ambiental, a INTERCAST adota práticas sustentáveis em seus processos produtivos, minimizando o impacto ambiental e promovendo o desenvolvimento social. Com uma trajetória de sucesso, a INTERCAST S.A. se destaca como uma referência no mercado, graças ao seu compromisso com a qualidade, inovação e sustentabilidade.

História e Fundação: A INTERCAST S.A. é uma fundição de ferro localizada em Itaúna, Minas Gerais. A empresa é especializada em Midcastings e tem um compromisso com a responsabilidade e o desenvolvimento de seus trabalhadores, clientes e comunidade. A INTERCAST faz parte do grupo industrial norte-americano American, fundado em 1905, conhecido por sua tradição e pioneirismo.

Localização: Itaúna, Minas Gerais, Brasil.
Grupo: A empresa é parte do grupo American, um conglomerado industrial com mais de um século de experiência e reconhecimento global.

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.

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
Valvulas

TODAS AS PEÇAS:
COD	PÇ
134	Ponte 212
194	Churrasq.Brasil
266	Port. Plan. 086
284	12 MM ND
289	8 MM ND
298	6FM-1ND
299	6 MM ND
300	Hyd. Base 511
309	Hyd. Base 810
318	Ponte 216
330	Carc. Redu. 704
337	Suporte 6200993M3  
338	ACW2524060
340	DIF.3200W1323
341	Dif.3200X1870U
361	Tamp Carter 170
362	Cx.Dif. 023672
378	Ponte 098
379	Ponte 075
380	CARCAÇA 128 III
382	Cx.Dif. 122247
383	Cx.Dif. 023642
384	CX.DIf. 1586
385	Cx.Dif. 1864
390	CARCAÇA 054 I
400	Trombeta 1003
403	CARC. EMB. 155
414	Port. Plan. 536
436	Ponte 03262
457	Disco 5788
458	DIF. 3200K2117 (1893)
479	Carcaça 6701
480	HUB 8408
483	Mancal 176
484	Mancal 176
485	10 MM ND
486	DIF. 1702
487	SUPORTE 403
491	Porta Engr.6382
493	Carc.Cent.334P5
495	Carcaça 6019-0
518	Carter 980
519	Carc.tub.710-0
525	BLOCO MOT.4313
527	CX. ACX2051400
534	Tampa PTO 7065
536	Torre Giro 4592
537	Peso 230
538	Peso 080
539	Suporte 908
543	Carter 021 B
545	Espaçador 9316
565	Carcaça 695
566	Port. Plan. 275
567	Port. Plan 276
571	Carc.tub.930
572	Roda 181
574	Carc. Redu. 733
575	Carc. Redu. 735
580	Roda 247
581	Suporte 222
582	Carcaça 289
591	Port. Plan. 143
594	Carc. Art. 226
595	Carc. Art. 227
597	Carc. Art. 245
598	Carc. Art. 244
621	HUB 1854
622	HUB 1853
634	Carc. Art. 627
635	Carc. Art. 628
636	Carcaça 181
637	Suporte 878
646	CARCAÇA 983
653	Carcaça 229
654	Trombeta 047
655	Trombeta 048
660	Trombeta 4303
661	Trombeta 4300
662	Suporte 686M5
664	Peso 469
667	Braçadeira 689
668	Braçadeira 690
670	PESO 782
676	Trombeta 832
677	Trombeta 1421
678	Trombeta 1422
679	Peso 153
681	Carcaça 198
682	Carcaça 071
684	SUPORTE 173
704	Cubo 266M1
705	Flange 267M3
706	Suporte 306JO
726	Trombeta 478
65005	MJ BASE 12662
65006	MJ BASE 12663
100066	Ponte 129
100067	Carcaça 703
100805	Carc. Art. 025
100806	Carc. Art. 026
100852	Disco 1978M1
100854	Centro 1071P1
100855	Suporte 2212M3
100856	Suporte 1842M2
101571	Trombeta 983
101572	Trombeta 984
101578	Suporte 767
103540	TAMPA FRONTAL 2698/5420

FICHA TECNICA:
""" + ficha_tecnica + """

CONTATO:
Para obter informações da empresa, recomenda-se acessar diretamente o site da empresa ou entrar em contato através do e-mail [email protected] ou pelo telefone [+55 37 3249 7000].
Para mais detalhes, você pode visitar o site oficial da INTERCAST (www.intercast.com.br) ou seguir o Instagram @intercastsa para atualizações e informações adicionais.
"""

def responder_pergunta(message, history):
    if "ficha técnica" in message.lower() or "código 289" in message.lower():
        resposta = qa_pipeline(question=message, context=ficha_tecnica)
    else:
        resposta = qa_pipeline(question=message, context=contexto)
    return resposta['answer']

interface = gr.ChatInterface(
    responder_pergunta,
    title="Assistente INTERCAST S.A.",
    description="Faça perguntas sobre a INTERCAST S.A., seus produtos, história, informações gerais e detalhes da ficha técnica."
)

interface.launch()