Update modules/pricing/pricing_app.py

modules/pricing/pricing_app.py

@@ -3,11 +3,14 @@ import pandas as pd
 import numpy as np
 import matplotlib.pyplot as plt
 import matplotlib.font_manager as fm
+import arabic_reshaper
+from bidi.algorithm import get_display
 import io
 import base64
 from datetime import datetime
 import random
 import json
+import os
 
 class PricingApp:
     """وحدة التسعير المتكاملة"""
@@ -139,6 +142,10 @@ class PricingApp:
                 'custom_factors': {}         # معاملات مخصصة لبنود محددة
             }
             
+        # تهيئة حالة حفظ التسعير
+        if 'saved_pricing' not in st.session_state:
+            st.session_state.saved_pricing = []
+            
     def render(self):
         """طريقة للتوافق مع الواجهة القديمة"""
         self.run()
@@ -193,6 +200,9 @@ class PricingApp:
                     
                 with tab5:
                     self._render_pricing_strategies(project_info)
+                    
+                # عرض أزرار التصدير والحفظ
+                self._render_export_save_buttons(project_info)
         
     def _render_new_project_form(self):
         """عرض نموذج إنشاء مشروع جديد"""
@@ -335,35 +345,45 @@ class PricingApp:
             
         # إنشاء DataFrame من قائمة المشاريع
         df = pd.DataFrame(st.session_state.projects)
-        df = df[['id', 'name', 'client', 'estimated_value', 'deadline', 'status', 'pricing_type']]
-        df.columns = ['الرقم', 'اسم المشروع', 'العميل', 'القيمة التقديرية', 'الموعد النهائي', 'الحالة', 'نوع التسعير']
-        
-        # تنسيق القيمة التقديرية
-        df['القيمة التقديرية'] = df['القيمة التقديرية'].apply(lambda x: f"{x:,.2f} ريال")
-        
-        # عرض الجدول
-        st.dataframe(df, use_container_width=True)
-        
-        # اختيار المشروع
-        col1, col2 = st.columns(2)
-        with col1:
-            selected_project_id = st.selectbox(
-                "اختر المشروع",
-                options=[p['id'] for p in st.session_state.projects],
-                format_func=lambda x: next((p['name'] for p in st.session_state.projects if p['id'] == x), ""),
-                index=[p['id'] for p in st.session_state.projects].index(st.session_state.current_project) if st.session_state.current_project in [p['id'] for p in st.session_state.projects] else 0,
-                key="select_project"
-            )
+        if len(df) > 0 and 'id' in df.columns:
+            df = df[['id', 'name', 'client', 'estimated_value', 'deadline', 'status', 'pricing_type']]
+            df.columns = ['الرقم', 'اسم المشروع', 'العميل', 'القيمة التقديرية', 'الموعد النهائي', 'الحالة', 'نوع التسعير']
             
-            if selected_project_id != st.session_state.current_project:
-                st.session_state.current_project = selected_project_id
-                st.rerun()
-                
-        with col2:
-            if st.button("تعديل المشروع", key="edit_project_btn"):
-                st.session_state.edit_project_id = st.session_state.current_project
-                st.session_state.show_edit_project_form = True
-                st.rerun()
+            # تنسيق القيمة التقديرية
+            df['القيمة التقديرية'] = df['القيمة التقديرية'].apply(lambda x: f"{x:,.2f} ريال")
+            
+            # عرض الجدول
+            st.dataframe(df, use_container_width=True)
+            
+            # اختيار المشروع
+            col1, col2 = st.columns(2)
+            with col1:
+                project_ids = [p['id'] for p in st.session_state.projects]
+                if st.session_state.current_project in project_ids:
+                    current_index = project_ids.index(st.session_state.current_project)
+                else:
+                    current_index = 0 if project_ids else None
+                    
+                if current_index is not None and project_ids:
+                    selected_project_id = st.selectbox(
+                        "اختر المشروع",
+                        options=project_ids,
+                        format_func=lambda x: next((p['name'] for p in st.session_state.projects if p['id'] == x), ""),
+                        index=current_index,
+                        key="select_project"
+                    )
+                    
+                    if selected_project_id != st.session_state.current_project:
+                        st.session_state.current_project = selected_project_id
+                        st.rerun()
+                        
+            with col2:
+                if st.button("تعديل المشروع", key="edit_project_btn"):
+                    st.session_state.edit_project_id = st.session_state.current_project
+                    st.session_state.show_edit_project_form = True
+                    st.rerun()
+        else:
+            st.info("لا توجد مشاريع. قم بإنشاء مشروع جديد للبدء.")
                 
     def _get_current_project_info(self):
         """الحصول على معلومات المشروع الحالي"""
