Update app.py

app.py

@@ -6,16 +6,15 @@ import matplotlib.patches as patches
 import io
 import csv
 import os
-from reportlab.platypus import SimpleDocTemplate, Paragraph, Spacer, Image as RLImage, Table, TableStyle, PageBreak, KeepTogether
-from reportlab.lib.styles import getSampleStyleSheet, ParagraphStyle
+from reportlab.platypus import SimpleDocTemplate, Paragraph, Spacer, Image as RLImage, Table, TableStyle, PageBreak
+from reportlab.lib.styles import getSampleStyleSheet
 from reportlab.lib import colors
 from reportlab.lib.pagesizes import letter
-from reportlab.lib.units import mm
 
 # ==========================================
 # CONFIGURAÇÃO GERAL
 # ==========================================
-st.set_page_config(page_title="SGP - Gestão de Embalagem v14.0", layout="wide", page_icon="🏭")
+st.set_page_config(page_title="SGP - Gestão de Embalagem v14.1", layout="wide", page_icon="🏭")
 
 PESO_IDEAL_MIN = 970
 PESO_IDEAL_MAX = 1080
@@ -121,10 +120,17 @@ class PackageCalculator:
             if catalog_item:
                 origem = "Catálogo (Norma)"
                 num_pecas = catalog_item['qtde']
+                
+                # Verifica se é Grade Reta (IT13/14 com ml e ma definidos e coerentes)
+                is_grid_rect = False
+                if catalog_item.get('ml') and catalog_item.get('ma'):
+                    if catalog_item['ml'] * catalog_item['ma'] == num_pecas:
+                         is_grid_rect = True
+
                 if catalog_item.get('sextavado') and 'padrao' in catalog_item:
                     layout = self._parse_layout(catalog_item['padrao'])
                     is_sext, gl, ga = True, 0, len(layout)
-                elif tipo == "Redondo":
+                elif tipo == "Redondo" and not is_grid_rect:
                     # Conversão Grade->Ninho
                     bw = catalog_item.get('ml', 0) or int(math.sqrt(num_pecas))
                     layout, rem, cw = [], num_pecas, bw
@@ -134,6 +140,7 @@ class PackageCalculator:
                         cw = bw-1 if cw==bw else bw
                     is_sext, gl, ga = True, 0, len(layout)
                 else:
+                    # Grade Reta (Metalon ou Redondo "quadrado")
                     is_sext, gl, ga = False, catalog_item.get('ml',0), catalog_item.get('ma',0)
             else:
                 origem = "Engenharia (Cálculo)"
@@ -161,7 +168,7 @@ class PackageCalculator:
 
         # Montagem Pacotes
         main = self._build_pkg(l, a, e, c, tipo, aba, num_pecas, peso_pct, origem, is_sext, layout, gl, ga, False, is_galv, wood_h)
-        main['catalog_data'] = catalog_item # Salva dados do catalogo para o PDF
+        main['catalog_data'] = catalog_item
         main['peso_metro'] = peso_metro
         main['area_secao'] = area
         
@@ -197,13 +204,19 @@ class PackageCalculator:
             info["layout_str"] = "-".join(map(str, r)) if r else ""
         else:
             info["larg_total"] = gl*l + (aba if tipo=="Enrijecido" else 0)
+            
+            # Altura Base
+            if tipo == "Redondo": 
+                h_base = ga * l
+            else: 
+                h_base = ga * a
+                
             if is_galv:
                 cm = max(0, ga-1)
-                hb = (l + (ga-1)*l*math.sin(math.radians(60))) if tipo=="Redondo" else ga*a
-                info["alt_total"] = hb + cm*wood_h
+                info["alt_total"] = h_base + cm*wood_h
                 info["qtd_madeiras"] = cm * 5
             else:
-                info["alt_total"] = (l + (ga-1)*l*math.sin(math.radians(60))) if tipo=="Redondo" else ga*a
+                info["alt_total"] = h_base
                 info["qtd_madeiras"] = 0
             info["layout_str"] = ""
 
@@ -224,7 +237,6 @@ class PackageCalculator:
 # ==========================================
 
 def plot_single_profile(info):
-    """Desenha um único perfil com cotas para o PDF"""
     fig, ax = plt.subplots(figsize=(4, 4))
     l, a, e = info['l'], info['a'], info['e']
     tipo = info['tipo']
@@ -232,25 +244,18 @@ def plot_single_profile(info):
     if tipo == "Redondo":
         ax.add_patch(patches.Circle((l/2, l/2), l/2, fc='lightgray', ec='black', lw=2))
         ax.add_patch(patches.Circle((l/2, l/2), l/2-e, fc='white', ec='black', lw=1))
-        # Cota Diametro
         ax.annotate(f"Ø {l:.2f}", xy=(l/2, -l*0.1), xytext=(l/2, -l*0.2), ha='center', arrowprops=dict(arrowstyle='<->'))
