Update app.py

app.py

@@ -1,221 +1,213 @@
-from flask import Flask, render_template_string, request, send_file, redirect, url_for
-import matplotlib.pyplot as plt
-import matplotlib
-from matplotlib.ticker import MaxNLocator
-import os
+from flask import Flask, request, render_template_string
 from datetime import datetime, timedelta
-from fpdf import FPDF
-
-# Ajustes para Hugging Face (sem permissão de escrita em /.config)
-matplotlib.use('Agg')
+import os
 
 app = Flask(__name__)
 
-# Estimativas de tempo de obra (em dias)
-TEMPOS_OBRA = {
-    "Demolição": 3,
-    "Hidráulica": 5,
-    "Elétrica": 5,
-    "Gesso": 4,
-    "Pintura": 4,
-    "Piso": 5,
-    "Ar Condicionado": 3,
-    "Nivelamento de Sacada": 2,
-    "Limpeza Final": 2
-}
-
-@app.route("/", methods=["GET", "POST"])
-def index():
-    if request.method == "POST":
-        # Dados do cliente e obra
-        cliente = request.form["cliente"]
-        endereco = request.form["endereco"]
-        data_inicio = request.form["data_inicio"]
-        metragem = float(request.form["metragem"])
-        quartos = int(request.form["quartos"])
-        banheiros = int(request.form["banheiros"])
-        tamanhos_banheiros = [request.form[f"tamanho_banheiro_{i}"] for i in range(banheiros)]
-        padroes_banheiros = [request.form[f"padrao_banheiro_{i}"] for i in range(banheiros)]
-        tamanho_cozinha = request.form["tamanho_cozinha"]
-        padrao_cozinha = request.form["padrao_cozinha"]
-        padrao_geral = request.form["padrao_geral"]
-        servicos = request.form.getlist("servicos")
-
-        # Lógica de cálculo
-        total = metragem * (1000 if padrao_geral == "econômico" else 1500 if padrao_geral == "médio" else 2000)
-        total += sum(500 if t == "pequeno" else 1000 if t == "médio" else 1500 for t in tamanhos_banheiros)
-        total += 500 if tamanho_cozinha == "pequeno" else 1000 if tamanho_cozinha == "médio" else 1500
-
-        # Adiciona valor de acordo com padrão de cozinha
-        total += 1000 if padrao_cozinha == "luxo" else 500 if padrao_cozinha == "médio" else 0
-
-        for s in servicos:
-            total += 1000  # valor base por serviço
-
-        # Cronograma
-        cronograma = []
-        data_atual = datetime.strptime(data_inicio, "%Y-%m-%d")
-        for etapa in ["Demolição"] + servicos + ["Limpeza Final"]:
-            duracao = TEMPOS_OBRA.get(etapa, 3)
-            cronograma.append((etapa, data_atual.strftime("%d/%m"), (data_atual + timedelta(days=duracao)).strftime("%d/%m")))
-            data_atual += timedelta(days=duracao)
-
-        # Gerar contrato PDF
-        contrato_path = gerar_contrato(cliente, endereco, data_inicio, total, cronograma)
-
-        # Gerar gráfico
-        gerar_grafico_gantt(cronograma)
-
-        return render_template_string(SUCESSO_HTML, cliente=cliente, total=total, contrato=contrato_path)
-
-    return render_template_string(FORM_HTML)
-
-@app.route("/grafico")
-def grafico():
-    return send_file("grafico.png", mimetype="image/png")
-
-@app.route("/contrato")
-def contrato():
-    return send_file("contrato.pdf", as_attachment=True)
-
-def gerar_contrato(cliente, endereco, data_inicio, total, cronograma):
-    pdf = FPDF()
-    pdf.add_page()
-    pdf.set_font("Arial", size=12)
-
-    pdf.cell(200, 10, txt="CONTRATO DE PRESTAÇÃO DE SERVIÇOS DE REFORMA", ln=True, align='C')
-    pdf.ln(10)
-    pdf.multi_cell(0, 10, f"""
-Pelo presente instrumento, de um lado {cliente}, residente no endereço {endereco},
-contrata os serviços de reforma, com início previsto em {data_inicio}.
-
-O valor total do contrato é de R$ {total:,.2f}, dividido em parcelas semanais conforme cronograma abaixo:
-""")
-
-    for etapa, inicio, fim in cronograma:
-        pdf.cell(0, 10, f"{etapa}: {inicio} a {fim}", ln=True)
-
-    pdf.ln(10)
-    pdf.multi_cell(0, 10, """
-Cláusulas adicionais:
-1. A contratada se compromete a seguir o cronograma acordado.
-2. Qualquer atraso ou alteração deverá ser comunicada com 48h de antecedência.
-3. Os materiais estão inclusos no orçamento total (salvo ajuste prévio).
-4. Pagamentos semanais deverão ser realizados na data de início de cada etapa.
-
-Assinaturas:
-______________________________________
-Cliente
-
-______________________________________
-Prestador de Serviços
-""")
-    pdf.output("contrato.pdf")
-    return "contrato.pdf"
-
-def gerar_grafico_gantt(cronograma):
-    fig, ax = plt.subplots(figsize=(10, 6))
-    for i, (etapa, inicio, fim) in enumerate(cronograma):
-        start = datetime.strptime(inicio, "%d/%m")
-        end = datetime.strptime(fim, "%d/%m")
-        ax.barh(i, (end - start).days, left=start.toordinal(), height=0.6, align='center')
-        ax.text(start.toordinal(), i, f"{etapa}", va='center', ha='left', fontsize=10)
-    ax.set_yticks(range(len(cronograma)))
-    ax.set_yticklabels([e[0] for e in cronograma])
-    ax.xaxis.set_major_locator(MaxNLocator(integer=True))
-    ax.set_title("Cronograma da Reforma (Gantt)")
-    fig.tight_layout()
-    plt.savefig("grafico.png")
-    plt.close()
-
-FORM_HTML = """
+# --- Página Inicial ---
+@app.route("/", methods=["GET"])
+def home():
+    return render_template_string(HTML_FORM)
+
+# --- Processamento da Obra ---
+@app.route("/salvar_obra", methods=["POST"])
+def salvar_obra():
+    dados = request.form
+
+    cliente = dados["cliente"]
+    endereco = dados["endereco"]
+    metros = int(dados["metros"])
+    banheiros = int(dados["banheiros"])
+    tamanho_banheiro = dados["tamanho_banheiro"]
+    cozinha_m2 = int(dados["cozinha_m2"])
+    acabamento = dados["acabamento"]
+    quartos = int(dados["quartos"])
+    inicio = datetime.strptime(dados["inicio"], "%Y-%m-%d")
+
+    fator_acabamento = {"simples": 1.0, "medio": 1.2, "luxo": 1.5}[acabamento]
+    fator_banheiro = {"pequeno": 3000, "medio": 4000, "grande": 6000}[tamanho_banheiro]
+
+    itens = [
+        {"nome": "Hidráulica", "dias": 2, "base": 30 * metros},
+        {"nome": "Elétrica", "dias": 2, "base": 25 * metros},
+        {"nome": "Gesso", "dias": 2, "base": 20 * metros},
+        {"nome": "Pintura", "dias": 2, "base": 18 * metros},
+        {"nome": "Ar Condicionado", "dias": 1, "base": 3000},
+        {"nome": "Nivelamento de Sacada", "dias": 1, "base": 2400},
+        {"nome": "Piso de Madeira", "dias": 3, "base": 60 * metros},
+        {"nome": "Banheiros", "dias": 3, "base": fator_banheiro * banheiros},
+        {"nome": "Cozinha", "dias": 3, "base": 5000 + (cozinha_m2 * 200)},
+    ]
+
+    itens_formatados = []
+    data_atual = inicio
+    for item in itens:
+        custo = round(item["base"] * fator_acabamento)
+        fim = data_atual + timedelta(days=item["dias"])
+        itens_formatados.append({
+            "nome": item["nome"],
+            "inicio": data_atual.strftime("%d/%m/%Y"),
+            "fim": fim.strftime("%d/%m/%Y"),
+            "custo": custo
+        })
+        data_atual = fim + timedelta(days=1)
+
+    total = sum(i["custo"] for i in itens_formatados)
+    pagamentos = []
+    for i in range(4):
+        data_pag = datetime.today() + timedelta(days=i * 15)
+        pagamentos.append({
+            "data": data_pag.strftime("%d/%m/%Y"),
+            "valor": round(total / 4, 2)
+        })
+
+    return render_template_string(HTML_RELATORIO,
+        cliente=cliente,
+        endereco=endereco,
+        inicio=inicio.strftime("%d/%m/%Y"),
+        itens=itens_formatados,
+        total=total,
+        pagamentos=pagamentos
+    )
+
+# --- HTMLs Embutidos ---
+
+HTML_FORM = """
 <!DOCTYPE html>
