Spaces:

Persano
/

reformape

Sleeping

App Files Files Community

Persano commited on Apr 22

Commit

b3108df

verified ·

1 Parent(s): 0f7e3cc

Update app.py

Browse files

Files changed (1) hide show

app.py +97 -209

app.py CHANGED Viewed

@@ -1,226 +1,114 @@
 from flask import Flask, render_template_string, request, send_file
 from fpdf import FPDF
 import os
 import tempfile
 from datetime import date, timedelta
-import gradio as gr
 app = Flask(__name__)
-# HTML para o formulário
-HTML_FORM = """
-<!DOCTYPE html>
-<html lang="pt-br">
-<head>
-    <meta charset="UTF-8">
-    <title>Orçamento de Reforma</title>
-    <style>
-        body {
-            font-family: Arial, sans-serif;
-            margin: 40px;
-            background-color: #f7f7f7;
-        }
-        h1 {
-            color: #333;
-            text-align: center;
-        }
-        form {
-            background-color: #fff;
-            padding: 20px;
-            border-radius: 10px;
-            box-shadow: 0 0 10px rgba(0,0,0,0.1);
-            max-width: 700px;
-            margin: auto;
-        }
-        label {
-            font-weight: bold;
-        }
-        input, select {
-            margin-bottom: 15px;
-            width: 100%;
-            padding: 8px;
-            box-sizing: border-box;
-        }
-        fieldset {
-            margin-bottom: 20px;
-            padding: 10px;
-            border-radius: 5px;
-            border: 1px solid #ccc;
-        }
-        legend {
-            font-weight: bold;
-        }
-        input[type=submit] {
-            background-color: #4CAF50;
-            color: white;
-            border: none;
-            cursor: pointer;
-        }
-        input[type=submit]:hover {
-            background-color: #45a049;
-        }
-        .checkboxes {
-            display: flex;
-            flex-wrap: wrap;
-            gap: 10px;
-        }
-        .checkboxes label {
-            display: flex;
-            align-items: center;
-            gap: 5px;
-        }
-        @media (max-width: 768px) {
-            form {
-                padding: 15px;
-            }
-        }
-    </style>
-    <script>
-        function gerarCamposBanheiros() {
-            const container = document.getElementById("banheiros_por_tamanho");
-            container.innerHTML = "";
-            const total = parseInt(document.getElementsByName("banheiros")[0].value);
-            for (let i = 1; i <= total; i++) {
-                container.innerHTML += `
-                    <fieldset>
-                        <legend>Banheiro ${i}</legend>
-                        <label>Tamanho:</label>
-                        <select name="banheiro_${i}_tamanho">
-                            <option value="pequeno">Pequeno</option>
-                            <option value="medio">Médio</option>
-                            <option value="grande">Grande</option>
-                        </select><br>
-                        <label>Acabamento:</label>
-                        <select name="banheiro_${i}_acabamento">
-                            <option value="simples">Simples</option>
-                            <option value="medio">Médio</option>
-                            <option value="luxo">Luxo</option>
-                        </select><br><br>
-                    </fieldset>`;
-            }
-        }
-    </script>
-</head>
-<body>
-    <h1>Orçamento de Reforma</h1>
-    <form method="post">
-        <label>Nome do contratante:</label>
-        <input type="text" name="nome_contratante" required><br>
-        <label>Endereço da obra:</label>
-        <input type="text" name="endereco_obra" required><br>
-        <label>Dimensão total da obra (m²):</label>
-        <input type="number" name="dimensao_obra" min="1" required><br>
-        <label>Quantidade total de banheiros:</label>
-        <input type="number" name="banheiros" min="0" required onchange="gerarCamposBanheiros()"><br>
-        <div id="banheiros_por_tamanho"></div>
-        <label>Tamanho da cozinha:</label>
-        <select name="cozinha_tamanho">
-            <option value="pequeno">Pequeno</option>
-            <option value="medio">Médio</option>
-            <option value="grande">Grande</option>
-        </select><br>
-        <label>Acabamento da cozinha:</label>
-        <select name="cozinha_acabamento">
-            <option value="simples">Simples</option>
-            <option value="medio">Médio</option>
-            <option value="luxo">Luxo</option>
-        </select><br>
-        <label>Serviços desejados:</label>
-        <div class="checkboxes">
-            <label><input type="checkbox" name="servicos" value="hidraulica"> Hidráulica</label>
-            <label><input type="checkbox" name="servicos" value="eletrica"> Elétrica</label>
-            <label><input type="checkbox" name="servicos" value="pintura"> Pintura</label>
-            <label><input type="checkbox" name="servicos" value="ar_condicionado"> Ar Condicionado</label>
-            <label><input type="checkbox" name="servicos" value="fechamento_sacada"> Fechamento de Sacada</label>
-            <label><input type="checkbox" name="servicos" value="gesso"> Gesso</label>
-        </div><br>
-        <input type="submit" name="acao" value="Calcular Orçamento">
-    </form>
-    {% if total is defined and detalhes is defined %}
-        <div style="background-color:#fff; margin-top:20px; padding:20px; border-radius:10px; box-shadow:0 0 10px rgba(0,0,0,0.1); max-width:700px; margin:auto;">
-            <h2>Resumo do Orçamento</h2>
-            <ul>
-            {% for item in detalhes %}
-                <li>{{ item }}</li>
-            {% endfor %}
-            </ul>
-            {% if total is not none %}
-                <p><strong>Valor total:</strong> R$ {{ "%.2f"|format(total) }}</p>
-            {% else %}
-                <p><strong>Valor total:</strong> Não calculado</p>
-            {% endif %}
-            <form method="post">
-                {% for key, value in request.form.items() %}
-                    <input type="hidden" name="{{ key }}" value="{{ value }}">
-                {% endfor %}
-                <input type="submit" name="acao" value="Gerar Contrato PDF">
-            </form>
-        </div>
-    {% endif %}
-</body>
-</html>
-"""
 @app.route("/", methods=["GET", "POST"])
 def index():
