Update app.py

app.py

@@ -5,11 +5,14 @@ import json
 
 JSON_PATH = "/tmp/material_db.json"
 
-def parse_excel(file):
+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 None, None, f"❌ خطا در خواندن فایل اکسل: {str(e)}"
+        return "", "", f"❌ خطا در خواندن فایل اکسل: {str(e)}"
 
     column_mapping = {
         "code": ["کد", "کد فهرست بها"],
@@ -22,7 +25,7 @@ def parse_excel(file):
     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}» در فایل پیدا نشد."
+            return "", "", f"❌ ستون «{key}» در فایل پیدا نشد."
         resolved[key] = found
 
     materials = []
@@ -51,74 +54,80 @@ def parse_excel(file):
         else:
             existing = []
 
-        new_materials = [m for m in materials if m not in existing]
-        existing.extend(new_materials)
+        new_record = {
+            "project_description": description.strip(),
+            "items": materials
+        }
+        existing.append(new_record)
 
         with open(JSON_PATH, "w", encoding="utf-8") as f:
             json.dump(existing, f, indent=2, ensure_ascii=False)
 
     except Exception as e:
-        return None, None, f"❌ خطا در ذخیره فایل JSON: {str(e)}"
+        return "", "", f"❌ خطا در ذخیره فایل JSON: {str(e)}"
+
+    return "\n".join(constraints), f"✅ {len(materials)} قلم ثبت شد.", JSON_PATH
+
+def download_json():
+    return JSON_PATH if os.path.exists(JSON_PATH) else None
 
-    return constraints, len(new_materials), f"✅ {len(new_materials)} مورد جدید با موفقیت ثبت شدند."
+def load_json_file(json_file):
+    if json_file is None:
+        return "⚠️ لطفاً یک فایل JSON معتبر انتخاب کنید."
 
-def load_json_file(file):
     try:
-        with open(file.name, "r", encoding="utf-8") as f:
-            data = json.load(f)
-        with open(JSON_PATH, "w", encoding="utf-8") as f_out:
-            json.dump(data, f_out, indent=2, ensure_ascii=False)
-        return f"✅ فایل JSON با {len(data)} مورد با موفقیت بارگذاری شد."
+        with open(json_file.name, "r", encoding="utf-8") as f:
+            content = json.load(f)
+        with open(JSON_PATH, "w", encoding="utf-8") as f2:
+            json.dump(content, f2, indent=2, ensure_ascii=False)
+        return f"✅ فایل JSON بارگذاری و جایگزین شد."
     except Exception as e:
-        return f"❌ خطا در بارگذاری فایل JSON: {str(e)}"
+        return f"❌ خطا در خواندن یا ذخیره فایل JSON: {str(e)}"
 
-def read_json_descriptions():
+def show_all_projects():
     if not os.path.exists(JSON_PATH):
-        return "ℹ️ تاکنون هیچ موردی ثبت نشده است."
+        return "⚠️ فایل JSON موجود نیست."
+
     try:
         with open(JSON_PATH, "r", encoding="utf-8") as f:
-            data = json.load(f)
-        descriptions = [f"{item['description']} ({item['quantity']} {item['unit']})" for item in data]
-        return "\n".join(descriptions)
+            content = json.load(f)
+        result = ""
+        for i, project in enumerate(content):
+            result += f"\n📁 پروژه {i+1}: {project['project_description']}\n"
+            for item in project["items"]:
+                result += f"- {item['description']} ({item['quantity']} {item['unit']})\n"
+        return result.strip()
     except Exception as e:
         return f"❌ خطا در خواندن فایل JSON: {str(e)}"
 
 with gr.Blocks(title="مدیریت مواد پروژه - نسخه 0.1.1") as demo:
-    with gr.Tabs():
-        with gr.TabItem("📤 بارگذاری فایل اکسل"):
-            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)
-
-            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
-
-            upload_btn.click(handle_upload, inputs=[file_input], outputs=[status_output, constraints_output])
-
-        with gr.TabItem("📂 بارگذاری فایل JSON"):
-            gr.Markdown("## بارگذاری اولیه اطلاعات ذخیره‌شده")
-
-            json_input = gr.File(label="فایل JSON")
-            load_btn = gr.Button("📥 بارگذاری")
-
-            load_status = gr.Textbox(label="نتیجه", lines=2)
-            load_btn.click(load_json_file, inputs=[json_input], outputs=[load_status])
-
-        with gr.TabItem("📑 نمایش اقلام ذخیره‌شده"):
-            gr.Markdown("## فهرست اقلام ثبت‌شده در فایل JSON")
-            refresh_btn = gr.Button("🔄 بازخوانی")
-            json_display = gr.Textbox(label="موارد موجود", lines=20)
-
-            refresh_btn.click(read_json_descriptions, outputs=[json_display])
+    with gr.Tab("📤 بارگذاری اکسل و ثبت پروژه"):
+        gr.Markdown("## آپلود فایل اکسل و وارد کردن توضیح پروژه")
+
+        with gr.Row():
+            file_input = gr.File(label="آپلود فایل اکسل")
+            project_desc = gr.Textbox(label="📝 توضیح پروژه")
+
+        upload_btn = gr.Button("📤 بارگذاری و ثبت")
+        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 = gr.Button("📥 دانلود فایل JSON")
+
+        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)
+
+    with gr.Tab("📚 پروژه‌های ثبت‌شده و بارگذاری JSON"):
+        gr.Markdown("## مشاهده پروژه‌های ذخیره‌شده در فایل JSON")
+        show_btn = gr.Button("📂 نمایش پروژه‌های ذخیره‌شده")
+        json_content_output = gr.Textbox(label="📑 محتوای فایل JSON", lines=20, interactive=False)
+        show_btn.click(show_all_projects, outputs=json_content_output)
+
+        gr.Markdown("## بارگذاری فایل JSON از سیستم شما")
+        json_input_file = gr.File(label="📥 انتخاب فایل JSON جدید")
+        load_json_btn = gr.Button("🔄 بارگذاری فایل JSON جدید")
+        json_load_status = gr.Textbox(label="وضعیت بارگذاری فایل JSON")
+        load_json_btn.click(load_json_file, inputs=json_input_file, outputs=json_load_status)
 
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch()