Update utils/helpers.py

utils/helpers.py

@@ -2,331 +2,281 @@
 # -*- coding: utf-8 -*-
 
 """
-وحدة المساعدة المركزية للنظام
-تحتوي على دوال مساعدة مشتركة تستخدم في جميع أنحاء التطبيق
+وحدة المساعدات العامة
+توفر هذه الوحدة مجموعة من الدوال المساعدة المستخدمة في مختلف أجزاء النظام
 """
 
 import os
-import sys
-import streamlit as st
-import pandas as pd
-import numpy as np
+import datetime
 import json
 import re
-import time
-import tempfile
-from datetime import datetime, timedelta
-import random
-import secrets
-import shutil
-import base64
-import logging
-from pathlib import Path
-
-# تكوين التسجيلات
-logging.basicConfig(
-    level=logging.INFO,
-    format='%(asctime)s - %(name)s - %(levelname)s - %(message)s'
-)
-logger = logging.getLogger("wahbi-ai")
-
+import streamlit as st
 
 def create_directory_if_not_exists(directory_path):
-    """إنشاء مسار إذا لم يكن موجوداً"""
-    try:
-        if not os.path.exists(directory_path):
-            os.makedirs(directory_path)
-            logger.info(f"تم إنشاء المجلد: {directory_path}")
+    """إنشاء مجلد إذا لم يكن موجوداً بالفعل"""
+    if not os.path.exists(directory_path):
+        os.makedirs(directory_path)
         return True
-    except Exception as e:
-        logger.error(f"خطأ في إنشاء المجلد {directory_path}: {e}")
-        return False
-
+    return False
 
 def get_data_folder():
-    """الحصول على مسار مجلد البيانات الرئيسي"""
-    # مسار بيانات النظام الرئيسي
-    data_folder = os.path.join(os.path.dirname(os.path.dirname(os.path.abspath(__file__))), "data")
+    """الحصول على مسار مجلد البيانات"""
+    data_folder = os.path.abspath(os.path.join(os.path.dirname(__file__), '..', '..', 'data'))
     create_directory_if_not_exists(data_folder)
     return data_folder
 
-
-def load_config():
-    """تحميل إعدادات التكوين من ملف config.json"""
-    config_path = os.path.join(get_data_folder(), "config.json")
-    
-    # إذا لم يكن ملف التكوين موجوداً، قم بإنشاء ملف افتراضي
-    if not os.path.exists(config_path):
-        default_config = {
-            "system": {
-                "version": "1.0.0",
-                "release_date": "2025-03-30",
-                "company_name": "شركة شبه الجزيرة للمقاولات",
-                "company_logo": "",
-                "language": "ar",
-                "theme": "light",
-                "debug_mode": False
-            },
-            "ai_models": {
-                "default_model": "claude-3-7-sonnet-20250219",
-                "openai_model": "gpt-4o",
-                "huggingface_model": "mistralai/Mistral-7B-Instruct-v0.2"
-            },
-            "notifications": {
-                "enable_email": False,
-                "enable_browser": True,
-                "check_interval": 60
-            }
-        }
-        
-        with open(config_path, 'w', encoding='utf-8') as f:
-            json.dump(default_config, f, ensure_ascii=False, indent=2)
-        
-        return default_config
-    
-    # تحميل ملف التكوين الموجود
-    try:
-        with open(config_path, 'r', encoding='utf-8') as f:
-            return json.load(f)
-    except Exception as e:
-        logger.error(f"خطأ في تحميل ملف التكوين: {e}")
-        return {}
-
-
-def save_config(config):
-    """حفظ إعدادات التكوين إلى ملف config.json"""
-    config_path = os.path.join(get_data_folder(), "config.json")
-    
-    try:
-        with open(config_path, 'w', encoding='utf-8') as f:
-            json.dump(config, f, ensure_ascii=False, indent=2)
-        return True
-    except Exception as e:
-        logger.error(f"خطأ في حفظ ملف التكوين: {e}")
-        return False
-
-
-def format_time(timestamp=None, format_str="%Y-%m-%d %H:%M:%S"):
-    """تنسيق الوقت إلى صيغة معينة"""
+def format_time(timestamp=None):
+    """تنسيق الوقت بصيغة قابلة للقراءة"""
     if timestamp is None:
-        timestamp = datetime.now()
-    elif isinstance(timestamp, (int, float)):
-        timestamp = datetime.fromtimestamp(timestamp)
-    
-    return timestamp.strftime(format_str)
-
+        timestamp = datetime.datetime.now()
+    return timestamp.strftime("%Y-%m-%d %H:%M:%S")
 
 def get_user_info():
     """الحصول على معلومات المستخدم الحالي"""
