Update app.py

app.py

@@ -28,26 +28,6 @@ def translate(text, model, tokenizer, source_lang="pt", target_lang="en"):
     translated_text = tokenizer.decode(outputs[0], skip_special_tokens=True)
     return translated_text
 
-def find_previous_explanation(row, table_data):
-    current_date = pd.to_datetime(row['Ano e mês'])
-    current_group = row['Grupo']
-    
-    while True:
-        # Subtract one year to look for the previous year's same month
-        current_date = current_date - pd.DateOffset(years=1)
-        previous_row = table_data[
-            (table_data['Ano e mês'] == current_date.strftime('%Y-%m-01')) & 
-            (table_data['Grupo'] == current_group)
-        ]
-        
-        if not previous_row.empty and previous_row.iloc[0]['Explicação']:
-            return f"Em {current_date.strftime('%B de %Y')}, a explicação foi: {previous_row.iloc[0]['Explicação']}"
-        
-        # Stop if we've searched 10 years back without finding anything
-        if current_date.year < pd.to_datetime(row['Ano e mês']).year - 10:
-            break
-    
-    return "Não foi encontrada nenhuma explicação em anos anteriores."
 
 def response(user_question, table_data):
     # Traduz a pergunta para o inglês
@@ -62,7 +42,7 @@ def response(user_question, table_data):
     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" in user_question:
+    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:
@@ -70,11 +50,7 @@ def response(user_question, table_data):
             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
-
-    
+            response_pt = "Não foi possível encontrar uma explicação correspondente."  
     return response_pt
 
 # Streamlit interface