Update app.py

app.py

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