-    total = None
-    detalhes = []
-    erro = None
     if request.method == "POST":
-        try:
-            nome_contratante = request.form["nome_contratante"]
-            endereco_obra = request.form["endereco_obra"]
-            dimensao_obra = float(request.form["dimensao_obra"])
-            banheiros = int(request.form["banheiros"])
-            cozinha_tamanho = request.form["cozinha_tamanho"]
-            cozinha_acabamento = request.form["cozinha_acabamento"]
-            servicos = request.form.getlist("servicos")
-            # Calculando os valores do orçamento
-            total = 10000 + dimensao_obra * 50  # Simples cálculo de exemplo
-            detalhes = [
-                f"Nome: {nome_contratante}",
-                f"Endereço: {endereco_obra}",
-                f"Dimensão da obra: {dimensao_obra} m²",
-                f"Quantidade de banheiros: {banheiros}",
-                f"Tamanho da cozinha: {cozinha_tamanho}",
-                f"Acabamento da cozinha: {cozinha_acabamento}",
-                f"Serviços: {', '.join(servicos)}",
-            ]
-            if "acao" in request.form and request.form["acao"] == "Gerar Contrato PDF":
-                # Gerar o contrato em PDF
-                pdf = FPDF()
-                pdf.add_page()
-                pdf.set_font("Arial", size=12)
-                pdf.cell(200, 10, txt=f"Orçamento de Reforma - {nome_contratante}", ln=True, align="C")
-                for item in detalhes:
-                    pdf.cell(200, 10, txt=item, ln=True)
-                # Salvar o PDF temporariamente
-                temp_file = tempfile.NamedTemporaryFile(delete=False, suffix=".pdf")
-                pdf.output(temp_file.name)
-                return send_file(temp_file.name, as_attachment=True)
-        except Exception as e:
-            erro = str(e)
-    return render_template_string(HTML_FORM, erro=erro, total=total, detalhes=detalhes)
 if __name__ == "__main__":
-    app.run(debug=True, host="0.0.0.0", port=7860)

+# app.py
 from flask import Flask, render_template_string, request, send_file
 from fpdf import FPDF
 import os
 import tempfile
 from datetime import date, timedelta
 app = Flask(__name__)
