Update app.py

app.py

@@ -1,42 +1,36 @@
 import gradio as gr
-from utils import load_material_db, save_material_db, build_db_from_excel, get_items_for_description
+import json
+from utils import parse_excel_file
 
-# بارگذاری دیتابیس موجود (در صورت وجود)
-material_db = load_material_db()
+material_db = {}
 
-def upload_excel_and_generate_db(file, description):
+def handle_file_upload(file, description):
     global material_db
-    material_db = build_db_from_excel(file.name, description)
-    save_material_db(material_db)
-    return f"✅ دیتابیس برای توصیف '{description}' با {len(material_db[description])} کالا ذخیره شد."
-
-def analyze_description(description):
-    items = get_items_for_description(material_db, description)
-    if not items:
-        return "❌ برای این توصیف داده‌ای پیدا نشد. ابتدا فایل مرتبط را آپلود کنید."
+    if file is None or description.strip() == "":
+        return "لطفاً فایل اکسل و توضیح توصیف را وارد کنید."
     
-    output = [f"### 📦 براورد برای: {description}\n"]
-    for item in items:
-        output.append(f"- {item['item_name']} | {item['quantity']} {item['unit']} | کد: {item['item_code']}")
-    return "\n".join(output)
+    try:
+        data = parse_excel_file(file.name)
+        material_db[description.strip()] = data
+
+        with open("material_db.json", "w", encoding="utf-8") as f:
+            json.dump(material_db, f, indent=2, ensure_ascii=False)
 
-demo = gr.Blocks()
+        return f"✅ اطلاعات برای «{description.strip()}» با موفقیت ذخیره شد."
+    except Exception as e:
+        return f"❌ خطا در پردازش فایل: {str(e)}"
 
-with demo:
-    gr.Markdown("## 📊 ساخت پایگاه داده براساس فایل اکسل + نمایش براورد بر اساس توصیف")
-    with gr.Tab("1️⃣ ساخت پایگاه داده"):
-        with gr.Row():
-            file_input = gr.File(label="آپلود فایل اکسل")
-            description_input = gr.Textbox(label="توصیف پروژه", placeholder="مثلاً: نصب کابل زمینی 120")
-        db_output = gr.Textbox(label="وضعیت", interactive=False)
-        generate_btn = gr.Button("ساخت دیتابیس")
-        generate_btn.click(upload_excel_and_generate_db, inputs=[file_input, description_input], outputs=db_output)
+with gr.Blocks(title="ساخت دیتابیس کتابخانه‌ای براورد") as demo:
+    gr.Markdown("### 📘 ساخت دیتابیس براورد بر اساس توصیف و فایل اکسل")
     
-    with gr.Tab("2️⃣ نمایش براورد"):
-        description_query = gr.Textbox(label="توصیف پروژه", placeholder="مثلاً: نصب کابل زمینی 120")
-        result_output = gr.Markdown()
-        search_btn = gr.Button("نمایش کالاها")
-        search_btn.click(analyze_description, inputs=description_query, outputs=result_output)
+    with gr.Row():
+        file_input = gr.File(label="📁 آپلود فایل اکسل", file_types=[".xlsx"])
+        desc_input = gr.Textbox(label="📝 توضیح توصیف (مثلاً: نصب ترانس روی پایه 12/800)")
+    
+    submit_btn = gr.Button("✅ ذخیره اطلاعات")
+    output = gr.Textbox(label="📢 نتیجه عملیات")
+
+    submit_btn.click(fn=handle_file_upload, inputs=[file_input, desc_input], outputs=output)
 
 if __name__ == "__main__":
     demo.launch()