Update app.py

app.py

@@ -50,22 +50,30 @@ def find_previous_explanation(row, table_data):
     return "Não foi encontrada nenhuma explicação em anos anteriores."
 
 def response(user_question, table_data):
-    # Translate question to English
+    # Traduz a pergunta para o inglês
     question_en = translate(user_question, pt_en_translator, tokenizer, source_lang="pt", target_lang="en")
 
-    # Generate response in English
+    # Gera a resposta em inglês
     encoding = tapex_tokenizer(table=table_data, query=[question_en], padding=True, return_tensors="pt", truncation=True)
     outputs = tapex_model.generate(**encoding)
     response_en = tapex_tokenizer.batch_decode(outputs, skip_special_tokens=True)[0]
 
-    # Translate response to Portuguese
+    # Traduz a resposta para o português
     response_pt = translate(response_en, en_pt_translator, tokenizer, source_lang="en", target_lang="pt")
-    
-    # Check if the response contains a request for an explanation
+
+    # Verifica se a resposta contém uma solicitação de explicação
     if "Explicação" in user_question:
-        row = table_data[table_data['Explicação'] == response_pt].iloc[0]
-        if not row['Explicação']:
-            response_pt = find_previous_explanation(row, table_data)
+        # Tenta encontrar a linha correspondente na tabela
+        matching_rows = table_data[table_data['Explicação'] == response_pt]
+        if not matching_rows.empty:
+            row = matching_rows.iloc[0]
+            if not row['Explicação']:
+                response_pt = find_previous_explanation(row, table_data)
+        else:
+            response_pt = "Não foi possível encontrar uma explicação correspondente."
+    
+    return response_pt
+
     
     return response_pt