Update app.py

app.py

@@ -1,22 +1,189 @@
 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")
 
-# Contexto de exemplo sobre uma empresa fictícia
+# 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 = """
-A empresa INTERCAST S.A. é conhecida por atuar no setor de fundição e metalurgia. INTERCAST S.A. foi fundada em 1970, com o objetivo de fornecer peças fundidas de alta qualidade para diversos setores industriais. Ao longo das décadas, a empresa expandiu suas operações, incorporando novas tecnologias e processos para melhorar a eficiência e a qualidade de seus produtos. Especializa-se na produção de peças fundidas em ferro e aço, atendendo principalmente os setores automotivo, agrícola, e de construção. A INTERCAST sempre se destacou pela inovação, investindo continuamente em pesquisa e desenvolvimento para aprimorar seus processos de fundição. A empresa possui diversas certificações de qualidade, que atestam seu compromisso com a excelência e a satisfação dos clientes. Além de consolidar sua presença no mercado nacional, a INTERCAST também exporta seus produtos para diversos países, fortalecendo sua posição no cenário global. Mantém parcerias estratégicas com grandes empresas dos setores que atende, garantindo a continuidade e o crescimento sustentado. A empresa adota práticas sustentáveis em seus processos produtivos, minimizando o impacto ambiental e promovendo a responsabilidade social. A INTERCAST S.A. é uma empresa sólida e inovadora, com uma longa trajetória de sucesso no setor de fundição e metalurgia. Seu compromisso com a qualidade, inovação e sustentabilidade a posiciona como uma referência no mercado.
+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 intercast@intercast.com.br 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):
-    resposta = qa_pipeline(question=message, context=contexto)
+    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="Conheça a História da Intercast",
-    description="Use este assistente para aprender sobre a história da nossa empresa, Intercast"
+    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()