Update app.py

app.py

@@ -32,25 +32,16 @@ def translate(text, model, tokenizer, source_lang="pt", target_lang="en"):
 def response(user_question, table_data):
     # Traduz a pergunta para o inglês
     question_en = translate(user_question, pt_en_translator, tokenizer, source_lang="pt", target_lang="en")
+    print(question_en)
 
     # 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]
+    print(response_en)
 
     # Traduz a resposta para o português
     response_pt = translate(response_en, en_pt_translator, tokenizer, source_lang="en", target_lang="pt")
-
-    # Verifica se a resposta contém uma solicitação de explicação
-    if 'Explicação' or 'explicação' in user_question:
-        # 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
 
 # Streamlit interface