Update app.py

app.py

@@ -13,9 +13,8 @@ from reportlab.lib.pagesizes import letter
 # ==========================================
 # CONFIGURAÇÃO
 # ==========================================
-st.set_page_config(page_title="Calculadora Formadoras v5.0", layout="wide")
+st.set_page_config(page_title="Calculadora Formadoras v6.0", layout="wide")
 
-# Lista de fitas disponíveis no estoque
 FITAS = [
     {"nome": "Fita 1.20", "tamanho": 1.20}, {"nome": "Fita 1.40", "tamanho": 1.40},
     {"nome": "Fita 1.50", "tamanho": 1.50}, {"nome": "Fita 1.60", "tamanho": 1.60},
@@ -66,20 +65,19 @@ class PackageCalculator:
         return best_item, best_dist
 
     def _escolher_fita(self, perimetro_m):
-        # Margem de 10% para o esticador pneumático e selo TR (igual código antigo)
+        # Margem de 10%
         perimetro_com_margem = perimetro_m * 1.10 
         for f in FITAS:
             if f["tamanho"] >= perimetro_com_margem:
                 return f["nome"], f["tamanho"], perimetro_com_margem
-        return "Sob Medida (>4m)", perimetro_com_margem, perimetro_com_margem
+        return "Sob Medida", perimetro_com_margem, perimetro_com_margem
 
     def calculate(self, modo, meta, l, a, e, c, tipo, acabamento, aba=0):
         l, a, e = float(l), float(a), float(e)
         is_galv = (acabamento == "Galvanizado (GI/GA)")
         wood_h = 6.60 if is_galv else 0.0
-        wood_w = 48.20 # Largura da madeira (visual)
+        wood_w = 48.20 
 
-        # Busca no Banco
         key_red = f"{l:.2f}_{e:.2f}"
         key_met = f"{l:.2f}_{a:.2f}_{e:.2f}"
         key_met_inv = f"{a:.2f}_{l:.2f}_{e:.2f}"
@@ -100,7 +98,6 @@ class PackageCalculator:
                 item = None
                 origem = "Engenharia (Cálculo)"
 
-        # Propriedades Físicas
         densidade = 7.85e-6
         if tipo == "Redondo":
             area = math.pi * ((l/2)**2 - (l/2-e)**2)
@@ -110,7 +107,6 @@ class PackageCalculator:
             area = (l + 2*a - 2*e) * e
         peso_unit = area * c * densidade
 
-        # Definição do Layout
         if item:
             num_pecas = item['qtde']
             if item.get('sextavado') or (tipo == "Redondo" and 'padrao' in item):
@@ -120,7 +116,6 @@ class PackageCalculator:
                 layout = None
                 is_sext, gl, ga = False, item.get('ml', 0), item.get('ma', 0)
         else:
-            # Fallback Matemático
             peso_alvo = 1000
             num_pecas = int(peso_alvo / peso_unit) if peso_unit > 0 else 1
             if tipo == "Redondo":
@@ -140,7 +135,6 @@ class PackageCalculator:
 
         peso_pct = num_pecas * peso_unit
 
-        # Totais
         if modo == "Toneladas":
             total_pcs = int((meta * 1000) / peso_unit) if peso_unit > 0 else 0
         else:
@@ -148,7 +142,6 @@ class PackageCalculator:
         qtd_pcts = total_pcs // num_pecas if num_pecas > 0 else 0
         sobra = total_pcs % num_pecas
 
-        # Montagem
         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, wood_w)
         
         sobra_pkg = None
@@ -178,7 +171,6 @@ class PackageCalculator:
             "is_galv": is_galv, "wood_h": wood_h, "wood_w": wood_w
         }
         
-        # Dimensões e Madeiras
         qtd_madeiras = 0
         if sext:
             rows = layout or []
@@ -186,9 +178,8 @@ class PackageCalculator:
             info["larg_total"] = max_c * l
             # Altura = Geometria + Madeira entre camadas
             h_geo = l + (len(rows) - 1) * l * math.sin(math.radians(60))
-            # Se Galv, 5 madeiras por linha (exceto a última talvez, mas vamos padronizar por camada usada)
             if is_galv:
-                h_wood = (len(rows)) * wood_h # Uma base + entre camadas
+                h_wood = (len(rows)) * wood_h 
                 qtd_madeiras = len(rows) * 5
             else:
                 h_wood = 0
