Update modules/ai_assistant/ai_app.py

modules/ai_assistant/ai_app.py

@@ -84,61 +84,40 @@ class ClaudeAIService:
             # الحصول على مفتاح API
             api_key = self.get_api_key()
             
-            # قراءة محتوى الصورة مع ضغطها إذا كان حجمها كبيرًا
-            file_size = os.path.getsize(image_path)
-            if file_size > 5 * 1024 * 1024:  # إذا كان الحجم أكبر من 5 ميجابايت
-                # فتح الصورة وضغطها
-                with Image.open(image_path) as img:
-                    # حفظ الصورة المضغوطة في ذاكرة مؤقتة
-                    compressed_img_buffer = io.BytesIO()
-                    
-                    # حساب معامل الجودة المناسب (كلما قل الرقم، قلت الجودة وحجم الملف)
-                    quality = min(95, int((5 * 1024 * 1024 / file_size) * 100))
-                    
-                    # حفظ الصورة بالجودة المحسوبة
-                    img.save(compressed_img_buffer, format='JPEG', quality=quality)
-                    compressed_img_buffer.seek(0)
-                    
-                    
-                    # استخدام البيانات المضغوطة
-                    file_content = compressed_img_buffer.read()
-                    print(f"تم ضغط الصورة من {file_size/1024/1024:.2f} ميجابايت إلى {len(file_content)/1024/1024:.2f} ميجابايت")
-            else:
-                # قراءة محتوى الملف مع ضغط الصور فقط
-                file_size = os.path.getsize(image_path)
-                _, ext = os.path.splitext(image_path)
-                ext = ext.lower()
-                
-            if ext in ['.jpg', '.jpeg', '.png', '.gif', '.webp'] and file_size > 5 * 1024 * 1024:          
-                # فقط ضغط الصور العادية وليس ملفات PDF
-                try:
-                    with Image.open(image_path) as img:
-                        # حفظ الصورة المضغوطة في ذاكرة مؤقتة
-                        compressed_img_buffer = io.BytesIO()
-                        
-                        # حساب معامل الجودة المناسب
-                        quality = min(95, int((5 * 1024 * 1024 / file_size) * 100))
-                        
-                        # حفظ الصورة بالجودة المحسوبة
-                        img.save(compressed_img_buffer, format='JPEG', quality=quality)
-                        compressed_img_buffer.seek(0)
-                        
-                        # استخدام البيانات المضغوطة
-                        file_content = compressed_img_buffer.read()
-                        logging.info(f"تم ضغط الصورة من {file_size/1024/1024:.2f} ميجابايت إلى {len(file_content)/1024/1024:.2f} ميجابايت")
-                        
-                except Exception as e:
-                    logging.error(f"خطأ أثناء ضغط الصورة: {str(e)}")
-                    with open(image_path, 'rb') as f:
-                        file_content = f.read()
-            else:
-                # استخدام الملف الأصلي للملفات غير الصور أو الصور الصغيرة
-                with open(image_path, 'rb') as f:
-                    file_content = f.read()
-                    
-                # التحقق من حجم الملف بعد القراءة
-                if len(file_content) > 5 * 1024 * 1024:
-                    raise ValueError(f"حجم الملف ({len(file_content)/1024/1024:.2f} ميجابايت) يتجاوز الحد الأقصى (5 ميجابايت). يرجى تقليل حجم الملف قبل الرفع.")
+# قراءة محتوى الملف مع ضغط الصور فقط
+file_size = os.path.getsize(image_path)
+_, ext = os.path.splitext(image_path)
+ext = ext.lower()
+
+if ext in ['.jpg', '.jpeg', '.png', '.gif', '.webp'] and file_size > 5 * 1024 * 1024:
+    # فقط ضغط الصور العادية وليس ملفات PDF
+    try:
+        with Image.open(image_path) as img:
+            # حفظ الصورة المضغوطة في ذاكرة مؤقتة
+            compressed_img_buffer = io.BytesIO()
+            
+            # حساب معامل الجودة المناسب
+            quality = min(95, int((5 * 1024 * 1024 / file_size) * 100))
+            
+            # حفظ الصورة بالجودة المحسوبة
+            img.save(compressed_img_buffer, format='JPEG', quality=quality)
+            compressed_img_buffer.seek(0)
+            
+            # استخدام البيانات المضغوطة
+            file_content = compressed_img_buffer.read()
+            logging.info(f"تم ضغط الصورة من {file_size/1024/1024:.2f} ميجابايت إلى {len(file_content)/1024/1024:.2f} ميجابايت")
+    except Exception as e:
+        logging.error(f"خطأ أثناء ضغط الصورة: {str(e)}")
+        with open(image_path, 'rb') as f:
+            file_content = f.read()
+else:
+    # استخدام الملف الأصلي للملفات غير الصور أو الصور الصغيرة
+    with open(image_path, 'rb') as f:
+        file_content = f.read()
+
+    # التحقق من حجم الملف بعد القراءة
+    if len(file_content) > 5 * 1024 * 1024:
+        raise ValueError(f"حجم الملف ({len(file_content)/1024/1024:.2f} ميجابايت) يتجاوز الحد الأقصى (5 ميجابايت). يرجى تقليل حجم الملف قبل الرفع.")