-    # في التطبيق الفعلي، يمكن استرداد معلومات المستخدم من قاعدة البيانات أو من حالة الجلسة
-    if "user_info" in st.session_state:
-        return st.session_state.user_info
-    
-    # معلومات افتراضية للتطوير
+    # في الوقت الحالي، نستخدم معلومات مستخدم افتراضية
+    # يمكن تعديل هذه الدالة لاحقاً للتكامل مع نظام المصادقة
     return {
         "id": 1,
         "username": "admin",
-        "full_name": "مدير النظام",
-        "email": "admin@example.com",
-        "role": "مدير",
-        "department": "الإدارة",
-        "last_login": format_time()
+        "name": "مدير النظام",
+        "role": "admin"
     }
 
-
-def get_current_project():
-    """الحصول على معلومات المشروع الحالي"""
-    if "current_project" in st.session_state:
-        return st.session_state.current_project
+def load_css():
+    """تحميل أنماط CSS المخصصة"""
+    st.markdown("""
+    <style>
+    .module-title {
+        color: #1E88E5;
+        text-align: center;
+        margin-bottom: 20px;
+        font-weight: bold;
+    }
     
-    # في حالة عدم وجود مشروع محدد
-    return None
-
-
-def load_icons():
-    """تحميل الأيقونات المستخدمة في النظام"""
-    icons_path = os.path.join(os.path.dirname(os.path.dirname(os.path.abspath(__file__))), "assets", "icons")
-    icons = {}
+    .section-title {
+        color: #1565C0;
+        margin-top: 20px;
+        margin-bottom: 10px;
+        font-weight: bold;
+    }
     
-    # التأكد من وجود مجلد الأيقونات
-    if not os.path.exists(icons_path):
-        create_directory_if_not_exists(icons_path)
-        return icons
+    .info-box {
+        background-color: #E3F2FD;
+        padding: 15px;
+        border-radius: 5px;
+        margin-bottom: 15px;
+    }
     
-    # تحميل جميع الأيقونات
-    for icon_file in os.listdir(icons_path):
-        if icon_file.endswith(('.png', '.svg', '.jpg')):
-            icon_name = os.path.splitext(icon_file)[0]
-            icon_path = os.path.join(icons_path, icon_file)
-            
-            try:
-                with open(icon_path, "rb") as f:
-                    icons[icon_name] = base64.b64encode(f.read()).decode()
-            except Exception as e:
-                logger.error(f"خطأ في تحميل الأيقونة {icon_name}: {e}")
+    .warning-box {
+        background-color: #FFF8E1;
+        padding: 15px;
+        border-radius: 5px;
+        margin-bottom: 15px;
+    }
     
-    return icons
-
-
-def get_random_id(length=8):
-    """إنشاء معرف عشوائي بطول محدد"""
-    return secrets.token_hex(length // 2)
-
-
-def compress_text(text, max_length=10000):
-    """اختصار النص إلى حد أقصى محدد مع الحفاظ على المعنى"""
-    if not text or len(text) <= max_length:
-        return text
+    .error-box {
+        background-color: #FFEBEE;
+        padding: 15px;
+        border-radius: 5px;
+        margin-bottom: 15px;
+    }
+    
+    .success-box {
+        background-color: #E8F5E9;
+        padding: 15px;
+        border-radius: 5px;
+        margin-bottom: 15px;
+    }
+    
+    .centered {
+        text-align: center;
+    }
+    
+    .footer {
+        text-align: center;
+        margin-top: 30px;
+        color: #78909C;
+        font-size: 0.8em;
+    }
     
-    # تقسيم النص إلى جمل
-    sentences = re.split(r'(?<=[.!?])\s+', text)
+    /* تحسين تصميم الجداول */
+    table {
+        width: 100%;
+    }
     
-    # حساب متوسط طول الجملة
-    avg_sentence_length = len(text) / len(sentences)
+    thead th {
+        background-color: #1976D2 !important;
+        color: white !important;
+    }
+    
+    tbody tr:nth-child(even) {
+        background-color: #f2f2f2;
+    }
     
-    # حساب عدد الجمل التي يمكن تضمينها
-    num_sentences_to_keep = int(max_length / avg_sentence_length)
+    tbody tr:hover {
+        background-color: #E3F2FD;
+    }
     
-    # الاحتفاظ بالجمل الأولى والأخيرة للحفاظ على السياق
-    keep_first = num_sentences_to_keep // 2
-    keep_last = num_sentences_to_keep - keep_first
+    /* تحسين ظهور الفورم */
+    input, select, textarea {
+        border-radius: 5px !important;
+        border: 1px solid #ddd !important;
+        padding: 10px !important;
+    }
     
-    # دمج الجمل المختارة
-    compressed_text = ' '.join(sentences[:keep_first] + sentences[-keep_last:])
+    .stButton>button {
+        border-radius: 5px !important;
+    }
     
-    # إضافة إشارة إلى أن النص تم اختصاره
-    if len(compressed_text) < len(text):
-        compressed_text += " [...المزيد من النص تم اختصاره...]"
+    /* زيادة حجم الخط للتوافق مع اللغة العربية */
+    html, body, [class*="css"] {
+        font-family: 'Tajawal', sans-serif;
+    }
+    </style>
     
-    return compressed_text
+    <link rel="stylesheet" href="https://fonts.googleapis.com/css2?family=Tajawal:wght@400;700&display=swap">
+    """, unsafe_allow_html=True)
 