@@ -196,9 +187,8 @@ class PackageCalculator:
         else:
             info["larg_total"] = gl * l
             if tipo == "Enrijecido": info["larg_total"] += aba
-            # Altura
             if is_galv:
-                h_wood = ga * wood_h # 1 camada de madeira para cada camada de tubo
+                h_wood = ga * wood_h 
                 qtd_madeiras = ga * 5
             else:
                 h_wood = 0
@@ -206,21 +196,17 @@ class PackageCalculator:
 
         info["qtd_madeiras"] = qtd_madeiras
 
-        # --- CÁLCULO DE FITAS ---
-        # 1. Fita Principal (Envolve tudo)
+        # Fita Principal
         perim_total = (info["larg_total"] + info["alt_total"]) * 2 / 1000
         nome_fita, tam_fita, real_fita = self._escolher_fita(perim_total)
         info["fita_nome"] = nome_fita
         info["fita_tamanho"] = tam_fita
-        info["fita_real_necessaria"] = real_fita
 
-        # 2. Fita do Meio (Só para normal)
+        # Fita Meio
         info["fita_meio_nome"] = "N/A"
         info["fita_meio_tamanho"] = 0
         
         if not is_galv:
-            # Fita que pega largura total + metade da altura
-            # Perimetro parcial = 2 * (Largura + Altura/2)
             perim_meio = (info["larg_total"] + (info["alt_total"] / 2)) * 2 / 1000
             n_meio, t_meio, r_meio = self._escolher_fita(perim_meio)
             info["fita_meio_nome"] = n_meio
@@ -238,75 +224,49 @@ class PackageCalculator:
 def plot_cross_section(info):
     fig, ax = plt.subplots(figsize=(10, 8))
     l, a, e = info['l'], info['a'], info['e']
-    wood_h, wood_w = info['wood_h'], info['wood_w']
+    wood_h = info['wood_h']
     is_galv = info['is_galv']
     
     fc = 'lightblue' if not info['is_rem'] else 'lightcoral'
     ec = 'black'
 
-    # Desenhar Peças e Madeira
     if info['sext']:
-        # Hexagonal
+        # --- SEXTAVADO ---
         rows = info['layout']
-        r = l / 2
+        radius = l / 2
         dy = l * 0.866
         total_w = info['larg_total']
         
         for i, count in enumerate(rows):
             row_w = count * l
             x0 = (total_w - row_w) / 2
-            # Y considera madeira acumulada
-            y_base = r + i * dy + (i * wood_h)
+            y_base = radius + i * dy + (i * wood_h)
             
-            # Desenhar 5 madeiras abaixo da linha (se galv)
+            # Madeira Transversal (Vista de Frente = Barra Contínua)
             if is_galv:
-                # Distribuir 5 madeiras na largura do pacote
-                # Y da madeira é logo abaixo da linha de tubos
-                wy = y_base - r - wood_h
-                espaco = total_w / 5
-                for m in range(5):
-                    wx = (m * espaco) + (espaco/2) - (wood_w/2)
-                    ax.add_patch(patches.Rectangle((wx, wy), wood_w, wood_h, fc='peru', ec='brown'))
+                # A madeira é um sarrafo transversal. No corte, vemos o comprimento da madeira atravessando a largura.
+                # Desenhamos uma barra inteira por baixo da camada
+                wy = y_base - radius - wood_h
+                ax.add_patch(patches.Rectangle((0, wy), total_w, wood_h, fc='peru', ec='brown'))
 
             for k in range(count):
-                cx = x0 + k * l + r
-                ax.add_patch(patches.Circle((cx, y_base), r, fc=fc, ec=ec))
-                ax.add_patch(patches.Circle((cx, y_base), r-e, fc='white', ec=ec, lw=0.5))
+                cx = x0 + k * l + radius
+                ax.add_patch(patches.Circle((cx, y_base), radius, fc=fc, ec=ec))
+                ax.add_patch(patches.Circle((cx, y_base), radius-e, fc='white', ec=ec, lw=0.5))
     else:
-        # Grade
+        # --- GRADE RETANGULAR ---
         gl, ga = info['gl'], info['ga']
         drawn = 0
         for i in range(ga):
             y = i * (a + wood_h)
             
