Update app.py

app.py

@@ -1,114 +1,95 @@
 import os
 import streamlit as st
+from groq import Groq
 import matplotlib.pyplot as plt
 from fpdf import FPDF
-from groq import Groq
-
-# Set your API key
-os.environ["GROQ_API_KEY"] = "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"
+import io
 
 # Initialize Groq client
-client = Groq(api_key=os.environ["GROQ_API_KEY"])
+client = Groq(api_key=os.environ.get("GROQ_API_KEY"))
 
-# Function to calculate cutting areas
-def calculate_areas(blank_plate, pcs, saw_cut, trim_size):
-    blank_area = blank_plate['thickness'] * blank_plate['width'] * blank_plate['length']
-    total_cut_area = sum([pc['width'] * pc['length'] for pc in pcs])
-    
-    # Assuming trimming and saw cuts reduce the usable area
-    trimmed_area = blank_area - (2 * trim_size * (blank_plate['width'] + blank_plate['length']))
-    usable_area = trimmed_area - (len(pcs) * saw_cut)
-    
-    non_used_area = blank_area - total_cut_area
-    usable_percentage = (usable_area / blank_area) * 100
-    
-    return usable_area, total_cut_area, non_used_area, usable_percentage
+# App title
+st.title("Cutting Plan Generator")
 
-# Function to generate a sketch (simplified)
-def generate_sketch(blank_plate, pcs):
-    fig, ax = plt.subplots(figsize=(6, 6))
-    ax.set_xlim([0, blank_plate['width']])
-    ax.set_ylim([0, blank_plate['length']])
-    
-    # Draw the blank plate
-    ax.add_patch(plt.Rectangle((0, 0), blank_plate['width'], blank_plate['length'], fill=None, edgecolor='black'))
-    
-    # Draw each piece to be cut
-    for i, pc in enumerate(pcs):
-        ax.add_patch(plt.Rectangle((pc['x'], pc['y']), pc['width'], pc['length'], fill=True, color="blue", alpha=0.5))
-    
-    plt.axis('off')
-    st.pyplot(fig)
+# Input for blank plates
+st.header("Blank Plates Details")
+num_blank_types = st.number_input("Number of Blank Plate Types", min_value=1, step=1, key="num_blank_types")
+blank_data = []
+for i in range(num_blank_types):
+    with st.expander(f"Blank Plate {i+1}"):
+        blank_width = st.number_input(f"Width (mm) for Blank {i+1}", min_value=0.0, step=0.1, key=f"blank_width_{i}")
+        blank_length = st.number_input(f"Length (mm) for Blank {i+1}", min_value=0.0, step=0.1, key=f"blank_length_{i}")
+        num_blank_plates = st.number_input(f"Number of Plates for Blank {i+1}", min_value=1, step=1, key=f"num_blank_plates_{i}")
+        blank_data.append({"width": blank_width, "length": blank_length, "count": num_blank_plates})
 
-# Function to generate PDF
-def generate_pdf(blank_plate, pcs, areas):
-    pdf = FPDF()
-    pdf.add_page()
-    
-    pdf.set_font("Arial", size=12)
-    pdf.cell(200, 10, txt="Cutting Plan Report", ln=True, align='C')
-    
-    # Blank Plate details
-    pdf.ln(10)
-    pdf.cell(200, 10, txt=f"Blank Plate: {blank_plate['thickness']}mm x {blank_plate['width']}mm x {blank_plate['length']}mm", ln=True)
-    
-    # Area details
-    pdf.ln(5)
-    pdf.cell(200, 10, txt=f"Usable Area: {areas[0]} mm²", ln=True)
-    pdf.cell(200, 10, txt=f"Cut Area: {areas[1]} mm²", ln=True)
-    pdf.cell(200, 10, txt=f"Non-Used Area: {areas[2]} mm²", ln=True)
-    pdf.cell(200, 10, txt=f"Usable Area Percentage: {areas[3]}%", ln=True)
-    
-    # Sketch generation (placeholder)
-    pdf.ln(10)
-    pdf.cell(200, 10, txt="Sketch of Cutting Plan:", ln=True)
-    
-    # Save the PDF
-    pdf.output("cutting_plan_report.pdf")
-    st.success("PDF generated successfully!")
+# Input for pieces
+st.header("Pieces to be Cut Details")
+num_piece_types = st.number_input("Number of Piece Types", min_value=1, step=1, key="num_piece_types")
+piece_data = []
+for i in range(num_piece_types):
+    with st.expander(f"Piece {i+1}"):
+        piece_width = st.number_input(f"Width (mm) for Piece {i+1}", min_value=0.0, step=0.1, key=f"piece_width_{i}")
+        piece_length = st.number_input(f"Length (mm) for Piece {i+1}", min_value=0.0, step=0.1, key=f"piece_length_{i}")
+        num_pieces = st.number_input(f"Number of Pieces for Piece {i+1}", min_value=1, step=1, key=f"num_pieces_{i}")
+        piece_data.append({"width": piece_width, "length": piece_length, "count": num_pieces})
 
-# Streamlit UI
-st.title("Cutting Plan Generator")
+# Saw cut size and trimming size
+st.header("Other Specifications")
+saw_cut_size = st.number_input("Saw Cut Size (mm)", min_value=0.0, step=0.1, key="saw_cut_size")
+trimming_size = st.number_input("Trimming Size (mm)", min_value=0.0, step=0.1, key="trimming_size")
 
