Update app.py

app.py

@@ -1,69 +1,236 @@
-from smolagents import CodeAgent,DuckDuckGoSearchTool, HfApiModel,load_tool,tool
-import datetime
-import requests
-import pytz
-import yaml
-from tools.final_answer import FinalAnswerTool
-
-from Gradio_UI import GradioUI
-
-# Below is an example of a tool that does nothing. Amaze us with your creativity !
-@tool
-def my_custom_tool(arg1:str, arg2:int)-> str: #it's import to specify the return type
-    #Keep this format for the description / args / args description but feel free to modify the tool
-    """A tool that does nothing yet 
-    Args:
-        arg1: the first argument
-        arg2: the second argument
-    """
-    return "What magic will you build ?"
-
-@tool
-def get_current_time_in_timezone(timezone: str) -> str:
-    """A tool that fetches the current local time in a specified timezone.
-    Args:
-        timezone: A string representing a valid timezone (e.g., 'America/New_York').
-    """
-    try:
-        # Create timezone object
-        tz = pytz.timezone(timezone)
-        # Get current time in that timezone
-        local_time = datetime.datetime.now(tz).strftime("%Y-%m-%d %H:%M:%S")
-        return f"The current local time in {timezone} is: {local_time}"
-    except Exception as e:
-        return f"Error fetching time for timezone '{timezone}': {str(e)}"
-
-
-final_answer = FinalAnswerTool()
-
-# If the agent does not answer, the model is overloaded, please use another model or the following Hugging Face Endpoint that also contains qwen2.5 coder:
-# model_id='https://pflgm2locj2t89co.us-east-1.aws.endpoints.huggingface.cloud' 
-
-model = HfApiModel(
-max_tokens=2096,
-temperature=0.5,
-model_id='Qwen/Qwen2.5-Coder-32B-Instruct',# it is possible that this model may be overloaded
-custom_role_conversions=None,
-)
-
-
-# Import tool from Hub
-image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
-
-with open("prompts.yaml", 'r') as stream:
-    prompt_templates = yaml.safe_load(stream)
-    
-agent = CodeAgent(
-    model=model,
-    tools=[final_answer], ## add your tools here (don't remove final answer)
-    max_steps=6,
-    verbosity_level=1,
-    grammar=None,
-    planning_interval=None,
-    name=None,
-    description=None,
-    prompt_templates=prompt_templates
-)
-
-
-GradioUI(agent).launch()
+import gradio as gr
+from datetime import date
+
+
+# =========================
+# FUNÇÕES
+# =========================
+
+def adicionar_transacao(tipo, descricao, categoria, valor, data, transacoes):
+    if not descricao:
+        return transacoes, atualizar_resumo(transacoes), "Digite uma descrição."
+
+    if not valor or valor <= 0:
+        return transacoes, atualizar_resumo(transacoes), "Digite um valor maior que zero."
+
+    nova_transacao = {
+        "tipo": tipo,
+        "descricao": descricao,
+        "categoria": categoria,
+        "valor": valor,
+        "data": data
+    }
+
+    transacoes.append(nova_transacao)
+
+    return (
+        transacoes,
+        atualizar_resumo(transacoes),
+        f"✅ {tipo} adicionada com sucesso!"
+    )
+
+
+def atualizar_resumo(transacoes):
+    receitas = sum(
+        t["valor"] for t in transacoes
+        if t["tipo"] == "Receita"
+    )
+
+    despesas = sum(
+        t["valor"] for t in transacoes
+        if t["tipo"] == "Despesa"
+    )
+
+    saldo = receitas - despesas
+
+    texto = f"""
+## 💰 Resumo financeiro
+
+**Receitas:** R$ {receitas:,.2f}
+
+**Despesas:** R$ {despesas:,.2f}
+
+**Saldo:** R$ {saldo:,.2f}
+
+**Quantidade de lançamentos:** {len(transacoes)}
+"""
+
+    return texto
+
+
+def mostrar_transacoes(transacoes):
+    if not transacoes:
+        return "Nenhuma transação cadastrada."
+
+    texto = "## 📋 Transações\n\n"
+
+    for i, t in enumerate(reversed(transacoes), 1):
+        sinal = "+" if t["tipo"] == "Receita" else "-"
+
+        texto += (
+            f"**{i}. {t['descricao']}**  \n"
+            f"📅 {t['data']} | 🏷️ {t['categoria']}  \n"
+            f"{sinal} R$ {t['valor']:,.2f}  \n\n"
+        )
+
+    return texto
+
+
+def adicionar_e_atualizar(
+    tipo,
+    descricao,
+    categoria,
+    valor,
+    data,
+    transacoes
+):
+    novas_transacoes, resumo, mensagem = adicionar_transacao(
+        tipo,
+        descricao,
+        categoria,
+        valor,
+        data,
+        transacoes
+    )
+
+    lista = mostrar_transacoes(novas_transacoes)
+
+    return (
+        novas_transacoes,
+        resumo,
+        lista,
+        mensagem,
+        "",
+        None
+    )
+
+
+def limpar_transacoes():
+    transacoes = []
+
+    return (
+        transacoes,
+        atualizar_resumo(transacoes),
+        mostrar_transacoes(transacoes),
+        "🗑️ Todas as transações foram apagadas."
+    )
+
+
+# =========================
+# INTERFACE
+# =========================
+
+with gr.Blocks(
+    title="Meu Controle Financeiro"
+) as app:
+
+    transacoes = gr.State([])
+
+    gr.Markdown(
+        """
+# 💰 Meu Controle Financeiro
+
+Controle suas **receitas, despesas e saldo** de forma simples.
+"""
+    )
+
+    with gr.Row():
+
+        with gr.Column():
+
+            gr.Markdown("## ➕ Nova transação")
+
+            tipo = gr.Radio(
+                ["Receita", "Despesa"],
+                value="Despesa",
+                label="Tipo"
+            )
+
+            descricao = gr.Textbox(
+                label="Descrição",
+                placeholder="Ex: Supermercado"
+            )
+
+            categoria = gr.Dropdown(
+                [
+                    "Alimentação",
+                    "Transporte",
+                    "Moradia",
+                    "Contas",
+                    "Lazer",
+                    "Saúde",
+                    "Educação",
+                    "Salário",
+                    "Investimentos",
+                    "Outros"
+                ],
+                value="Outros",
+                label="Categoria"
+            )
+
+            valor = gr.Number(
+                label="Valor (R$)",
+                minimum=0
+            )
+
+            data = gr.Textbox(
+                label="Data",
+                value=str(date.today())
+            )
+
+            adicionar = gr.Button(
+                "➕ Adicionar",
+                variant="primary"
+            )
+
+            mensagem = gr.Markdown()
+
+        with gr.Column():
+
+            resumo = gr.Markdown(
+                atualizar_resumo([])
+            )
+
+    gr.Markdown("---")
+
+    lista = gr.Markdown(
+        "Nenhuma transação cadastrada."
+    )
+
+    apagar = gr.Button(
+        "🗑️ Apagar todas as transações"
+    )
+
+    adicionar.click(
+        adicionar_e_atualizar,
+        inputs=[
+            tipo,
+            descricao,
+            categoria,
+            valor,
+            data,
+            transacoes
+        ],
+        outputs=[
+            transacoes,
+            resumo,
+            lista,
+            mensagem,
+            descricao,
+            valor
+        ]
+    )
+
+    apagar.click(
+        limpar_transacoes,
+        outputs=[
+            transacoes,
+            resumo,
+            lista,
+            mensagem
+        ]
+    )
+
+
+if __name__ == "__main__":
+    app.launch()