Spaces:
Sleeping
Sleeping
| 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() |