-            # Madeira (Abaixo da camada)
-            if is_galv:
-                wy = y - wood_h if i > 0 else y # Na base desenha em 0? Vamos assumir que Y começa deslocado
-                # Ajuste: Vamos desenhar a madeira em y, e o tubo em y + wood_h
-                # Correção visual: 
-                # Camada 0: Tubo em 0? Se tiver madeira na base, tubo sobe.
-                # O User disse: "começa na primeira fileira". 
-                # Vamos desenhar a madeira embaixo de cada tubo.
-                tube_y = y 
-                if i == 0: # Base
-                    # Madeira base
-                    wy = 0
-                    tube_y = wood_h
-                else:
-                    # Madeira entre camadas
-                    wy = i * (a + wood_h)
-                    tube_y = wy + wood_h
-                
-                # Mas para manter simples: Y calculado acima já considera espaçamento
-                # Vamos desenhar a madeira 'sob' o tubo dessa iteração
-                # Distribui 5 na largura
-                espaco = info['larg_total'] / 5
-                for m in range(5):
-                    wx = (m * espaco) + (espaco/2) - (wood_w/2)
-                    # Desenha madeira na base da camada i
-                    ax.add_patch(patches.Rectangle((wx, i*(a+wood_h)), wood_w, wood_h, fc='peru', ec='brown'))
-            
-            tube_y_draw = i * (a + wood_h) + (wood_h if is_galv else 0)
+            # Madeira Transversal (Barra Contínua)
+            if is_galv: 
+                # Desenha madeira na base da camada
+                ax.add_patch(patches.Rectangle((0, y), info['larg_total'], wood_h, fc='peru', ec='brown'))
+                tube_y_draw = y + wood_h
+            else:
+                tube_y_draw = y
 
             for j in range(gl):
                 if drawn >= info['num']: break
@@ -320,24 +280,14 @@ def plot_cross_section(info):
                     ax.add_patch(patches.Rectangle((x+inset, tube_y_draw+inset), l-2*inset, a-2*inset, fc='white', ec=ec, lw=0.5))
                 drawn += 1
 
-    # Cotas e Fitas
+    # Cotas
     L_tot, A_tot = info['larg_total'], info['alt_total']
     margin = max(L_tot, A_tot) * 0.2
     
-    # Bounding Box
     ax.add_patch(patches.Rectangle((-2, -2), L_tot+4, A_tot+4, fill=False, ec='red', ls='--', lw=1.5, label='Limite'))
     
-    # Cinta Principal
-    off = 5
-    ax.add_patch(patches.Rectangle((-off, -off), L_tot+2*off, A_tot+2*off, fill=False, ec='purple', ls='-', lw=2, label='Fita Principal'))
-    
-    # Cinta Meio (Se houver)
-    if info['fita_meio_tamanho'] > 0:
-        # Representação visual da fita de meio (metade da altura)
-        ax.add_patch(patches.Rectangle((-off-2, -off-2), L_tot+2*(off+2), (A_tot/2)+2*(off+2), fill=False, ec='orange', ls='--', lw=2, label='Fita Meio'))
-
-    ax.text(L_tot/2, -margin/2, f"{L_tot:.1f} mm", ha='center', va='center', fontweight='bold')
-    ax.text(-margin/2, A_tot/2, f"{A_tot:.1f} mm", ha='center', va='center', fontweight='bold', rotation=90)
+    ax.text(L_tot/2, -margin/2, f"L: {L_tot:.1f} mm", ha='center', va='center', fontweight='bold')
+    ax.text(-margin/2, A_tot/2, f"A: {A_tot:.1f} mm", ha='center', va='center', fontweight='bold', rotation=90)
     
     ax.set_xlim(-margin, L_tot+margin)
     ax.set_ylim(-margin, A_tot+margin)
@@ -345,8 +295,7 @@ def plot_cross_section(info):
     ax.axis('off')
     
     if is_galv:
-        # Legenda madeira dummy
-        ax.add_patch(patches.Rectangle((0,0),0,0, fc='peru', label=f"{info['qtd_madeiras']} Madeiras"))
+        ax.plot([], [], color='peru', label='Madeira (Vista Transversal)', lw=4)
 
     plt.title(f"{info['origem']} | {info['num']} Pçs | {info['peso']:.1f}kg", fontsize=11)
     plt.legend(loc='upper right', fontsize='small')
@@ -356,29 +305,63 @@ def plot_side_view(info):
     fig, ax = plt.subplots(figsize=(10, 3))
     c, h = info['c'], info['alt_total']
     is_galv = info['is_galv']