-<html>
+<html lang="pt-br">
 <head>
+    <meta charset="UTF-8">
     <title>Simulador de Reforma</title>
     <style>
-        body { font-family: sans-serif; max-width: 800px; margin: auto; padding: 20px; }
-        input, select { width: 100%; margin-bottom: 10px; padding: 8px; }
+        body {
+            font-family: 'Segoe UI', sans-serif;
+            background-color: #f0f4f8;
+            color: #333;
+            padding: 40px;
+        }
+        .container {
+            max-width: 700px;
+            margin: auto;
+            background: white;
+            padding: 30px;
+            border-radius: 15px;
+            box-shadow: 0 4px 20px rgba(0, 0, 0, 0.1);
+        }
+        h1 {
+            color: #004080;
+        }
+        label {
+            display: block;
+            margin: 15px 0 5px;
+        }
+        input, select {
+            width: 100%;
+            padding: 8px;
+            margin-bottom: 10px;
+            border-radius: 6px;
+            border: 1px solid #ccc;
+        }
+        button {
+            background-color: #0070cc;
+            color: white;
+            padding: 10px 20px;
+            border: none;
+            border-radius: 8px;
+            font-size: 16px;
+            cursor: pointer;
+        }
+        button:hover {
+            background-color: #005bb5;
+        }
     </style>
 </head>
 <body>