+def render_credits():
+    """عرض المعلومات عن حقوق الملكية وإصدار النظام"""
+    st.markdown("""
+    <div class="footer">
+        <p>هذا النظام يعمل لصالح شركة شبه الجزيرة للمقاولات، جميع الحقوق محفوظة 2025</p>
+        <p>نظام WAHBi AI - الإصدار 2.0</p>
+    </div>
+    """, unsafe_allow_html=True)
 
-def str_to_bool(text):
-    """تحويل النص إلى قيمة منطقية"""
-    return text.lower() in ('yes', 'true', 'y', 't', '1', 'نعم', 'صحيح')
+def load_icons():
+    """تحميل الأيقونات المستخدمة في النظام"""
+    icons = {
+        "project": "🏗️",
+        "document": "📄",
+        "analysis": "🔍",
+        "warning": "⚠️",
+        "success": "✅",
+        "error": "❌",
+        "info": "ℹ️",
+        "settings": "⚙️",
+        "user": "👤",
+        "money": "💰",
+        "time": "⏱️",
+        "location": "📍",
+        "notification": "🔔",
+        "edit": "✏️",
+        "delete": "🗑️",
+        "upload": "📤",
+        "download": "📥",
+        "save": "💾",
+        "cancel": "❌",
+        "add": "➕",
+        "calendar": "📅",
+        "chat": "💬",
+        "search": "🔎",
+        "star": "⭐",
+        "trophy": "🏆",
+        "medal": "🥇",
+        "chart": "📊",
+        "map": "🗺️",
+        "building": "🏢",
+        "road": "🛣️",
+        "bridge": "🌉",
+    }
+    return icons
 
+def format_number(number, decimal_places=2):
+    """تنسيق الأرقام بطريقة أنيقة"""
+    if isinstance(number, (int, float)):
+        if decimal_places == 0:
+            return "{:,.0f}".format(number)
+        else:
+            return "{:,.{dp}f}".format(number, dp=decimal_places)
+    return str(number)
 
-def handle_arabic_text(text):
-    """معالجة النص العربي للعرض بشكل صحيح"""
-    if not text:
-        return ""
-    
-    # إضافة علامة RTL لضمان عرض النص العربي بشكل صحيح
-    return f"<div dir='rtl'>{text}</div>"
+def format_currency(amount, currency="ريال", decimal_places=2):
+    """تنسيق المبالغ المالية"""
+    if amount is None:
+        return "غير محدد"
+    formatted = format_number(amount, decimal_places)
+    return f"{formatted} {currency}"
 
+def styled_button(label, key=None, type="primary", on_click=None, args=None, full_width=False, icon=None, is_link=False, help=None):
+    """
+    إنشاء زر بتنسيق معين
+    :param label: نص الزر
+    :param key: مفتاح الزر الفريد
+    :param type: نوع التنسيق ('primary', 'secondary', 'success', 'warning', 'danger', 'info', 'glass', 'flat')
+    :param on_click: الدالة التي سيتم تنفيذها عند النقر
+    :param args: معاملات الدالة
+    :param full_width: هل يأخذ الزر العرض كاملاً
+    :param icon: أيقونة لعرضها قبل النص (emoji أو HTML)
+    :param is_link: إذا كان الزر رابطاً بدلاً من زر عادي
+    :param help: نص المساعدة للزر
+    :return: زر مُنسّق
+    """
+    if is_link:
+        btn_class = f"{type}-btn"
+        if icon:
+            btn_class += " action-btn"
+            label_with_icon = f"{icon} {label}"
+        else:
+            label_with_icon = label
+            
+        button_html = f"""
+        <div class="{btn_class}">
+            {label_with_icon}
+        </div>
+        """
+        return st.markdown(button_html, unsafe_allow_html=True)
+    else:
+        with st.container():
+            btn_class = f"{type}-btn"
+            if icon:
+                btn_class += " action-btn"
+                label_with_icon = f"{icon} {label}"
+            else:
+                label_with_icon = label
+            
+            st.markdown(f'<div class="{btn_class}">', unsafe_allow_html=True)
+            clicked = st.button(label_with_icon, key=key, on_click=on_click, args=args, use_container_width=full_width, help=help)
+            st.markdown('</div>', unsafe_allow_html=True)
+            return clicked
+
+def filter_dataframe(df, column, value):
+    """ترشيح إطار البيانات"""
+    if value == "الكل":
+        return df
+    return df[df[column] == value]
+
+def get_file_extension(filename):
+    """استخراج امتداد الملف"""
+    if not filename:
+        return ""
+    return os.path.splitext(filename)[-1].lower()
 
