Update app.py

app.py

@@ -1,22 +1,40 @@
-from flask import Flask, render_template, request, redirect, session, url_for
+from flask import Flask, render_template, request, redirect, session, url_for, send_file
+from datetime import datetime
 from supabase_client import supabase
+from dotenv import load_dotenv
 import os
+import io
+from weasyprint import HTML
+
+load_dotenv()
 
 app = Flask(__name__)
 app.secret_key = os.urandom(24)
 
-# -------- Autenticação --------
+def calcular_total(dados):
+    area = float(dados.get('area', 0))
+    quartos = int(dados.get('quartos', 0))
+    banheiros = int(dados.get('banheiros', 0))
+    padrao = dados.get('padrao', 'simples')
+
+    fator = {'simples': 1000, 'medio': 1500, 'luxo': 2000}.get(padrao, 1000)
+
+    total = (area * fator) + (quartos * 5000) + (banheiros * 7000)
+    return total
+
+# --------- Autenticação ---------
+
 @app.route('/login', methods=['GET', 'POST'])
 def login():
     if request.method == 'POST':
         email = request.form['email']
-        password = request.form['password']
-        result = supabase.auth.sign_in_with_password({'email': email, 'password': password})
+        senha = request.form['senha']
+        result = supabase.auth.sign_in_with_password({'email': email, 'password': senha})
         if result.user:
-            session['user'] = result.user.id
+            session['user_id'] = result.user.id
             return redirect('/dashboard')
         else:
-            return "Login inválido"
+            return render_template('login.html', erro="Login inválido")
     return render_template('login.html')
 
 @app.route('/logout')
@@ -25,67 +43,107 @@ def logout():
     return redirect('/login')
 
 def login_required(f):
-    def wrap(*args, **kwargs):
-        if 'user' in session:
-            return f(*args, **kwargs)
-        else:
+    def wrapper(*args, **kwargs):
+        if 'user_id' not in session:
             return redirect('/login')
-    wrap.__name__ = f.__name__
-    return wrap
+        return f(*args, **kwargs)
+    wrapper.__name__ = f.__name__
+    return wrapper
+
+# --------- Dashboard ---------
 
-# -------- Dashboard --------
 @app.route('/dashboard')
 @login_required
 def dashboard():
-    user_id = session['user']
-    response = supabase.table('orcamentos').select('*').eq('user_id', user_id).execute()
-    orcamentos = response.data
+    user_id = session['user_id']
+    orcamentos = supabase.table('orcamentos').select('*').eq('user_id', user_id).execute().data
     return render_template('dashboard.html', orcamentos=orcamentos)
 
-# -------- Criar Orçamento --------
-@app.route('/orcamentos/novo', methods=['GET', 'POST'])
+# --------- Criar Orçamento ---------
+
+@app.route('/orcamento/novo', methods=['GET', 'POST'])
 @login_required
-def criar_orcamento():
+def novo_orcamento():
     if request.method == 'POST':
-        dados = {
-            'user_id': session['user'],
-            'cliente': request.form['cliente'],
-            'endereco': request.form['endereco'],
-            'valor_total': request.form['valor_total']
-        }
-        supabase.table('orcamentos').insert(dados).execute()
+        dados = request.form
+        total = calcular_total(dados)
+        supabase.table('orcamentos').insert({
+            'user_id': session['user_id'],
+            'cliente': dados['cliente'],
+            'endereco': dados['endereco'],
+            'area': float(dados['area']),
+            'quartos': int(dados['quartos']),
+            'banheiros': int(dados['banheiros']),
+            'padrao': dados['padrao'],
+            'total': total,
+            'data': datetime.now().isoformat()
+        }).execute()
         return redirect('/dashboard')
-    return render_template('create_orcamento.html')
+    return render_template('form_orcamento.html')
+
+# --------- Visualizar ---------
 
-# -------- Editar Orçamento --------
-@app.route('/orcamentos/<int:orcamento_id>/editar', methods=['GET', 'POST'])
+@app.route('/orcamento/<int:orc_id>')
 @login_required
-def editar_orcamento(orcamento_id):
+def visualizar_orcamento(orc_id):
+    user_id = session['user_id']
+    orc = supabase.table('orcamentos').select('*').eq('id', orc_id).eq('user_id', user_id).single().execute().data
+    if not orc:
+        return "Orçamento não encontrado", 404
+    return render_template('visualizar_orcamento.html', orc=orc)
+
+# --------- Editar ---------
+
+@app.route('/orcamento/<int:orc_id>/editar', methods=['GET', 'POST'])
+@login_required
+def editar_orcamento(orc_id):
+    user_id = session['user_id']
+    orc = supabase.table('orcamentos').select('*').eq('id', orc_id).eq('user_id', user_id).single().execute().data
+    if not orc:
+        return "Orçamento não encontrado", 404
+
     if request.method == 'POST':
-        update_data = {
-            'cliente': request.form['cliente'],
-            'endereco': request.form['endereco'],
-            'valor_total': request.form['valor_total']
-        }
-        supabase.table('orcamentos').update(update_data).eq('id', orcamento_id).execute()
+        dados = request.form
+        total = calcular_total(dados)
+        supabase.table('orcamentos').update({
+            'cliente': dados['cliente'],
+            'endereco': dados['endereco'],
+            'area': float(dados['area']),
+            'quartos': int(dados['quartos']),
+            'banheiros': int(dados['banheiros']),
+            'padrao': dados['padrao'],
+            'total': total
+        }).eq('id', orc_id).execute()
         return redirect('/dashboard')
-    orcamento = supabase.table('orcamentos').select('*').eq('id', orcamento_id).single().execute().data
-    return render_template('edit_orcamento.html', orcamento=orcamento)
 
-# -------- Ver Orçamento --------
-@app.route('/orcamentos/<int:orcamento_id>')
-@login_required
-def visualizar_orcamento(orcamento_id):
-    orcamento = supabase.table('orcamentos').select('*').eq('id', orcamento_id).single().execute().data
-    return render_template('view_orcamento.html', orcamento=orcamento)
+    return render_template('form_orcamento.html', orc=orc)
 
-# -------- Excluir Orçamento --------
-@app.route('/orcamentos/<int:orcamento_id>/excluir', methods=['POST'])
+# --------- Excluir ---------
+
+@app.route('/orcamento/<int:orc_id>/excluir')
 @login_required
-def excluir_orcamento(orcamento_id):
-    supabase.table('orcamentos').delete().eq('id', orcamento_id).execute()
+def excluir_orcamento(orc_id):
+    user_id = session['user_id']
+    supabase.table('orcamentos').delete().eq('id', orc_id).eq('user_id', user_id).execute()
     return redirect('/dashboard')
 
+# --------- Geração PDF ---------
+
+@app.route('/orcamento/<int:orc_id>/pdf')
+@login_required
+def gerar_pdf(orc_id):
+    user_id = session['user_id']
+    orc = supabase.table('orcamentos').select('*').eq('id', orc_id).eq('user_id', user_id).single().execute().data
+    if not orc:
+        return "Orçamento não encontrado", 404
+
+    html = render_template('contrato_pdf.html', orc=orc)
+    pdf = HTML(string=html).write_pdf()
+    return send_file(io.BytesIO(pdf), download_name="contrato.pdf", as_attachment=True)
+
+# --------- Run Hugging Face ---------
+
 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)))
+