Update app.py

app.py

@@ -1,63 +1,118 @@
+# app.py
 import gradio as gr
+from PIL import Image, ImageDraw
+from rectpack import newPacker, PackingMode, PackingAlgorithm
 import pandas as pd
 
-def read_google_sheet(csv_url):
-    try:
-        if not csv_url:
-            return "⚠️ Silakan masukkan URL CSV Google Sheet"
-        
-        # Baca CSV dari URL
-        df = pd.read_csv(csv_url)
-        
-        # Bersihkan DataFrame:
-        # Hapus baris dan kolom yang seluruhnya kosong
-        df = df.dropna(how="all")
-        df = df.dropna(axis=1, how="all")
-        
-        # Ubah nama kolom yang tidak informatif (misalnya 'Unnamed: ...') menjadi nama generik
-        df.columns = [f"Column_{i}" if "Unnamed" in col else col for i, col in enumerate(df.columns)]
+def urut_dan_nesting(lebar_roll, daftar_banner):
+    tinggi_maks_roll = 2000 
+    packer = newPacker(
+        mode=PackingMode.Rectangle,
+        bin_algo=PackingAlgorithm.MaxRectsBl,
+        rotation=True
+    )
+    packer.add_bin(lebar_roll, tinggi_maks_roll)
+    
+    for banner in daftar_banner:
+        w, h, nama = banner
+        packer.add_rect(int(w), int(h), rid=str(nama))
         
-        # Konversi DataFrame ke JSON agar lebih mudah diproses oleh AI
-        json_output = df.to_json(orient="records", force_ascii=False, indent=2)
-        return json_output
+    packer.pack()
+    
+    hasil_layout = []
+    tinggi_pakai_maksimal = 0
     
-    except Exception as e:
-        return f"❌ Error: {str(e)}"
+    for rect in packer.rect_list():
+        b_idx, x, y, w, h, rid = rect
+        hasil_layout.append({
+            "nama": rid,
+            "x": x,
+            "y": y,
+            "lebar_cetak": w,
+            "tinggi_cetak": h
+        })
+        if y + h > tinggi_pakai_maksimal:
+            tinggi_pakai_maksimal = y + h
+            
+    return hasil_layout, max(tinggi_pakai_maksimal, 10)
 
-with gr.Blocks(theme=gr.themes.Soft(), title="Google Sheet Viewer") as demo:
-    with gr.Column():
-        gr.Markdown("""
-        # 📊 Google Sheet Viewer
-        **Cara Penggunaan:**
-        1. Buka Google Sheet anda
-        2. Pilih *File > Share > Publish to web*
-        3. Pilih format **CSV** dan publish
-        4. Salin URL CSV yang dihasilkan
-        """)
+def proses_nesting(lebar_roll, df_order):
+    if df_order is None or df_order.empty:
+        return None, "❌ Daftar banner masih kosong."
         
-        # Input URL menggunakan Textbox
-        url_input = gr.Textbox(
-            label="URL CSV Google Sheet",
-            placeholder="Contoh: https://docs.google.com/spreadsheets/d/.../export?format=csv&gid=0",
-            interactive=True
-        )
+    daftar_banner = []
+    for index, row in df_order.iterrows():
+        try:
+            w = float(row.iloc[0])
+            h = float(row.iloc[1])
+            nama = str(row.iloc[2])
+            if w > 0 and h > 0:
+                daftar_banner.append((w, h, nama))
+        except Exception:
+            continue
+            
+    if not daftar_banner:
+        return None, "❌ Format data salah. Pastikan kolom Lebar dan Tinggi berupa angka."
+
+    layout, total_panjang_bahan = urut_dan_nesting(lebar_roll, daftar_banner)
+    
+    SKALA = 4 
+    img_w = int(lebar_roll * SKALA)
+    img_h = int(total_panjang_bahan * SKALA)
+    
+    canvas = Image.new("RGB", (img_w, img_h), "#F0F0F0")
+    draw = ImageDraw.Draw(canvas)
+    
+    for item in layout:
+        x1 = item["x"] * SKALA
+        y1 = (total_panjang_bahan - item["y"] - item["tinggi_cetak"]) * SKALA 
+        x2 = x1 + (item["lebar_cetak"] * SKALA)
+        y2 = y1 + (item["tinggi_cetak"] * SKALA)
         
-        # Tombol submit
-        submit_button = gr.Button("Tampilkan Data", variant="primary", size="lg")
+        draw.rectangle([x1, y1, x2, y2], fill="#A8DADC", outline="#1D3557", width=2)
+        draw.text((x1 + 10, y1 + 10), f"{item['nama']}\n({item['lebar_cetak']}x{item['tinggi_cetak']})", fill="#1D3557")
         
-        # Output JSON sebagai teks
-        output = gr.Textbox(
-            label="JSON Output (AI Friendly)",
-            lines=10,
-            interactive=False
-        )
+    info = f"✅ **Berhasil dihitung!**\n- Lebar Roll: **{lebar_roll} cm**\n- Panjang Bahan Terpakai: **{total_panjang_bahan} cm**"
+    return canvas, info
+
+# Data awal default untuk demo
+data_default = pd.DataFrame([
+    [100, 100, "Banner_A_Agus"],
+    [60, 90, "Banner_B_Budi_1"],
+    [60, 90, "Banner_B_Budi_2"],
+    [40, 80, "Banner_C_Cici"],
+    [110, 50, "Banner_D_Deni"],
+    [30, 30, "Logo_Kecil_1"],
+    [30, 30, "Logo_Kecil_2"],
+    [50, 70, "Banner_E_Eka"]
+], columns=["Lebar (cm)", "Tinggi (cm)", "Nama / ID Order"])
+
+app = gr.Blocks(theme=gr.themes.Soft())
+
+with app:
+    gr.Markdown("# 🖨️ Kalkulator Nesting & Layout Banner Otomatis")
+    gr.Markdown("Atur ukuran roll dan masukkan daftar pesanan banner untuk mengoptimalkan pemotongan bahan secara otomatis.")
     
-    # Handler untuk tombol submit
-    submit_button.click(
-        fn=read_google_sheet,
-        inputs=url_input,
-        outputs=output
-    )
+    with gr.Row():
+        with gr.Column(scale=1):
+            lebar_roll_input = gr.Number(value=120, label="Lebar Roll Mesin (cm)")
+            tabel_order = gr.Dataframe(
+                value=data_default,
+                headers=["Lebar (cm)", "Tinggi (cm)", "Nama / ID Order"],
+                datatype=["number", "number", "str"],
+                row_count=(5, "dynamic"),
+                col_count=(3, "fixed"),
+                label="Daftar Order Banner"
+            )
+            btn_proses = gr.Button("🚀 Hitung & Generate Layout", variant="primary")
+            
+        with gr.Column(scale=1):
+            output_info = gr.Markdown(label="Informasi Hasil")
+            output_image = gr.Image(label="Pratinjau Layout Operator", type="pil")
+            
+    btn_proses.fn = proses_nesting
+    btn_proses.inputs = [lebar_roll_input, tabel_order]
+    btn_proses.outputs = [output_image, output_info]
 
 if __name__ == "__main__":
-    demo.launch()
+    app.launch()