Update app.py

app.py

@@ -1,76 +1,90 @@
 import gradio as gr
+import pandas as pd
 import os
 import json
-from utils import parse_excel_file, save_to_json
 
-DATA_FILE = "material_db.json"
+# 📁 مسیر پوشه و فایل داده‌ها
+DATA_FOLDER = "data"
+DATA_FILE = os.path.join(DATA_FOLDER, "material_db.json")
 
-def is_duplicate(record, all_data):
-    for item in all_data:
-        if (
-            item.get("project_description") == record.get("project_description")
-            and item.get("materials") == record.get("materials")
-        ):
-            return True
-    return False
+# ✅ اگر پوشه وجود نداره، بساز
+os.makedirs(DATA_FOLDER, exist_ok=True)
 
-def process_excel(file, description):
-    if file is None:
-        return "❌ لطفاً یک فایل اکسل بارگذاری کنید.", ""
 
+def extract_materials_from_excel(file):
     try:
-        # ۱. پردازش فایل اکسل
-        materials = parse_excel_file(file.name)
-        record = {
-            "project_description": description.strip(),
-            "materials": materials
-        }
-
-        # ۲. بارگذاری داده‌های قبلی
-        all_data = []
-        if os.path.exists(DATA_FILE):
-            with open(DATA_FILE, "r", encoding="utf-8") as f:
-                try:
-                    existing_data = json.load(f)
-                    if isinstance(existing_data, list):
-                        all_data = existing_data
-                    else:
-                        all_data = [existing_data]
-                except json.JSONDecodeError:
-                    pass  # فایل خالی یا خراب
-
-        # ۳. بررسی تکراری بودن
-        if is_duplicate(record, all_data):
-            return "⚠️ پروژه‌ای با همین توصیف و مواد قبلاً ثبت شده است.", ""
-
-        # ۴. ذخیره در فایل JSON
-        all_data.append(record)
-        save_to_json(all_data, DATA_FILE)
-
-        return f"✅ اطلاعات برای «{len(materials)}» ماده با موفقیت ذخیره شد.\n📝 توصیف پروژه: {description}", json.dumps(record, indent=2, ensure_ascii=False)
-
+        df = pd.read_excel(file.name)
+        materials = []
+        for _, row in df.iterrows():
+            material = row.to_dict()
+            if all(pd.notna(val) for val in material.values()):
+                materials.append(material)
+        return materials
     except Exception as e:
-        return str(e), ""
+        print("خطای خواندن فایل:", e)
+        return []
+
+
+def generate_conditions(materials):
+    return [f"{m['نام کالا']} به مقدار {m['مقدار']}" for m in materials if 'نام کالا' in m and 'مقدار' in m]
+
+
+def is_duplicate(record, data_list):
+    return any(rec["توصیف پروژه"] == record["توصیف پروژه"] and rec["مواد"] == record["مواد"] for rec in data_list)
+
+
+def save_to_json(record):
+    all_data = []
+    if os.path.exists(DATA_FILE):
+        with open(DATA_FILE, "r", encoding="utf-8") as f:
+            try:
+                existing_data = json.load(f)
+                if isinstance(existing_data, list):
+                    all_data = existing_data
+                else:
+                    all_data = [existing_data]
+            except json.JSONDecodeError:
+                print("⚠️ فایل JSON خراب بود. فایل جدید ساخته خواهد شد.")
+                all_data = []
+
+    if is_duplicate(record, all_data):
+        return False, "⚠️ پروژه‌ای با همین توصیف و مواد قبلاً ثبت شده است."
+
+    all_data.append(record)
+    with open(DATA_FILE, "w", encoding="utf-8") as f:
+        json.dump(all_data, f, ensure_ascii=False, indent=2)
+    return True, "✅ پروژه با موفقیت ذخیره شد."
+
+
+def process_excel(file, description):
+    if not file or not description:
+        return "لطفاً فایل و توصیف پروژه را وارد کنید.", ""
+
+    materials = extract_materials_from_excel(file)
+    if not materials:
+        return "خطا در استخراج مواد از فایل. لطفاً فایل اکسل معتبر وارد کنید.", ""
+
+    conditions = generate_conditions(materials)
+    record = {"توصیف پروژه": description, "مواد": materials}
+
+    success, message = save_to_json(record)
+    return message, "\n".join(conditions)
+
 
-with gr.Blocks(title="کتابخانه پروژه‌های برآورد") as demo:
-    gr.Markdown("## 📦 ابزار ساخت کتابخانه مواد پروژه")
+with gr.Blocks(title="نسخه 0.0.2 - ثبت پروژه و استخراج مواد") as demo:
+    gr.Markdown("## 📦 ثبت پروژه بر اساس فایل اکسل")
 
-    with gr.Tabs():
-        with gr.TabItem("📤 ثبت پروژه جدید"):
-            gr.Markdown("در این بخش فایل اکسل خود را بارگذاری کنید و توصیف پروژه را بنویسید.")
-            with gr.Row():
-                excel_input = gr.File(label="فایل اکسل", file_types=[".xlsx"])
-                description_input = gr.Textbox(label="توصیف پروژه", lines=3, placeholder="مثلاً: برآورد نصب چراغ روشنایی با لامپ 70 وات...")
+    with gr.Row():
+        file_input = gr.File(label="فایل اکسل مواد", file_types=[".xlsx"])
+        description_input = gr.Textbox(label="توصیف پروژه", placeholder="مثلاً برآورد تعمیرات شبکه روستایی ...")
 
-            with gr.Row():
-                submit_btn = gr.Button("📥 ذخیره در کتابخانه")
-            
-            output_msg = gr.Textbox(label="وضعیت ذخیره‌سازی", interactive=False)
-            output_json = gr.Code(label="📦 خروجی ذخیره‌شده", language="json")
+    output_message = gr.Textbox(label="وضعیت ذخیره‌سازی", interactive=False)
+    output_conditions = gr.Textbox(label="شرط‌های استخراج شده", lines=6)
 
-            submit_btn.click(fn=process_excel, inputs=[excel_input, description_input], outputs=[output_msg, output_json])
+    process_btn = gr.Button("📥 پردازش و ذخیره")
 
-        with gr.TabItem("📚 مشاهده توضیحات (در نسخه‌های بعدی)"):
-            gr.Markdown("در نسخه‌های آینده، اینجا می‌تونید پروژه‌های ثبت‌شده رو ببینید و تحلیل GPT بگیرید.")
+    process_btn.click(fn=process_excel,
+                      inputs=[file_input, description_input],
+                      outputs=[output_message, output_conditions])
 
 demo.launch()