-    <h1>Simulador de Reforma</h1>
-    <form method="post">
-        <h3>Dados do Cliente</h3>
-        <input name="cliente" placeholder="Nome completo" required>
-        <input name="endereco" placeholder="Endereço da obra" required>
-        <label>Data de Início:</label>
-        <input type="date" name="data_inicio" required>
-
-        <h3>Informações do Imóvel</h3>
-        <input type="number" name="metragem" placeholder="Metragem total (m²)" required>
-        <input type="number" name="quartos" placeholder="Número de quartos" required>
-        <input type="number" name="banheiros" placeholder="Quantidade de banheiros" min="1" max="5" required>
-        {% for i in range(5) %}
-            <div>
-                <label>Banheiro {{ i+1 }} - Tamanho:</label>
-                <select name="tamanho_banheiro_{{ i }}">
+    <div class="container">
+        <h1>Simulador de Reforma</h1>
+        <form action="/salvar_obra" method="post">
+            <label>Nome do Cliente: <input type="text" name="cliente" required></label>
+            <label>Endereço da Obra: <input type="text" name="endereco" required></label>
+            <label>M² da Obra: <input type="number" name="metros" required></label>
+            <label>Quantidade de Quartos: <input type="number" name="quartos" required></label>
+            <label>Número de Banheiros: <input type="number" name="banheiros" required></label>
+            <label>Tamanho dos Banheiros:
+                <select name="tamanho_banheiro">
                     <option value="pequeno">Pequeno</option>
-                    <option value="médio">Médio</option>
+                    <option value="medio">Médio</option>
                     <option value="grande">Grande</option>
                 </select>
-                <label>Banheiro {{ i+1 }} - Padrão:</label>
-                <select name="padrao_banheiro_{{ i }}">
-                    <option value="econômico">Econômico</option>
-                    <option value="médio">Médio</option>
+            </label>
+            <label>Tamanho da Cozinha (m²): <input type="number" name="cozinha_m2" required></label>
+            <label>Padrão de Acabamento:
+                <select name="acabamento">
+                    <option value="simples">Simples</option>
+                    <option value="medio">Médio</option>
                     <option value="luxo">Luxo</option>
                 </select>