-        # Cota Espessura
         ax.annotate(f"e={e:.2f}", xy=(l/2, l-e), xytext=(l*1.2, l), arrowprops=dict(arrowstyle='->'))
         ax.set_xlim(-l*0.2, l*1.2); ax.set_ylim(-l*0.2, l*1.2)
-    
-    else: # Quadrado/Ret
+    else: 
         ax.add_patch(patches.Rectangle((0,0), l, a, fc='lightgray', ec='black', lw=2))
         ax.add_patch(patches.Rectangle((e,e), l-2*e, a-2*e, fc='white', ec='black', lw=1))
-        # Cota L
         ax.annotate(f"L={l:.2f}", xy=(l/2, -a*0.1), xytext=(l/2, -a*0.2), ha='center', arrowprops=dict(arrowstyle='<->'))
-        # Cota A
         ax.annotate(f"A={a:.2f}", xy=(-l*0.1, a/2), xytext=(-l*0.2, a/2), va='center', rotation=90, arrowprops=dict(arrowstyle='<->'))
-        # Cota E
         ax.annotate(f"e={e:.2f}", xy=(l-e/2, a/2), xytext=(l+l*0.2, a/2), arrowprops=dict(arrowstyle='->'))
         ax.set_xlim(-l*0.3, l*1.3); ax.set_ylim(-a*0.3, a*1.3)
 
-    ax.set_aspect('equal')
-    ax.axis('off')
+    ax.set_aspect('equal'); ax.axis('off')
     plt.title(f"Perfil Individual ({tipo})", fontsize=10)
     return fig
 
@@ -283,24 +288,29 @@ def plot_cross_section(info):
         drawn = 0
         rad = l/2; dy = l*0.866
         for i in range(ga):
-            if info['tipo']=="Redondo": y = rad + i*dy + i*wood_h
-            else: y = i*(a+wood_h)
+            if info['tipo']=="Redondo": 
+                # Grade Redonda (Alinhada)
+                y = i*(l+wood_h)
+                cy = y + l/2
+            else: 
+                y = i*(a+wood_h)
+                tube_y = y
             
             if is_galv and i>0:
-                wy = y-rad-wood_h if info['tipo']=="Redondo" else y-wood_h
+                wy = y-wood_h
                 wx = (total_w - wood_len)/2
                 ax.add_patch(patches.Rectangle((wx, wy), wood_len, wood_h, fc='peru', ec='brown'))
-                tube_y = y+wood_h if info['tipo']!="Redondo" else y
-            else: tube_y = y
             
             for j in range(gl):
                 if drawn >= info['num']: break
+                x = j*l
+                
                 if info['tipo']=="Redondo":
-                    x = j*l if i%2==0 else j*l+rad
-                    ax.add_patch(patches.Circle((x+rad, y), rad, fc=fc, ec=ec))
-                    ax.add_patch(patches.Circle((x+rad, y), rad-e, fc='white', ec=ec, lw=0.5))
+                    # Grade Redonda
+                    cx = x + rad
+                    ax.add_patch(patches.Circle((cx, cy), rad, fc=fc, ec=ec))
+                    ax.add_patch(patches.Circle((cx, cy), rad-e, fc='white', ec=ec, lw=0.5))
                 else:
-                    x = j*l
                     ax.add_patch(patches.Rectangle((x, tube_y), l, a, fc=fc, ec=ec))
                     ax.add_patch(patches.Rectangle((x+e, tube_y+e), l-2*e, a-2*e, fc='white', ec=ec, lw=0.5))
                 drawn+=1
@@ -309,7 +319,6 @@ def plot_cross_section(info):
     margin = max(total_w, info['alt_total']) * 0.15
     ax.add_patch(patches.Rectangle((-2, -2), total_w+4, info['alt_total']+4, fill=False, ec='red', ls='--', lw=1))
     
-    # Texto Grande
     if not info['sext']:
         ax.text(total_w/2, info['alt_total']+margin*0.2, f"BASE: {info['gl']}", ha='center', color='blue', weight='bold', fontsize=12)
         ax.text(-margin*0.2, info['alt_total']/2, f"ALT: {info['ga']}", ha='right', va='center', color='blue', weight='bold', rotation=90, fontsize=12)
@@ -330,8 +339,14 @@ def plot_side_view(info):
     if info['is_galv']:
         lays = info['ga'] if not info['sext'] else len(info['layout'])
         for i in range(1, lays):
-            y = i*(info['l']*0.866 + info['wood_h']) if info['sext'] else i*(info['a']+info['wood_h'])
-            y -= info['wood_h']
+            # Cálculo Y da madeira na vista lateral
+            if info['sext']:
+                 y = i*(info['l']*0.866 + info['wood_h']) - info['wood_h'] # Aprox
+            else:
+                 y = i*(info['a'] + info['wood_h']) - info['wood_h']
+                 if info['tipo'] == "Redondo":
+                     y = i*(info['l'] + info['wood_h']) - info['wood_h']
+            
             for p in [0.1, 0.3, 0.5, 0.7, 0.9]:
                 ax.add_patch(patches.Rectangle((c*p-24, y), 48, info['wood_h'], fc='peru'))
                 