+    wood_w = info['wood_w']
+    wood_h = info['wood_h']
     
     ax.add_patch(patches.Rectangle((0, 0), c, h, fc='lightgray', ec='black'))
     
+    # --- DESENHO DAS 5 MADEIRAS (VISTA LATERAL) ---
+    if is_galv:
+        # Temos 'ga' camadas (ou len(layout)). Para cada camada, desenhamos 5 tocos de madeira.
+        # Precisamos saber a altura Y de cada camada de madeira.
+        layers = info['ga'] if not info['sext'] else len(info['layout'])
+        
+        # Distribuição ao longo do comprimento (5 posições)
+        positions_wood = [0.1, 0.3, 0.5, 0.7, 0.9]
+        
+        for i in range(layers):
+            # Calcular Y da madeira dessa camada
+            if info['sext']:
+                # Sextavado: madeira entra abaixo da camada i
+                radius = info['l'] / 2
+                dy = info['l'] * 0.866
+                y_wood = (radius + i * dy + (i * wood_h)) - radius - wood_h
+            else:
+                # Retangular
+                y_wood = i * (info['a'] + wood_h)
+            
+            # Desenhar os 5 tocos nessa altura Y
+            for pos in positions_wood:
+                wx = (c * pos) - (wood_w / 2)
+                ax.add_patch(patches.Rectangle((wx, y_wood), wood_w, wood_h, fc='peru', ec='brown'))
+
     # Fitas
-    positions = [0.1, 0.3, 0.5, 0.7, 0.9]
-    for idx, p in enumerate(positions):
+    positions_fita = [0.1, 0.3, 0.5, 0.7, 0.9]
+    for idx, p in enumerate(positions_fita):
         x = c * p
         if not is_galv and idx == 2: # Meio
-            ax.plot([x, x], [0, h/2], color='orange', lw=3, ls='--') # Fita meio até metade
+            ax.plot([x, x], [0, h/2], color='orange', lw=3, ls='--') 
         else:
-            ax.plot([x, x], [0, h], color='purple', lw=3) # Fita completa
+            ax.plot([x, x], [0, h], color='purple', lw=3)
 
     ax.text(c/2, -h*0.2, f"COMPRIMENTO: {c:.0f} mm", ha='center', va='top', fontweight='bold', color='black')
     ax.set_xlim(-c*0.05, c*1.05)
     ax.set_ylim(-h*0.3, h*1.1)
     ax.axis('off')
-    plt.title(f"Vista Lateral - {info['desc_fitas']}", fontsize=10)
+    plt.title(f"Vista Lateral - {info['desc_fitas']} + Madeiras", fontsize=10)
     return fig
 
 # ==========================================
 # EXPORTAÇÃO
 # ==========================================
 
+def fig_to_img(fig):
+    buf = io.BytesIO()
+    fig.savefig(buf, format='png', dpi=100, bbox_inches='tight')
+    buf.seek(0)
+    plt.close(fig)
+    return RLImage(buf, width=450, height=350, kind='proportional')
+
 def generate_pdf(res):
     buffer = io.BytesIO()
     doc = SimpleDocTemplate(buffer, pagesize=letter)
@@ -387,40 +370,41 @@ def generate_pdf(res):
     m = res['main']
     
     story.append(Paragraph(f"Ficha: {m['tipo']} {m['l']}x{m['a']} #{m['e']}", styles['Title']))
+    story.append(Spacer(1, 12))
     
-    # Detalhes Fitas e Madeira
-    fita_txt = f"4x {m['fita']} ({m['fita_tamanho']}m)"
+    acabamento = "Galvanizado (Com Madeira)" if m['is_galv'] else "Preto (Normal)"
+    
+    # CORREÇÃO AQUI: Usar fita_nome
+    fita_txt = f"4x {m['fita_nome']} ({m['fita_tamanho']}m)"
     if m['fita_meio_tamanho'] > 0:
         fita_txt += f" + 1x {m['fita_meio_nome']} ({m['fita_meio_tamanho']}m)"
     else:
