Update app.py

app.py

@@ -10,175 +10,53 @@ qa_pipeline = pipeline("question-answering", model="pierreguillou/bert-base-case
 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
+        try:
+            df = pd.read_excel(caminho_arquivo)
+            info = df.to_string(index=False)
+            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 completo da INTERCAST
-contexto = """
+# Dividindo o contexto em partes menores
+contexto_geral = """
 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.
+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. Localizada em Itaúna, Minas Gerais, Brasil, faz parte do grupo industrial norte-americano American, fundado em 1905.
 
 SEGMENTOS DA EMPRESA:
 A INTERCAST está presente em diversos segmentos, incluindo:
-Caminhões
-Equipamentos agrícolas
-Máquinas para construção
-Componentes para saneamento
+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.
+E-mail: [email protected]
+Telefone: +55 37 3249 7000
+Site: www.intercast.com.br
+Instagram: @intercastsa
+"""
+
+contexto_produtos = """
+LINHA DE PEÇAS:
+A INTERCAST desenvolve peças fundidas técnicas (Midcastings) de 20 a 250 kg, em ferro cinzento ou nodular. Alguns produtos incluem:
+Trombeta, Carcaça Central de Transmissão, Suporte, Cárter de Óleo, Quinta Roda, Hub Case, Caixa do Diferencial, Bottom Hidrante, Valvulas
 """
 
 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']
+    try:
+        if "ficha técnica" in message.lower() or "código 289" in message.lower():
+            resposta = qa_pipeline(question=message, context=ficha_tecnica)
+        elif any(palavra in message.lower() for palavra in ["peça", "produto", "linha"]):
+            resposta = qa_pipeline(question=message, context=contexto_produtos)
+        else:
+            resposta = qa_pipeline(question=message, context=contexto_geral)
+        return resposta['answer']
+    except Exception as e:
+        return f"Desculpe, ocorreu um erro ao processar sua pergunta: {str(e)}"
 
 interface = gr.ChatInterface(
     responder_pergunta,