@@ -440,7 +460,7 @@ class PricingApp:
         # أزرار التصدير والتعديل
         col1, col2 = st.columns(2)
         with col1:
-            if st.button("تصدير", key="export_boq_btn_1"):
+            if st.button("تصدير جدول الكميات", key="export_boq_btn_1"):
                 # إنشاء CSV للتصدير
                 export_df = pd.DataFrame(project_items)
                 export_df = export_df[['code', 'description', 'unit', 'quantity', 'unit_price', 'total_price', 'resource_type']]
@@ -448,21 +468,22 @@ class PricingApp:
                 
                 csv = export_df.to_csv(index=False)
                 b64 = base64.b64encode(csv.encode()).decode()
-                href = f'<a href="data:file/csv;base64,{b64}" download="boq.csv">تحميل CSV</a>'
+                href = f'<a href="data:file/csv;base64,{b64}" download="boq_{st.session_state.current_project}.csv">تحميل CSV</a>'
                 st.markdown(href, unsafe_allow_html=True)
                 
         with col2:
-            selected_item_id = st.selectbox(
-                "اختر بند للتعديل",
-                options=[item['id'] for item in project_items],
-                format_func=lambda x: next((item['description'] for item in project_items if item['id'] == x), ""),
-                key="select_boq_item"
-            )
-            
-            if st.button("تعديل البند", key="edit_boq_item_btn"):
-                st.session_state.edit_boq_item_id = selected_item_id
-                st.session_state.show_edit_boq_item_form = True
-                st.rerun()
+            if len(project_items) > 0:
+                selected_item_id = st.selectbox(
+                    "اختر بند للتعديل",
+                    options=[item['id'] for item in project_items],
+                    format_func=lambda x: next((item['description'] for item in project_items if item['id'] == x), ""),
+                    key="select_boq_item"
+                )
+                
+                if st.button("تعديل البند", key="edit_boq_item_btn"):
+                    st.session_state.edit_boq_item_id = selected_item_id
+                    st.session_state.show_edit_boq_item_form = True
+                    st.rerun()
                 
     def _render_new_boq_item_form(self):
         """عرض نموذج إضافة بند جديد"""
@@ -944,7 +965,9 @@ class PricingApp:
         self._setup_arabic_fonts()
         
         # إنشاء بيانات الرسم البياني