-        fita_txt = f"5x {m['fita']} ({m['fita_tamanho']}m)"
+        fita_txt = f"5x {m['fita_nome']} ({m['fita_tamanho']}m)"
     
     wood_txt = f"{m['qtd_madeiras']} peças (48.2x6.6mm)" if m['is_galv'] else "N/A"
 
     data = [
         ["Item", f"{m['l']}x{m['a']} #{m['e']}mm"],
-        ["Acabamento", "Galvanizado" if m['is_galv'] else "Preto"],
-        ["Total Produção", f"{res['total_pcts']} pacotes"],
+        ["Acabamento", acabamento],
+        ["Produção", f"{res['total_pcts']} pct + {1 if res['sobra'] else 0} sobra"],
+        ["Peso Total", f"{res['peso_total']:.2f} kg"],
         ["Cintas", fita_txt],
-        ["Madeiras", wood_txt],
-        ["Peso Total", f"{res['peso_total']:.2f} kg"]
+        ["Madeiras", wood_txt]
     ]
     t = Table(data, colWidths=[150, 300])
     t.setStyle(TableStyle([('GRID', (0,0), (-1,-1), 1, colors.black), ('BACKGROUND', (0,0), (0,-1), colors.lightgrey)]))
     story.append(t)
     story.append(Spacer(1, 20))
     
-    # Imagens
-    buf1 = io.BytesIO()
-    plot_cross_section(m).savefig(buf1, format='png', dpi=100, bbox_inches='tight')
-    buf1.seek(0)
-    story.append(RLImage(buf1, width=400, height=300, kind='proportional'))
-    
-    buf2 = io.BytesIO()
-    plot_side_view(m).savefig(buf2, format='png', dpi=100, bbox_inches='tight')
-    buf2.seek(0)
-    story.append(RLImage(buf2, width=400, height=150, kind='proportional'))
+    story.append(Paragraph(f"1. Pacote Padrão ({m['num']} pçs)", styles['Heading2']))
+    story.append(fig_to_img(plot_cross_section(m)))
+    story.append(fig_to_img(plot_side_view(m)))
     
+    if res['sobra']:
+        story.append(PageBreak())
+        story.append(Paragraph(f"2. Pacote de Sobra ({res['sobra']['num']} pçs)", styles['Heading2']))
+        story.append(fig_to_img(plot_cross_section(res['sobra'])))
+
     doc.build(story)
     buffer.seek(0)
     return buffer
@@ -431,14 +415,15 @@ def generate_csv(res):
     m = res['main']
     writer.writerow(["Tipo", "L", "A", "E", "Acab", "Qtd Pcts", "Pecas/Pct", "Fita Princ", "Fita Meio", "Qtd Madeira"])
     acab = "GI/GA" if m['is_galv'] else "Normal"
-    writer.writerow([m['tipo'], m['l'], m['a'], m['e'], acab, res['total_pcts'], m['num'], m['fita'], m['fita_meio_nome'], m['qtd_madeiras']])
+    # CORREÇÃO AQUI TAMBÉM: Usar fita_nome
+    writer.writerow([m['tipo'], m['l'], m['a'], m['e'], acab, res['total_pcts'], m['num'], m['fita_nome'], m['fita_meio_nome'], m['qtd_madeiras']])
     return output.getvalue()
 
 # ==========================================
 # INTERFACE
 # ==========================================
 
-st.title("🏭 Calculadora Formadoras v5.0")
+st.title("🏭 Calculadora Formadoras v6.0")
 
 with st.sidebar:
     st.header("Parâmetros")
@@ -446,10 +431,11 @@ with st.sidebar:
     acabamento = st.radio("Acabamento", ["Preto (Normal)", "Galvanizado (GI/GA)"])
     
     c1, c2 = st.columns(2)
-    l = c1.number_input("Largura/Diâm (mm)", value=50.0)
-    a = c2.number_input("Altura (mm)", value=50.0, disabled=(tipo=="Redondo"))
-    e = st.number_input("Espessura (mm)", value=2.00)
-    c = st.number_input("Comprimento (mm)", value=6000)
+    l = c1.number_input("Largura/Diâm (mm)", value=50.0, step=0.1)
+    a = c2.number_input("Altura (mm)", value=50.0, step=0.1, disabled=(tipo=="Redondo"))
+    e = c3.number_input("Espessura (mm)", value=2.00, step=0.05)
+    c = c4.number_input("Comp. (mm)", value=6000)
+    
     aba = st.number_input("Aba (U)", value=0.0, disabled=(tipo!="U/Enrijecido"))
     
     st.markdown("---")