Update app.py

app.py

@@ -9,15 +9,24 @@ DB_PATH = "material_db.json"
 
 class ElectricPoleEstimator:
     def __init__(self):
+        # ایجاد فایل دیتابیس اگر وجود ندارد
+        self._initialize_db()
         self.db = self._load_db()
         self.rule_engine = RuleEngine()
         
+    def _initialize_db(self):
+        """ایجاد فایل دیتابیس اولیه"""
+        if not os.path.exists(DB_PATH):
+            with open(DB_PATH, 'w', encoding='utf-8') as f:
+                json.dump([], f, ensure_ascii=False, indent=2)
+    
     def _load_db(self) -> List[Dict]:
-        """بارگذاری دیتابیس از فایل یا ایجاد جدید"""
-        if os.path.exists(DB_PATH):
+        """بارگذاری دیتابیس از فایل"""
+        try:
             with open(DB_PATH, 'r', encoding='utf-8') as f:
                 return json.load(f)
-        return []
+        except (json.JSONDecodeError, FileNotFoundError):
+            return []
 
     def _save_db(self):
         """ذخیره دیتابیس در فایل"""
@@ -114,10 +123,6 @@ class RuleEngine:
             self._passing_config_rule
         ]
     
-    def add_rule(self, rule_func):
-        """اضافه کردن قانون جدید"""
-        self.rules.append(rule_func)
-    
     def apply_rules(self, context: Dict) -> Dict:
         """اعمال تمام قوانین"""
         overrides = {}
@@ -146,7 +151,7 @@ with gr.Blocks(title="برآورد هوشمند مصالح تاسیسات برق
     
     with gr.Tab("بارگذاری داده‌ها"):
         gr.Markdown("### بارگذاری فایل اکسل جدید")
-        file_input = gr.File(label="فایل اکسل برآورد")
+        file_input = gr.File(label="فایل اکسل برآورد", type="filepath")
         upload_btn = gr.Button("بارگذاری و ذخیره در دیتابیس")
         upload_output = gr.Textbox(label="نتایج بارگذاری")
         upload_btn.click(estimator.add_materials, inputs=file_input, outputs=upload_output)
@@ -169,7 +174,8 @@ with gr.Blocks(title="برآورد هوشمند مصالح تاسیسات برق
         
         results_table = gr.Dataframe(
             headers=["کد", "عنوان", "مقدار", "واحد", "شرایط"],
-            datatype=["str", "str", "number", "str", "str"]
+            datatype=["str", "str", "number", "str", "str"],
+            wrap=True
         )
         
         estimate_btn.click(
@@ -178,14 +184,18 @@ with gr.Blocks(title="برآورد هوشمند مصالح تاسیسات برق
             outputs=results_table
         )
 
-    with gr.Tab("مدیریت قوانین"):
-        gr.Markdown("### قوانین فعلی:")
-        gr.Json(value={
-            "قانون سیم مینک": "اگر هادی مینک باشد، توان پایه 1200 در نظر گرفته می‌شود",
-            "قانون آرایش عبوری": "اگر آرایش عبوری باشد، توان پایه 400 در نظر گرفته می‌شود"
-        })
-        
-        gr.Markdown("### اضافه کردن قانون جدید (در نسخه بعدی)")
+    with gr.Tab("مشاهده دیتابیس"):
+        gr.Markdown("### محتوای فعلی دیتابیس")
+        db_viewer = gr.Dataframe(
+            headers=["کد", "عنوان", "واحد", "ارتفاع", "توان", "هادی"],
+            datatype=["str", "str", "str", "number", "number", "str"],
+            interactive=False
+        )
+        refresh_btn = gr.Button("بروزرسانی نمایش")
+        refresh_btn.click(
+            lambda: pd.DataFrame(estimator.db)[['code', 'title', 'unit', 'pole_height', 'pole_capacity', 'conductor_type']],
+            outputs=db_viewer
+        )
 
 if __name__ == "__main__":
     demo.launch()