Update app.py

app.py

@@ -150,6 +150,68 @@ def get_index_price(index: str) -> str:
         return f"The latest closing price for {index} is {price:.2f}."
     except Exception as e:
         return f"Error fetching index price: {str(e)}"
+
+@tool
+def get_interest_rates() -> str:
+    """
+    Fetches the latest Selic, CDI, and IPCA rates.
+
+    Returns:
+        The latest values for Brazil's key interest rates.
+    """
+    import requests
+    try:
+        response = requests.get("https://api.bcb.gov.br/dados/serie/bcdata.sgs.11/dados/ultimos/1?formato=json")  # SELIC
+        selic = response.json()[0]['valor']
+
+        response = requests.get("https://api.bcb.gov.br/dados/serie/bcdata.sgs.12/dados/ultimos/1?formato=json")  # CDI
+        cdi = response.json()[0]['valor']
+
+        response = requests.get("https://api.bcb.gov.br/dados/serie/bcdata.sgs.433/dados/ultimos/1?formato=json")  # IPCA
+        ipca = response.json()[0]['valor']
+
+        return (f"Selic atual: {selic}% ao ano\n"
+                f"CDI atual: {cdi}% ao ano\n"
+                f"IPCA acumulado: {ipca}%")
+    except Exception as e:
+        return f"Erro ao buscar taxas de juros: {str(e)}"
+
+
+@tool
+def compare_fixed_income(amount: float, period: int, cdi_percentage: float) -> str:
+    """
+    Compares the return of a CDB vs. savings over a given period.
+
+    Args:
+        amount: The initial investment amount.
+        period: The investment duration in months.
+        cdi_percentage: The CDB return as a percentage of the CDI.
+    
+    Returns:
+        The final amount in CDB vs. savings.
+    """
+    import requests
+    try:
+        # Buscar CDI atual
+        response = requests.get("https://api.bcb.gov.br/dados/serie/bcdata.sgs.12/dados/ultimos/1?formato=json")
+        cdi = float(response.json()[0]['valor']) / 100
+
+        # Poupança rende 70% da Selic quando Selic > 8.5%
+        response = requests.get("https://api.bcb.gov.br/dados/serie/bcdata.sgs.11/dados/ultimos/1?formato=json")
+        selic = float(response.json()[0]['valor']) / 100
+        poupanca_rendimento = 0.7 * selic if selic > 0.085 else 0.005
+
+        # Calcular rendimentos
+        cdb_final = amount * ((1 + (cdi * cdi_percentage / 100)) ** (period / 12))
+        poupanca_final = amount * ((1 + poupanca_rendimento) ** (period / 12))
+
+        return (f"Após {period} meses:\n"
+                f"CDB ({cdi_percentage}% do CDI): R$ {cdb_final:.2f}\n"
+                f"Poupança: R$ {poupanca_final:.2f}")
+    except Exception as e:
+        return f"Erro ao calcular rendimentos: {str(e)}"
+
+
         
 final_answer = FinalAnswerTool()