-        labels = ['المواد', 'المعدات', 'العمالة', 'المصاريف العمومية', 'الربح']
+        # استخدام arabic_reshaper و bidi لمعالجة النص العربي
+        labels_ar = ['المواد', 'المعدات', 'العمالة', 'المصاريف العمومية', 'الربح']
+        labels = [get_display(arabic_reshaper.reshape(label)) for label in labels_ar]
         
         # التحقق من وجود قيم NaN واستبدالها بأصفار
         values = [
@@ -974,7 +997,9 @@ class PricingApp:
         plt.setp(autotexts, size=10, weight="bold")
         
         # إضافة العنوان
-        ax.set_title('توزيع تكلفة البند', fontsize=16)
+        title_ar = 'توزيع تكلفة البند'
+        title = get_display(arabic_reshaper.reshape(title_ar))
+        ax.set_title(title, fontsize=16)
         
         # عرض الرسم البياني
         st.pyplot(fig)
@@ -1042,8 +1067,9 @@ class PricingApp:
         fig, ax = plt.subplots(figsize=(10, 6))
         
         # التحقق من وجود قيم NaN واستبدالها بأصفار
-        labels = list(cost_by_resource_type.keys())
-        values = [cost_by_resource_type[label] if not np.isnan(cost_by_resource_type[label]) else 0 for label in labels]
+        labels_ar = list(cost_by_resource_type.keys())
+        labels = [get_display(arabic_reshaper.reshape(label)) for label in labels_ar]
+        values = [cost_by_resource_type[label_ar] if not np.isnan(cost_by_resource_type[label_ar]) else 0 for label_ar in labels_ar]
         
         # التحقق من أن جميع القيم موجبة (أو صفر) لتجنب الأخطاء في الرسم البياني
         values = [max(0, val) for val in values]
@@ -1061,7 +1087,9 @@ class PricingApp:
         plt.setp(autotexts, size=10, weight="bold")
         
         # إضافة العنوان
-        ax.set_title('توزيع التكلفة حسب نوع المورد', fontsize=16)
+        title_ar = 'توزيع التكلفة حسب نوع المورد'
+        title = get_display(arabic_reshaper.reshape(title_ar))
+        ax.set_title(title, fontsize=16)
         
         # عرض الرسم البياني
         st.pyplot(fig)
@@ -1157,11 +1185,14 @@ class PricingApp:
                 p_amount = (total_cost + o_cost) * (p_margin / 100)
                 f_price = total_cost + o_cost + p_amount
                 
+                # حساب نسبة الربح المعدلة
+                adjusted_profit_percentage = p_amount / f_price * 100 if f_price > 0 else 0
+                
                 sensitivity_data.append({
                     "نسبة الربح": p_margin,
                     "نسبة المصاريف العمومية": o_percentage,
                     "السعر النهائي": f_price,
-                    "نسبة الربح المعدلة": p_amount / f_price * 100 if f_price > 0 else 0
+                    "نسبة الربح المعدلة": adjusted_profit_percentage
                 })
                 
         # إنشاء DataFrame من بيانات تحليل الحساسية
@@ -1228,10 +1259,20 @@ class PricingApp:
             )
             
         # إضافة التسميات
-        ax.set_xlabel('نسبة الربح', fontsize=12)
-        ax.set_ylabel('السعر النهائي (ريال)', fontsize=12)
-        ax.set_title('تحليل حساسية الربحية', fontsize=16)
-        ax.legend(title='نسبة المصاريف العمومية')
+        x_label_ar = 'نسبة الربح'
+        y_label_ar = 'السعر النهائي (ريال)'
+        title_ar = 'تحليل حساسية الربحية'
+        legend_title_ar = 'نسبة المصاريف العمومية'
+        
+        x_label = get_display(arabic_reshaper.reshape(x_label_ar))
+        y_label = get_display(arabic_reshaper.reshape(y_label_ar))
+        title = get_display(arabic_reshaper.reshape(title_ar))
+        legend_title = get_display(arabic_reshaper.reshape(legend_title_ar))
+        
+        ax.set_xlabel(x_label, fontsize=12)
+        ax.set_ylabel(y_label, fontsize=12)
+        ax.set_title(title, fontsize=16)
+        ax.legend(title=legend_title)
         ax.grid(True)
         
         # تنسيق المحور Y
@@ -1332,52 +1373,53 @@ class PricingApp:
             st.dataframe(df, use_container_width=True)
             
             # اختيار البند لتعديل معامله