-# Inputs for blank plates
-blank_plates = []
-num_blank_plates = st.number_input("Number of Blank Plates", min_value=1, value=1)
+# Priority of blank sizes
+priority = st.selectbox("Priority of Blank Sizes", options=["Smallest First", "Largest First"], key="priority")
 
-for i in range(num_blank_plates):
-    st.header(f"Blank Plate {i+1}")
-    thickness = st.number_input(f"Blank Plate {i+1} Thickness (mm)", min_value=1, value=10)
-    width = st.number_input(f"Blank Plate {i+1} Width (mm)", min_value=1, value=1000)
-    length = st.number_input(f"Blank Plate {i+1} Length (mm)", min_value=1, value=2000)
-    num_plates = st.number_input(f"Blank Plate {i+1} Quantity", min_value=1, value=1)
-    
-    blank_plates.append({'thickness': thickness, 'width': width, 'length': length, 'num_plates': num_plates})
+# Generate Cutting Plan and Sketch
+if st.button("Generate Cutting Plan"):
+    total_cut_area = 0
+    blank_areas = []
+    for blank in blank_data:
+        blank_area = (blank["width"] - 2 * trimming_size) * (blank["length"] - 2 * trimming_size) * blank["count"]
+        blank_areas.append(blank_area)
 
-# Inputs for pieces to be cut
-pcs = []
-num_pcs = st.number_input("Number of Pieces to Cut", min_value=1, value=5)
+    for piece in piece_data:
+        total_cut_area += piece["width"] * piece["length"] * piece["count"]
 
-for i in range(num_pcs):
-    st.header(f"Piece {i+1}")
-    pc_width = st.number_input(f"Piece {i+1} Width (mm)", min_value=1, value=200)
-    pc_length = st.number_input(f"Piece {i+1} Length (mm)", min_value=1, value=500)
-    pcs.append({'width': pc_width, 'length': pc_length, 'x': 0, 'y': 0})
+    total_blank_area = sum(blank_areas)
+    unused_area = total_blank_area - total_cut_area
+    usable_area_percentage = (total_cut_area / total_blank_area) * 100
 
-# Saw cut and trimming size
-saw_cut = st.number_input("Saw Cut Size (mm)", min_value=0, value=5)
-trim_size = st.number_input("Trimming Size (mm)", min_value=0, value=10)
+    st.subheader("Results")
+    st.write(f"Total Usable Surface Area: {total_blank_area:.2f} mm²")
+    st.write(f"Total Cut Square Area: {total_cut_area:.2f} mm²")
+    st.write(f"Total Non-used Square Area: {unused_area:.2f} mm²")
+    st.write(f"Percentage of Usable Area: {usable_area_percentage:.2f}%")
 
-# Calculate and display the cutting areas for each blank plate
-for blank_plate in blank_plates:
-    areas = calculate_areas(blank_plate, pcs, saw_cut, trim_size)
-    generate_sketch(blank_plate, pcs)
+    # Generate sketch
+    fig, ax = plt.subplots(figsize=(8, 6))
+    for i, blank in enumerate(blank_data):
+        ax.add_patch(plt.Rectangle((0, i * (blank["length"] + 10)), blank["width"], blank["length"], edgecolor="blue", facecolor="none", label=f"Blank {i+1}" if i == 0 else ""))
+        for j, piece in enumerate(piece_data):
+            ax.add_patch(plt.Rectangle((j * (piece["width"] + 5), i * (blank["length"] + 10) + trimming_size), piece["width"], piece["length"], edgecolor="red", facecolor="pink", label=f"Piece {j+1}" if i == 0 else ""))
+
+    ax.set_xlim(0, max(blank["width"] for blank in blank_data) + 50)
+    ax.set_ylim(0, (len(blank_data) * max(blank["length"] for blank in blank_data)) + 100)
+    ax.set_title("Cutting Plan Sketch")
+    ax.legend(loc="upper right")
+    st.pyplot(fig)
 
-    st.write(f"Blank Plate {blank_plates.index(blank_plate)+1}:")
-    st.write(f"Usable Area: {areas[0]} mm²")
-    st.write(f"Cut Area: {areas[1]} mm²")
-    st.write(f"Non-Used Area: {areas[2]} mm²")
-    st.write(f"Usable Area Percentage: {areas[3]}%")
+    # Save as PDF
+    pdf = FPDF()
+    pdf.add_page()
+    pdf.set_font("Arial", size=12)
+    pdf.cell(200, 10, txt="Cutting Plan Report", ln=True, align="C")
+    pdf.cell(200, 10, txt=f"Total Usable Surface Area: {total_blank_area:.2f} mm²", ln=True)
+    pdf.cell(200, 10, txt=f"Total Cut Square Area: {total_cut_area:.2f} mm²", ln=True)
+    pdf.cell(200, 10, txt=f"Total Non-used Square Area: {unused_area:.2f} mm²", ln=True)
+    pdf.cell(200, 10, txt=f"Percentage of Usable Area: {usable_area_percentage:.2f}%", ln=True)
 
-# Generate PDF
-if st.button("Generate PDF"):
-    generate_pdf(blank_plates[0], pcs, areas)
-    st.download_button("Download PDF", "cutting_plan_report.pdf")
+    pdf_output = io.BytesIO()
+    pdf.output(pdf_output)
+    st.download_button(
+        label="Download Cutting Plan as PDF",
+        data=pdf_output.getvalue(),
+        file_name="cutting_plan.pdf",
+        mime="application/pdf",
+    )