-def render_credits():
-    """عرض معلومات النظام وحقوق الملكية"""
-    st.markdown("---")
+def extract_numbers_from_text(text):
+    """استخراج الأرقام من النص
     
-    config = load_config()
-    system_info = config.get("system", {})
+    Args:
+        text (str): النص المراد استخراج الأرقام منه
+        
+    Returns:
+        list: قائمة بالأرقام المستخرجة
+    """
+    if not text:
+        return []
     
-    col1, col2, col3 = st.columns([1, 2, 1])
+    # نمط للعثور على الأرقام (صحيحة أو عشرية) في النص
+    pattern = r'[-+]?\d*\.\d+|\d+'
     
-    with col2:
-        st.markdown(
-            f"""
-            <div style="text-align: center; color: #666;">
-                <p>{system_info.get('company_name', 'شركة شبه الجزيرة للمقاولات')}</p>
-                <p>الإصدار: {system_info.get('version', '1.0.0')}</p>
-                <p>© جميع الحقوق محفوظة 2025</p>
-            </div>
-            """,
-            unsafe_allow_html=True
-        )
-
-
-# دالة للحصول على اتصال قاعدة البيانات
-def get_connection():
-    """
-    دالة للحصول على اتصال بقاعدة البيانات
+    # استخراج جميع الأرقام من النص
+    numbers = re.findall(pattern, text)
     
-    ملاحظة: ينبغي أن تكون مستبدلة بالدالة من db_connector.py في البيئة الإنتاجية
-    """
-    try:
-        # استيراد المستوى الفعلي للاتصال بقاعدة البيانات
-        from database.db_connector import get_connection as get_db_connection
-        return get_db_connection()
-    except ImportError:
-        # إذا كان الاتصال بقاعدة البيانات غير متاح
-        logger.warning("لم يتم العثور على وحدة اتصال قاعدة البيانات. استخدام مخزن بيانات مؤقت.")
-        # إرجاع None للإشارة إلى عدم وجود اتصال
-        return None
-
-
-def load_css(file_name=None):
-    """تحميل ملف CSS مخصص"""
-    try:
-        if file_name:
-            css_file = os.path.join(os.path.dirname(os.path.dirname(os.path.abspath(__file__))), "assets", "css", file_name)
-            
-            if os.path.exists(css_file):
-                with open(css_file, "r", encoding="utf-8") as f:
-                    css_content = f.read()
-            else:
-                logger.warning(f"ملف CSS غير موجود: {css_file}")
-                return
+    # تحويل النصوص المستخرجة إلى أرقام (صحيحة أو عشرية)
+    converted_numbers = []
+    for num in numbers:
+        if '.' in num:
+            converted_numbers.append(float(num))
         else:
-            # CSS افتراضي
-            css_content = """
-            .sidebar .sidebar-content {
-                direction: rtl;
-                text-align: right;
-            }
-            div[data-testid="stForm"] {
-                border: 1px solid #ddd;
-                padding: 10px;
-                border-radius: 10px;
-            }
-            .module-title {
-                color: #1E88E5;
-                text-align: center;
-                font-size: 1.8rem;
-                margin-bottom: 1rem;
-            }
-            .instructions {
-                background-color: #f8f9fa;
-                border-right: 3px solid #4CAF50;
-                padding: 10px;
-                margin-bottom: 15px;
-            }
-            .results-container {
-                background-color: #f5f5f5;
-                padding: 15px;
-                border-radius: 5px;
-                margin-top: 20px;
-            }
-            .risk-high {
-                color: #d32f2f;
-                font-weight: bold;
-            }
-            .risk-medium {
-                color: #f57c00;
-                font-weight: bold;
-            }
-            .risk-low {
-                color: #388e3c;
-                font-weight: bold;
-            }
-            .form-container {
-                background-color: #f9f9f9;
-                padding: 20px;
-                border-radius: 10px;
-                margin-bottom: 20px;
-            }
-            .section-header {
-                color: #2196F3;
-                font-size: 1.2rem;
-                font-weight: bold;
-                margin-top: 20px;
-                margin-bottom: 10px;
-                border-bottom: 1px solid #eee;
-                padding-bottom: 5px;
-            }
-            """
-        
-        st.markdown(f"<style>{css_content}</style>", unsafe_allow_html=True)
-        
-    except Exception as e:
-        logger.error(f"خطأ في تحميل ملف CSS: {e}")
+            converted_numbers.append(int(num))
+    
+    return converted_numbers