Update index.html

index.html

@@ -21,59 +21,64 @@
 
     <script type="module">
         import { pipeline, env } from 'https://cdn.jsdelivr.net/npm/@xenova/[email protected]';
-
+    
         // Configuração para desativar modelos locais
         env.allowLocalModels = false;
-
+    
         // Reconhecimento de Entidades Nomeadas (NER)
         const textInput = document.getElementById('text-input');
         const analyzeTextButton = document.getElementById('analyze-text');
         const textOutput = document.getElementById('text-output');
-
-        // Carregar modelo NER com quantização q4
-        let nerModel;
-        try {
-            textOutput.textContent = 'Carregando modelo de NER...';
-            nerModel = await pipeline('ner', 'Xenova/distilbert-base-multilingual-cased-ner-hrl', {
-                dtype: 'q4'
-            });
-            textOutput.textContent = 'Modelo de NER pronto!';
-        } catch (error) {
-            console.error('Erro ao carregar o modelo:', error);
-            textOutput.textContent = 'Erro ao carregar o modelo. Tente novamente mais tarde.';
-        }
-
-        analyzeTextButton.addEventListener('click', async () => {
-            const inputText = textInput.value.trim();
-            if (!inputText) {
-                textOutput.textContent = 'Por favor, insira um texto para análise.';
-                return;
-            }
-
-            textOutput.textContent = 'Analisando...';
+    
+        // Verifique se os elementos HTML foram encontrados
+        if (!textInput || !analyzeTextButton || !textOutput) {
+            console.error('Elementos HTML necessários não foram encontrados.');
+        } else {
+            let nerModel;
             try {
-                const nerOutput = await nerModel(inputText);
-                renderEntities(nerOutput);
+                textOutput.textContent = 'Carregando modelo de NER...';
+                nerModel = await pipeline('ner', 'Xenova/distilbert-base-multilingual-cased-ner-hrl', {
+                    dtype: 'q4',
+                });
+                textOutput.textContent = 'Modelo de NER pronto!';
             } catch (error) {
-                console.error('Erro durante a análise:', error);
-                textOutput.textContent = 'Erro durante a análise. Verifique o texto inserido.';
+                console.error('Erro ao carregar o modelo:', error);
+                textOutput.textContent = 'Erro ao carregar o modelo. Tente novamente mais tarde.';
+                return; // Saia se o modelo não puder ser carregado
             }
-        });
-
-        function renderEntities(entities) {
-            textOutput.innerHTML = '';
-            entities.forEach(entity => {
-                const { word, entity_group, score } = entity;
-
-                const entityElement = document.createElement('div');
-                entityElement.className = 'entity';
-                entityElement.innerHTML = `
-                    <strong>Palavra:</strong> ${word} <br>
-                    <strong>Entidade:</strong> ${entity_group} <br>
-                    <strong>Confiança:</strong> ${(score * 100).toFixed(2)}%
-                `;
-                textOutput.appendChild(entityElement);
+    
+            analyzeTextButton.addEventListener('click', async () => {
+                const inputText = textInput.value.trim();
+                if (!inputText) {
+                    textOutput.textContent = 'Por favor, insira um texto para análise.';
+                    return;
+                }
+    
+                textOutput.textContent = 'Analisando...';
+                try {
+                    const nerOutput = await nerModel(inputText);
+                    renderEntities(nerOutput);
+                } catch (error) {
+                    console.error('Erro durante a análise:', error);
+                    textOutput.textContent = 'Erro durante a análise. Verifique o texto inserido.';
+                }
             });
+    
+            function renderEntities(entities) {
+                textOutput.innerHTML = '';
+                entities.forEach(entity => {
+                    const { word, entity_group, score } = entity;
+    
+                    const entityElement = document.createElement('div');
+                    entityElement.className = 'entity';
+                    entityElement.innerHTML = `
+                        <strong>Palavra:</strong> ${word} <br>
+                        <strong>Entidade:</strong> ${entity_group} <br>
+                        <strong>Confiança:</strong> ${(score * 100).toFixed(2)}%
+                    `;
+                    textOutput.appendChild(entityElement);
+                });
+            }
         }
     </script>
 </body>