-            selected_item_id = st.selectbox(
-                "اختر البند لتعديل معامله",
-                options=[item['id'] for item in project_items],
-                format_func=lambda x: next((item['description'] for item in project_items if item['id'] == x), ""),
-                key="select_item_for_custom_factor"
-            )
-            
-            # الحصول على البند المحدد
-            selected_item = next((item for item in project_items if item['id'] == selected_item_id), None)
-            
-            if selected_item:
-                # الحصول على المعامل الحالي للبند
-                current_factor = st.session_state.unbalanced_pricing_factors.get('custom_factors', {}).get(str(selected_item_id), 1.0)
-                
-                # عرض شريط تمرير لتعديل معامل البند
-                custom_factor = st.slider(
-                    f"معامل البند: {selected_item['description']}",
-                    min_value=0.5,
-                    max_value=2.0,
-                    value=current_factor,
-                    step=0.01,
-                    key="custom_factor_slider"
+            if len(project_items) > 0:
+                selected_item_id = st.selectbox(
+                    "اختر البند لتعديل معامله",
+                    options=[item['id'] for item in project_items],
+                    format_func=lambda x: next((item['description'] for item in project_items if item['id'] == x), ""),
+                    key="select_item_for_custom_factor"
                 )
                 
-                # تحديث معامل البند
-                if 'custom_factors' not in st.session_state.unbalanced_pricing_factors:
-                    st.session_state.unbalanced_pricing_factors['custom_factors'] = {}
-                    
-                st.session_state.unbalanced_pricing_factors['custom_factors'][str(selected_item_id)] = custom_factor
-                
-                # عرض السعر الأصلي والسعر المعدل
-                original_price = selected_item['unit_price']
-                adjusted_price = original_price * custom_factor
+                # الحصول على البند المحدد
+                selected_item = next((item for item in project_items if item['id'] == selected_item_id), None)
                 
-                col3, col4 = st.columns(2)
-                with col3:
-                    st.metric("السعر الأصلي", f"{original_price:,.2f} ريال")
-                with col4:
-                    st.metric("السعر المعدل", f"{adjusted_price:,.2f} ريال", delta=f"{(custom_factor - 1) * 100:.1f}%")
+                if selected_item:
+                    # الحصول على المعامل الحالي للبند
+                    current_factor = st.session_state.unbalanced_pricing_factors.get('custom_factors', {}).get(str(selected_item_id), 1.0)
                     
-            # عرض زر تطبيق التسعير الغير متزن
-            if st.button("تطبيق التسعير الغير متزن", key="apply_unbalanced_pricing_btn"):
-                # تطبيق التسعير الغير متزن على جميع البنود
-                self._apply_unbalanced_pricing()
-                st.success("تم تطبيق التسعير الغير متزن بنجاح!")
-                st.rerun()
+                    # عرض شريط تمرير لتعديل معامل البند
+                    custom_factor = st.slider(
+                        f"معامل البند: {selected_item['description']}",
+                        min_value=0.5,
+                        max_value=2.0,
+                        value=current_factor,
+                        step=0.01,
+                        key="custom_factor_slider"
+                    )
+                    
+                    # تحديث معامل البند
+                    if 'custom_factors' not in st.session_state.unbalanced_pricing_factors:
+                        st.session_state.unbalanced_pricing_factors['custom_factors'] = {}
+                        
+                    st.session_state.unbalanced_pricing_factors['custom_factors'][str(selected_item_id)] = custom_factor
+                    
+                    # عرض السعر الأصلي والسعر المعدل
+                    original_price = selected_item['unit_price']
+                    adjusted_price = original_price * custom_factor
+                    
+                    col3, col4 = st.columns(2)
+                    with col3:
+                        st.metric("السعر الأصلي", f"{original_price:,.2f} ريال")
+                    with col4:
+                        st.metric("السعر المعدل", f"{adjusted_price:,.2f} ريال", delta=f"{(custom_factor - 1) * 100:.1f}%")
+                        
+                # عرض زر تطبيق التسعير الغير متزن
+                if st.button("تطبيق التسعير الغير متزن", key="apply_unbalanced_pricing_btn"):
+                    # تطبيق التسعير الغير متزن على جميع البنود
+                    self._apply_unbalanced_pricing()
+                    st.success("تم تطبيق التسعير الغير متزن بنجاح!")
+                    st.rerun()
                 
         elif pricing_strategy == "تسعير موجه ربحية":
             st.info("""
@@ -1460,16 +1502,25 @@ class PricingApp:
         fig, ax = plt.subplots(figsize=(10, 6))
         
         # إنشاء بيانات الرسم البياني
-        strategies = comparison_data['استراتيجية التسعير']
+        strategies_ar = comparison_data['استراتيجية التسعير']
+        strategies = [get_display(arabic_reshaper.reshape(strategy)) for strategy in strategies_ar]
         prices = [standard_price, unbalanced_price, profit_oriented_price, competitive_price, strategic_price]
         
         # رسم الأعمدة
         bars = ax.bar(strategies, prices, color=['blue', 'green', 'red', 'purple', 'orange'])
         
         # إضافة التسميات
-        ax.set_xlabel('استراتيجية التسعير', fontsize=12)
-        ax.set_ylabel('السعر النهائي (ريال)', fontsize=12)
-        ax.set_title('مقارنة بين استراتيجيات التسعير', fontsize=16)
+        x_label_ar = 'استراتيجية التسعير'
+        y_label_ar = 'السعر النهائي (ريال)'
+        title_ar = 'مقارنة بين استراتيجيات التسعير'
+        
+        x_label = get_display(arabic_reshaper.reshape(x_label_ar))
+        y_label = get_display(arabic_reshaper.reshape(y_label_ar))
+        title = get_display(arabic_reshaper.reshape(title_ar))
+        
+        ax.set_xlabel(x_label, fontsize=12)
+        ax.set_ylabel(y_label, fontsize=12)
+        ax.set_title(title, fontsize=16)
         
         # تنسيق المحور Y
         ax.get_yaxis().set_major_formatter(
@@ -1557,3 +1608,156 @@ class PricingApp:
                 if item['project_id'] == st.session_state.current_project:
                     st.session_state.boq_items[i]['unit_price'] *= adjustment_factor
                     st.session_state.boq_items[i]['total_price'] = item['quantity'] * st.session_state.boq_items[i]['unit_price']
+                    
+    def _render_export_save_buttons(self, project_info):
+        """عرض أزرار التصدير والحفظ"""
+        st.subheader("تصدير وحفظ التسعير")
+        
+        col1, col2 = st.columns(2)
+        
+        with col1:
+            if st.button("💾 حفظ التسعير الحالي", key="save_pricing_btn", type="primary"):
+                self._save_current_pricing(project_info)
+                st.success("تم حفظ التسعير بنجاح!")
+                
+        with col2:
+            if st.button("📤 تصدير التسعير الكامل", key="export_full_pricing_btn"):
+                self._export_full_pricing(project_info)
+                
+        # عرض التسعيرات المحفوظة
+        if st.session_state.saved_pricing:
+            st.subheader("التسعيرات المحفوظة")
+            
+            # إنشاء DataFrame من التسعيرات المحفوظة
+            saved_pricing_data = []
+            for pricing in st.session_state.saved_pricing:
+                saved_pricing_data.append({
+                    'المشروع': pricing['project_name'],
+                    'تاريخ الحفظ': pricing['save_date'],
+                    'إجمالي التكلفة': f"{pricing['total_cost']:,.2f} ريال",
+                    'نوع التسعير': pricing['pricing_type'],
+                    'الملاحظات': pricing['notes'] if 'notes' in pricing else ""
+                })
+                
+            df = pd.DataFrame(saved_pricing_data)
+            st.dataframe(df, use_container_width=True)
+            
+            # اختيار تسعير محفوظ للتحميل
+            if len(st.session_state.saved_pricing) > 0:
+                selected_pricing_index = st.selectbox(
+                    "اختر تسعير محفوظ للتحميل",
+                    options=list(range(len(st.session_state.saved_pricing))),
+                    format_func=lambda x: f"{st.session_state.saved_pricing[x]['project_name']} - {st.session_state.saved_pricing[x]['save_date']}",
+                    key="select_saved_pricing"
+                )
+                
+                if st.button("تحميل التسعير المحفوظ", key="load_saved_pricing_btn"):
+                    self._load_saved_pricing(selected_pricing_index)
+                    st.success("تم تحميل التسعير المحفوظ بنجاح!")
+                    st.rerun()
+                    
+    def _save_current_pricing(self, project_info):
+        """حفظ التسعير الحالي"""
+        # الحصول على بنود المشروع الحالي
+        project_items = [item for item in st.session_state.boq_items if item['project_id'] == st.session_state.current_project]
+        
+        if not project_items:
+            st.warning("لا توجد بنود في جدول الكميات. لا يمكن حفظ التسعير.")
+            return
+            
+        # حساب إجمالي تكلفة المشروع
+        total_cost = sum(item['total_price'] for item in project_items)
+        
+        # إنشاء كائن التسعير المحفوظ
+        saved_pricing = {
+            'project_id': project_info['id'],
+            'project_name': project_info['name'],
+            'save_date': datetime.now().strftime("%Y-%m-%d %H:%M:%S"),
+            'pricing_type': project_info.get('pricing_type', 'قياسي'),
+            'total_cost': total_cost,
+            'items': project_items,
+            'unbalanced_pricing_factors': st.session_state.unbalanced_pricing_factors.copy() if 'unbalanced_pricing_factors' in st.session_state else {}
+        }
+        
+        # إضافة التسعير المحفوظ إلى قائمة التسعيرات المحفوظة
+        if 'saved_pricing' not in st.session_state:
+            st.session_state.saved_pricing = []
+            
+        st.session_state.saved_pricing.append(saved_pricing)
+        
+    def _load_saved_pricing(self, index):
+        """تحميل تسعير محفوظ"""
+        if index < 0 or index >= len(st.session_state.saved_pricing):
+            st.error("مؤشر التسعير المحفوظ غير صالح.")
+            return
+            
+        saved_pricing = st.session_state.saved_pricing[index]
+        
+        # تحديث المشروع الحالي
+        st.session_state.current_project = saved_pricing['project_id']
+        
+        # تحديث بنود المشروع
+        for item in saved_pricing['items']:
+            # البحث عن البند في القائمة الحالية
+            existing_item = next((i for i in st.session_state.boq_items if i['id'] == item['id']), None)
+            
+            if existing_item:
+                # تحديث البند الموجود
+                for i, boq_item in enumerate(st.session_state.boq_items):
+                    if boq_item['id'] == item['id']:
+                        st.session_state.boq_items[i] = item
+                        break
+            else:
+                # إضافة البند الجديد
+                st.session_state.boq_items.append(item)
+                
+        # تحديث معاملات التسعير الغير متزن
+        if 'unbalanced_pricing_factors' in saved_pricing:
+            st.session_state.unbalanced_pricing_factors = saved_pricing['unbalanced_pricing_factors']
+            
+    def _export_full_pricing(self, project_info):
+        """تصدير التسعير الكامل"""
+        # الحصول على بنود المشروع الحالي
+        project_items = [item for item in st.session_state.boq_items if item['project_id'] == st.session_state.current_project]
+        
+        if not project_items:
+            st.warning("لا توجد بنود في جدول الكميات. لا يمكن تصدير التسعير.")
+            return
+            
+        # حساب إجمالي تكلفة المشروع
+        total_cost = sum(item['total_price'] for item in project_items)
+        
+        # إنشاء DataFrame من بنود المشروع
+        df = pd.DataFrame(project_items)
+        df = df[['code', 'description', 'unit', 'quantity', 'unit_price', 'total_price', 'resource_type']]
+        df.columns = ['الكود', 'الوصف', 'الوحدة', 'الكمية', 'سعر الوحدة', 'السعر الإجمالي', 'نوع المورد']
+        
+        # تنسيق الأسعار
+        df['سعر الوحدة'] = df['سعر الوحدة'].apply(lambda x: f"{x:,.2f}")
+        df['السعر الإجمالي'] = df['السعر الإجمالي'].apply(lambda x: f"{x:,.2f}")
+        
+        # إضافة صف المجموع
+        df.loc[len(df)] = ['', 'الإجمالي', '', '', '', f"{total_cost:,.2f}", '']
+        
+        # إنشاء CSV للتصدير
+        csv = df.to_csv(index=False)
+        b64 = base64.b64encode(csv.encode()).decode()
+        
+        # إنشاء رابط التحميل
+        filename = f"pricing_{project_info['id']}_{datetime.now().strftime('%Y%m%d_%H%M%S')}.csv"
+        href = f'<a href="data:file/csv;base64,{b64}" download="{filename}">تحميل ملف التسعير الكامل (CSV)</a>'
+        st.markdown(href, unsafe_allow_html=True)
+        
+        # إنشاء Excel للتصدير
+        excel_buffer = io.BytesIO()
+        df.to_excel(excel_buffer, index=False, engine='openpyxl')
+        excel_buffer.seek(0)
+        b64_excel = base64.b64encode(excel_buffer.read()).decode()
+        
+        # إنشاء رابط التحميل
+        filename_excel = f"pricing_{project_info['id']}_{datetime.now().strftime('%Y%m%d_%H%M%S')}.xlsx"
+        href_excel = f'<a href="data:application/vnd.openxmlformats-officedocument.spreadsheetml.sheet;base64,{b64_excel}" download="{filename_excel}">تحميل ملف التسعير الكامل (Excel)</a>'
+        st.markdown(href_excel, unsafe_allow_html=True)
+        
+        # إنشاء تقرير PDF
+        st.info("لتصدير التسعير بتنسيق PDF، يرجى استخدام خيار 'طباعة الصفحة' في المتصفح واختيار 'حفظ كـ PDF'.")