Update modules/document_comparison/document_comparator.py

modules/document_comparison/document_comparator.py

@@ -38,8 +38,24 @@ class DocumentComparator:
         self.comparison_dir = os.path.join(os.path.dirname(__file__), '..', '..', 'data', 'document_comparison')
         create_directory_if_not_exists(self.comparison_dir)
         
+        # تهيئة NLTK وتنزيل حزمة punkt إذا لم تكن موجودة
+        self._initialize_nltk()
+        
         # إعداد مقيم ROUGE لمقارنة النصوص
         self.rouge_scorer = rouge_scorer.RougeScorer(['rouge1', 'rouge2', 'rougeL'], use_stemmer=False)
+        
+    def _initialize_nltk(self):
+        """تهيئة مكتبة NLTK وتنزيل الحزم المطلوبة"""
+        try:
+            # محاولة استخدام sent_tokenize للتحقق من وجود حزمة punkt
+            from nltk.tokenize import sent_tokenize
+            sent_tokenize("This is a test sentence.")
+        except LookupError:
+            # تنزيل حزمة punkt تلقائيًا إذا لم تكن موجودة
+            import nltk
+            nltk.download('punkt', quiet=True)
+            # طباعة رسالة تأكيد التنزيل
+            st.info("تم تنزيل حزمة NLTK punkt بنجاح للاستخدام في مقارنة المستندات.")
     
     def _preprocess_text(self, text):
         """معالجة النص قبل التحليل"""