Update app.py

app.py

@@ -1,107 +1,91 @@
 import gradio as gr
 import pandas as pd
-import os
 import json
+import os
 
-JSON_PATH = "/tmp/material_db.json"
+DB_PATH = "/tmp/material_db.json"
+
+def extract_materials(file, description):
+    if file is None or description.strip() == "":
+        return "❌ لطفاً فایل اکسل و توضیح پروژه را وارد کنید.", "", None
 
-def parse_excel(file):
     try:
         df = pd.read_excel(file.name, header=2)
     except Exception as e:
-        return None, None, f"❌ خطا در خواندن فایل اکسل: {str(e)}"
-
-    column_mapping = {
-        "code": ["کد", "کد فهرست بها"],
-        "description": ["عنوان", "شرح کالا", "نام کالا"],
-        "quantity": ["تعداد", "تعداد مورد نیاز"],
-        "unit": ["واحد", "واحد شمارش", "نوع واحد"]
-    }
+        return f"❌ خطا در خواندن فایل: {e}", "", None
 
-    resolved = {}
-    for key, options in column_mapping.items():
-        found = next((col for col in options if col in df.columns), None)
-        if not found:
-            return None, None, f"❌ ستون «{key}» در فایل پیدا نشد."
-        resolved[key] = found
+    required_cols = ["شرح کالا", "تعداد", "واحد"]
+    for col in required_cols:
+        if col not in df.columns:
+            return f"❌ ستون '{col}' در فایل یافت نشد.", "", None
 
-    materials = []
     constraints = []
+    materials = []
+
     for _, row in df.iterrows():
         try:
-            qty = float(row[resolved["quantity"]])
-            if pd.isna(qty) or qty <= 0:
-                continue
+            desc = str(row["شرح کالا"]).strip()
+            qty = float(row["تعداد"])
+            unit = str(row["واحد"]).strip()
         except:
             continue
 
-        material = {
-            "code": str(row[resolved["code"]]).strip(),
-            "description": str(row[resolved["description"]]).strip(),
+        if qty <= 0 or desc == "":
+            continue
+
+        constraint = f"{desc} ({qty} {unit})"
+        constraints.append(constraint)
+        materials.append({
+            "description": desc,
             "quantity": qty,
-            "unit": str(row[resolved["unit"]]).strip()
-        }
-        materials.append(material)
+            "unit": unit
+        })
 
-        constraints.append(f"{material['description']} ({material['quantity']} {material['unit']})")
+    record = {
+        "project_description": description.strip(),
+        "items": materials
+    }
 
+    # Save to JSON file
     try:
-        if os.path.exists(JSON_PATH):
-            with open(JSON_PATH, "r", encoding="utf-8") as f:
-                existing = json.load(f)
+        if os.path.exists(DB_PATH):
+            with open(DB_PATH, "r", encoding="utf-8") as f:
+                data = json.load(f)
         else:
-            existing = []
+            data = []
 
-        new_materials = [m for m in materials if m not in existing]
-        existing.extend(new_materials)
+        data.append(record)
 
-        with open(JSON_PATH, "w", encoding="utf-8") as f:
-            json.dump(existing, f, indent=2, ensure_ascii=False)
+        with open(DB_PATH, "w", encoding="utf-8") as f:
+            json.dump(data, f, ensure_ascii=False, indent=2)
 
     except Exception as e:
-        return None, None, f"❌ خطا در ذخیره فایل JSON: {str(e)}"
-
-    return constraints, len(new_materials), f"✅ {len(new_materials)} مورد جدید با موفقیت ثبت شدند."
+        return f"❌ خطا در ذخیره فایل JSON: {e}", "", None
 
+    return f"✅ {len(materials)} قلم با موفقیت ثبت شد.", "\n".join(constraints), DB_PATH
 
-def get_json_file():
-    if os.path.exists(JSON_PATH):
-        return JSON_PATH
-    else:
-        return None
+def download_json():
+    return DB_PATH if os.path.exists(DB_PATH) else None
 
-
-with gr.Blocks(title="مدیریت مواد پروژه") as demo:
-    gr.Markdown("""
-    # 🛠️ سامانه ثبت اقلام پروژه
-
-    در این ابزار می‌توانید فایل اکسل برآورد مصالح پروژه را بارگذاری کرده،
-    اقلام را استخراج و ثبت نمایید. همچنین امکان دریافت خروجی JSON فراهم است.
-    """)
+with gr.Blocks(title="ثبت اقلام و ساخت کتابخانه قیود") as demo:
+    gr.Markdown("## 📁 ثبت اقلام پروژه و ساخت کتابخانه قیود")
 
     with gr.Row():
-        file_input = gr.File(label="📂 انتخاب فایل اکسل")
-        upload_btn = gr.Button("📤 ثبت اقلام")
-
-    status_output = gr.Textbox(label="وضعیت", lines=2)
-    constraints_output = gr.Textbox(label="🔍 اقلام استخراج‌شده", lines=10)
+        file_input = gr.File(label="📂 بارگذاری فایل اکسل")
+        description_input = gr.Textbox(label="📝 توصیف پروژه", placeholder="مثلاً: پروژه خط انتقال 63 کیلوولت تبریز")
 
-    download_btn = gr.Button("📥 دریافت فایل JSON")
-    json_file_output = gr.File(label="دانلود فایل", visible=False)
+    upload_btn = gr.Button("📤 ثبت اقلام")
+    status_output = gr.Textbox(label="📌 وضعیت", lines=2)
+    constraints_output = gr.Textbox(label="🧾 قیود استخراج‌شده", lines=10)
 
-    def handle_upload(file):
-        if file is None:
-            return "", "", "⚠️ لطفاً یک فایل اکسل انتخاب کنید."
-        constraints, count, msg = parse_excel(file)
-        joined = "\n".join(constraints) if constraints else ""
-        return msg, joined
+    download_btn = gr.Button("📥 دانلود فایل JSON")
+    json_file_output = gr.File(label="فایل خروجی", visible=True)
 
-    def handle_download():
-        path = get_json_file()
-        return path if path else None
+    upload_btn.click(fn=extract_materials,
+                     inputs=[file_input, description_input],
+                     outputs=[status_output, constraints_output, json_file_output])
 
-    upload_btn.click(handle_upload, inputs=[file_input], outputs=[status_output, constraints_output])
-    download_btn.click(handle_download, outputs=[json_file_output])
+    download_btn.click(fn=download_json, outputs=json_file_output)
 
 if __name__ == "__main__":
     demo.launch()