+# HTML_FORM vem aqui (omitido neste trecho por brevidade)
+HTML_FORM = """(HTML completo da pergunta anterior)"""
+# Funções auxiliares
+def calcular_valor_por_m2(tamanho, acabamento):
+    tabela = {
+        "simples": 1000,
+        "medio": 1500,
+        "luxo": 2000
+    }
+    fator_tamanho = {
+        "pequeno": 1,
+        "medio": 1.5,
+        "grande": 2
+    }
+    return tabela[acabamento] * fator_tamanho[tamanho]
+def calcular_orcamento(form):
+    detalhes = []
+    total = 0
+    metros_totais = float(form.get("dimensao_obra", 0))
+    # Cozinha
+    cozinha_tamanho = form.get("cozinha_tamanho")
+    cozinha_acabamento = form.get("cozinha_acabamento")
+    preco_cozinha = calcular_valor_por_m2(cozinha_tamanho, cozinha_acabamento)
+    valor_cozinha = preco_cozinha * 5  # Cozinha = 5m² fixo
+    detalhes.append(f"Cozinha ({cozinha_tamanho}, {cozinha_acabamento}): R$ {valor_cozinha:.2f}")
+    total += valor_cozinha
+    # Banheiros
+    num_banheiros = int(form.get("banheiros", 0))
+    for i in range(1, num_banheiros + 1):
+        tam = form.get(f"banheiro_{i}_tamanho")
+        acb = form.get(f"banheiro_{i}_acabamento")
+        preco = calcular_valor_por_m2(tam, acb)
+        m2 = 3 if tam == "pequeno" else 5 if tam == "medio" else 7
+        valor = preco * m2
+        detalhes.append(f"Banheiro {i} ({tam}, {acb}): R$ {valor:.2f}")
+        total += valor
+    # Serviços adicionais (hidráulica, elétrica, pintura = 10% cada)
+    servicos = request.form.getlist("servicos")
+    for s in ["hidraulica", "eletrica", "pintura"]:
+        if s in servicos:
+            valor = total * 0.1
+            total += valor
+            detalhes.append(f"{s.capitalize()}: R$ {valor:.2f}")
+    # Outros serviços fixos
+    servicos_fixos = ["ar_condicionado", "fechamento_sacada", "gesso"]
+    precos_fixos = {
+        "ar_condicionado": 1500,
+        "fechamento_sacada": 3000,
+        "gesso": 1000
+    }
+    for s in servicos_fixos:
+        if s in servicos:
+            valor = precos_fixos[s]
+            total += valor
+            detalhes.append(f"{s.replace('_', ' ').capitalize()}: R$ {valor:.2f}")
+    return total, detalhes
+def gerar_pdf(form, total, detalhes):
+    pdf = FPDF()
+    pdf.add_page()
+    pdf.set_font("Arial", size=12)
+    pdf.cell(200, 10, txt="Contrato de Reforma", ln=True, align="C")
+    pdf.cell(200, 10, txt=f"Contratante: {form.get('nome_contratante')}", ln=True)
+    pdf.cell(200, 10, txt=f"Endereço da Obra: {form.get('endereco_obra')}", ln=True)
+    pdf.cell(200, 10, txt=f"Data: {date.today().strftime('%d/%m/%Y')}", ln=True)
+    pdf.cell(200, 10, txt="Serviços contratados:", ln=True)
+    for item in detalhes:
+        pdf.multi_cell(0, 10, txt=f"- {item}")
+    pdf.cell(200, 10, txt=f"Valor Total: R$ {total:.2f}", ln=True)
+    pdf.cell(200, 10, txt="Assinatura do Contratante: __________________________", ln=True)
+    tmp_dir = tempfile.mkdtemp()
+    caminho_pdf = os.path.join(tmp_dir, "contrato.pdf")
+    pdf.output(caminho_pdf)
+    return caminho_pdf
+# Rotas
 @app.route("/", methods=["GET", "POST"])
 def index():
     if request.method == "POST":
+        if request.form.get("acao") == "Calcular Orçamento":
+            total, detalhes = calcular_orcamento(request.form)
+            return render_template_string(HTML_FORM, total=total, detalhes=detalhes, request=request)
+        elif request.form.get("acao") == "Gerar Contrato PDF":
+            total, detalhes = calcular_orcamento(request.form)
+            caminho_pdf = gerar_pdf(request.form, total, detalhes)
+            return send_file(caminho_pdf, as_attachment=True)
+    return render_template_string(HTML_FORM)
 if __name__ == "__main__":
+    app.run(host="0.0.0.0", port=7860)