Update app.py

app.py

@@ -28,6 +28,27 @@ 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):
     # Translate question to English
     question_en = translate(user_question, pt_en_translator, tokenizer, source_lang="pt", target_lang="en")
@@ -39,6 +60,13 @@ def response(user_question, table_data):
 
     # Translate response to Portuguese
     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
+    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)
+    
     return response_pt
 
 # Streamlit interface