Update app.py

app.py

@@ -1,36 +1,42 @@
 import gradio as gr
-from utils import load_material_db, filter_items
+from utils import load_material_db, save_material_db, build_db_from_excel, get_items_for_description
 
-# بارگذاری پایگاه داده از فایل json
+# بارگذاری دیتابیس موجود (در صورت وجود)
 material_db = load_material_db()
 
-def analyze_description(description, pole_height, pole_power, conductor_size):
-    # در این نسخه اول، صرفاً براساس ویژگی‌های پایه فیلتر می‌کنیم
-    filtered_items = filter_items(material_db, pole_height, pole_power, conductor_size)
+def upload_excel_and_generate_db(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])} کالا ذخیره شد."
 
-    if not filtered_items:
-        return "هیچ کالایی با این مشخصات یافت نشد."
+def analyze_description(description):
+    items = get_items_for_description(material_db, description)
+    if not items:
+        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)
 
-    # ساخت جدول خروجی
-    output_lines = ["**کالاهای پیشنهادی:**\n"]
-    for item in filtered_items:
-        output_lines.append(f"- 📦 {item['item_name']} | 📏 {item['unit']} | 🔢 {item['quantity']} | کد فهرست بها: `{item['item_code']}`")
+demo = gr.Blocks()
 
-    return "\n".join(output_lines)
-
-# رابط کاربری Gradio
-demo = gr.Interface(
-    fn=analyze_description,
-    inputs=[
-        gr.Textbox(label="توصیف پروژه (برای نسخه‌های بعدی استفاده می‌شود)"),
-        gr.Textbox(label="ارتفاع پایه (مثلاً 12)"),
-        gr.Textbox(label="توان پایه (مثلاً 800)"),
-        gr.Textbox(label="سایز هادی (مثلاً 120)")
-    ],
-    outputs=gr.Markdown(label="نتیجه تحلیل براوردی"),
-    title="تحلیل براورد بر اساس توصیف و شرایط پایه",
-    description="نسخه اولیه برای نمایش کالاهای مرتبط از روی فایل اکسل و شرایط پروژه"
-)
+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.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)
 
 if __name__ == "__main__":
     demo.launch()