-            </div>
-        {% endfor %}
-        <label>Tamanho da Cozinha:</label>
-        <select name="tamanho_cozinha">
-            <option value="pequeno">Pequeno</option>
-            <option value="médio">Médio</option>
-            <option value="grande">Grande</option>
-        </select>
-        <label>Padrão da Cozinha:</label>
-        <select name="padrao_cozinha">
-            <option value="econômico">Econômico</option>
-            <option value="médio">Médio</option>
-            <option value="luxo">Luxo</option>
-        </select>
-        <label>Padrão Construtivo do Imóvel:</label>
-        <select name="padrao_geral">
-            <option value="econômico">Econômico</option>
-            <option value="médio">Médio</option>
-            <option value="alto padrão">Alto Padrão</option>
-        </select>
-
-        <h3>Serviços Desejados:</h3>
-        <label><input type="checkbox" name="servicos" value="Hidráulica"> Hidráulica</label><br>
-        <label><input type="checkbox" name="servicos" value="Elétrica"> Elétrica</label><br>
-        <label><input type="checkbox" name="servicos" value="Gesso"> Gesso</label><br>
-        <label><input type="checkbox" name="servicos" value="Pintura"> Pintura</label><br>
-        <label><input type="checkbox" name="servicos" value="Piso"> Piso</label><br>
-        <label><input type="checkbox" name="servicos" value="Ar Condicionado"> Ar Condicionado</label><br>
-        <label><input type="checkbox" name="servicos" value="Nivelamento de Sacada"> Nivelamento de Sacada</label><br>
-
-        <button type="submit">Gerar Contrato e Cronograma</button>
-    </form>
+            </label>
+            <label>Início da Obra: <input type="date" name="inicio" required></label>
+            <button type="submit">Gerar Orçamento</button>
+        </form>
+    </div>
 </body>
 </html>
 """
 
-SUCESSO_HTML = """
+HTML_RELATORIO = """
 <!DOCTYPE html>
-<html>
-<head><title>Resultado</title></head>
+<html lang="pt-br">
+<head>
+    <meta charset="UTF-8">
+    <title>Relatório da Obra</title>
+    <style>
+        body {
+            font-family: 'Segoe UI', sans-serif;
+            background-color: #f8f9fa;
+            padding: 40px;
+        }
+        .container {
+            max-width: 800px;
+            margin: auto;
+            background: white;
+            padding: 30px;
+            border-radius: 15px;
+            box-shadow: 0 4px 20px rgba(0,0,0,0.1);
+        }
+        h1, h2 {
+            color: #004080;
+        }
+        .item {
+            padding: 6px 0;
+            border-bottom: 1px solid #ddd;
+        }
+        .total {
+            font-size: 1.2em;
+            color: #006600;
+            margin-top: 20px;
+            font-weight: bold;
+        }
+    </style>
+</head>
 <body>
-    <h1>Simulação Finalizada</h1>
-    <p>Cliente: {{ cliente }}</p>
-    <p>Valor total estimado: R$ {{ '{:,.2f}'.format(total).replace(',', 'X').replace('.', ',').replace('X', '.') }}</p>
-    <a href="/contrato">📄 Baixar Contrato em PDF</a><br><br>
-    <a href="/grafico">📊 Ver Gráfico do Cronograma</a>
+    <div class="container">
+        <h1>Relatório da Obra</h1>
+        <p><strong>Cliente:</strong> {{cliente}}</p>
+        <p><strong>Endereço:</strong> {{endereco}}</p>
+        <p><strong>Início da Obra:</strong> {{inicio}}</p>
+        <h2>Itens e Custos:</h2>
+        {% for i in itens %}
+            <div class="item">{{i.nome}} ({{i.inicio}} - {{i.fim}}) → R$ {{i.custo}}</div>
+        {% endfor %}
+        <p class="total">Total: R$ {{total}}</p>
+        <h2>Pagamentos:</h2>
+        {% for p in pagamentos %}
+            <div class="item">{{p.data}} → R$ {{p.valor}}</div>
+        {% endfor %}
+    </div>
 </body>
 </html>
 """
 
+# --- Executa no Hugging Face Spaces ---
 if __name__ == "__main__":
-    app.run(host="0.0.0.0", port=7860)
-
-
+    app.run(host="0.0.0.0", port=int(os.environ.get("PORT", 7860)))