@@ -352,7 +367,7 @@ def fig_to_img(fig, w=400, h=300):
     return RLImage(buf, width=w, height=h, kind='proportional')
 
 # ==========================================
-# GERADOR DE PDF DETALHADO
+# GERADOR DE PDF DETALHADO (CORRIGIDO)
 # ==========================================
 
 def generate_pdf(res):
@@ -362,16 +377,19 @@ def generate_pdf(res):
     story = []
     m = res['main']
 
-    # Cabeçalho
     story.append(Paragraph(f"FICHA TÉCNICA DE PRODUÇÃO", styles['Title']))
     story.append(Spacer(1, 12))
 
-    # 1. Dados do Produto (Compara Pedido vs Norma)
     catalog_data = m.get('catalog_data')
-    if catalog_data:
-        cat_desc = f"Norma IT{13 if m['tipo']=='Redondo' else 14}: Qtde {catalog_data['qtde']} | Peso {catalog_data['peso']}kg"
-    else:
-        cat_desc = "Não catalogado (Cálculo Engenharia)"
+    cat_desc = f"Norma IT{13 if m['tipo']=='Redondo' else 14}: Qtde {catalog_data['qtde']} | Peso {catalog_data['peso']}kg" if catalog_data else "Não catalogado (Cálculo Engenharia)"
+    
+    # Definindo estilo explicitamente para evitar o erro getStyle()
+    t_style = TableStyle([
+        ('GRID', (0,0), (-1,-1), 1, colors.black),
+        ('BACKGROUND', (0,0), (0,0), colors.lightgrey),
+        ('FONTNAME', (0,0), (-1,0), 'Helvetica-Bold'),
+        ('ALIGN', (0,0), (-1,-1), 'LEFT')
+    ])
 
     tbl_data = [
         ["Parâmetro", "Valor Solicitado", "Dados de Referência"],
@@ -381,34 +399,27 @@ def generate_pdf(res):
     ]
     
     t1 = Table(tbl_data, colWidths=[120, 180, 220])
-    t1.setStyle(TableStyle([
-        ('GRID', (0,0), (-1,-1), 1, colors.black),
-        ('BACKGROUND', (0,0), (0,0), colors.lightgrey),
-        ('FONTNAME', (0,0), (-1,0), 'Helvetica-Bold'),
-        ('ALIGN', (0,0), (-1,-1), 'LEFT')
-    ]))
+    t1.setStyle(t_style)
     story.append(t1)
     story.append(Spacer(1, 15))
 
-    # 2. Dados Técnicos Individuais (Novo)
     story.append(Paragraph("Especificações do Tubo Individual", styles['Heading2']))
     
-    # Cria duas colunas: Texto e Desenho do Tubo
     tech_data = [
         ["Peso Teórico:", f"{m.get('peso_metro', 0):.3f} kg/m"],
         ["Área Seção:", f"{m.get('area_secao', 0):.2f} mm²"],
         ["Peso Total Peça:", f"{(m['peso']/m['num']):.3f} kg"]
     ]
     
-    # Desenho Individual
-    img_single = fig_to_img(plot_single_profile(m), w=150, h=150)
+    # Nested Table com o estilo definido
+    t_nested = Table(tech_data)
+    t_nested.setStyle(t_style)
     
-    # Tabela de layout (Texto esquerda, Imagem direita)
-    t_tech = Table([[Table(tech_data, style=t1.getStyle()), img_single]], colWidths=[300, 200])
+    img_single = fig_to_img(plot_single_profile(m), w=150, h=150)
+    t_tech = Table([[t_nested, img_single]], colWidths=[300, 200])
     story.append(t_tech)
     story.append(Spacer(1, 15))
 
-    # 3. Dados do Pacote (Embalagem)
     story.append(Paragraph(f"Configuração do Pacote ({m['num']} Peças)", styles['Heading2']))
     
     pack_data = [
@@ -423,12 +434,10 @@ def generate_pdf(res):
     story.append(t_pack)
     story.append(Spacer(1, 10))
 
-    # Imagens do Pacote
     story.append(fig_to_img(plot_cross_section(m), w=400, h=300))
     story.append(Spacer(1, 5))
     story.append(fig_to_img(plot_side_view(m), w=400, h=120))
 
-    # 4. Sobra
     if res['sobra']:
         story.append(PageBreak())
         story.append(Paragraph(f"Detalhe do Pacote de Sobra ({res['sobra']['num']} peças)", styles['Heading2']))
@@ -441,7 +450,7 @@ def generate_pdf(res):
 # ==========================================
 # INTERFACE
 # ==========================================
-st.title("🏭 SGP - Sistema de Gestão de Produção v14.0")
+st.title("🏭 SGP - Sistema de Gestão de Produção v14.1")
 
 with st.sidebar:
     st.header("Parâmetros")