Update app.py

app.py

@@ -12,7 +12,9 @@ def ler_ficha_tecnica():
     if os.path.exists(caminho_arquivo):
         try:
             df = pd.read_excel(caminho_arquivo)
-            info = df.to_string(index=False)
+            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)}"
@@ -21,47 +23,56 @@ def ler_ficha_tecnica():
 
 ficha_tecnica = ler_ficha_tecnica()
 
-# Dividindo o contexto em partes menores
-contexto_geral = """
+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. Localizada em Itaúna, Minas Gerais, Brasil, faz parte do grupo industrial norte-americano American, fundado em 1905.
+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
+- 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.
+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:
-E-mail: intercast@intercast.com.br
-Telefone: +55 37 3249 7000
-Site: www.intercast.com.br
-Instagram: @intercastsa
-"""
+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
 
-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
-"""
+""" + ficha_tecnica
 
 def responder_pergunta(message, history):
     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)
+        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="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."
+    title="IA INTERCAST",
+    description="Faça perguntas sobre a INTERCAST, seus produtos, história, localização e informações gerais."
 )
 
 interface.launch()