Update modules/ai_assistant/ai_app.py

modules/ai_assistant/ai_app.py

@@ -1,3 +1,6 @@
+"""
+تعديل ملف ai_app.py لدمج وحدة التحقق من الصور مباشرة وإصلاح مشكلة الاستيراد
+"""
 import os
 import sys
 import logging
@@ -13,9 +16,77 @@ import requests
 from PIL import Image
 import pandas as pd
 import numpy as np
+import plotly.express as px
 
-# استيراد وحدة التحقق من الصور
-from image_validator import validate_image
+# دمج وحدة التحقق من الصور مباشرة لتجنب مشاكل الاستيراد
+def validate_image(image_path, max_size_mb=10):
+    """
+    التحقق من صحة الصورة وتحسينها قبل إرسالها إلى API
+    
+    المعلمات:
+        image_path: مسار الصورة
+        max_size_mb: الحجم الأقصى المسموح به بالميجابايت
+        
+    العوائد:
+        tuple: (صورة محسنة، نوع الملف، حالة الصلاحية، رسالة الخطأ)
+    """
+    try:
+        # التحقق من وجود الملف
+        if not os.path.exists(image_path):
+            return None, None, False, f"الملف غير موجود: {image_path}"
+            
+        # التحقق من حجم الملف
+        file_size_mb = os.path.getsize(image_path) / (1024 * 1024)
+        if file_size_mb > max_size_mb:
+            return None, None, False, f"حجم الملف كبير جداً: {file_size_mb:.2f} ميجابايت (الحد الأقصى: {max_size_mb} ميجابايت)"
+            
+        # تحديد نوع الملف من امتداده
+        _, ext = os.path.splitext(image_path)
+        ext = ext.lower()
+        
+        if ext in ('.jpg', '.jpeg'):
+            file_type = "image/jpeg"
+        elif ext == '.png':
+            file_type = "image/png"
+        elif ext == '.gif':
+            file_type = "image/gif"
+        elif ext == '.webp':
+            file_type = "image/webp"
+        else:
+            return None, None, False, f"نوع الملف غير مدعوم: {ext}"
+            
+        # فتح الصورة باستخدام PIL للتحقق من صحتها
+        try:
+            with Image.open(image_path) as img:
+                # التحقق من أبعاد الصورة
+                width, height = img.size
+                if width > 4000 or height > 4000:
+                    # تغيير حجم الصورة إذا كانت كبيرة جداً
+                    img = img.resize((min(width, 4000), min(height, 4000)), Image.LANCZOS)
+                    
+                # تحويل الصورة إلى تنسيق مناسب
+                if ext == '.png' and img.mode == 'RGBA':
+                    # تحويل الصور الشفافة إلى RGB
+                    background = Image.new('RGB', img.size, (255, 255, 255))
+                    background.paste(img, mask=img.split()[3])
+                    img = background
+                
+                # حفظ الصورة في ذاكرة مؤقتة
+                buffer = io.BytesIO()
+                img.save(buffer, format=file_type.split('/')[1].upper())
+                buffer.seek(0)
+                
+                # تحويل إلى base64
+                file_content = buffer.getvalue()
+                file_base64 = base64.b64encode(file_content).decode('utf-8')
+                
+                return file_base64, file_type, True, ""
+                
+        except Exception as e:
+            return None, None, False, f"خطأ في معالجة الصورة: {str(e)}"
+            
+    except Exception as e:
+        return None, None, False, f"خطأ في التحقق من الصورة: {str(e)}"
 
 # محاولة استيراد المحللات المختلفة
 try:
@@ -1278,8 +1349,8 @@ def _render_faq_tab(self):
             للحصول على المساعدة، يمكنك:
             1. استخدام المساعد الذكي في تبويب "المساعد الذكي" لطرح أسئلتك
             2. الاطلاع على الأسئلة الشائعة في هذا التبويب
-            3. التواصل مع فريق الدعم الفني عبر البريد الإلكتروني: support@wahbi-ai.com
-            4. زيارة موقع الدعم الفني: https://support.wahbi-ai.com
+            3. التواصل مع فريق الدعم الفني عبر البريد الإلكتروني: tgoahry@sajco.com.sa
+            4. زيارة موقع الدعم الفني: https://sajco.com
             """
         }
     ]