Update func_ai.py

func_ai.py

@@ -58,20 +58,49 @@ def retrieve_from_vdb(query):
         print(f"Ошибка при поиске: {response.text}")
         return []
 
+from deep_translator import GoogleTranslator
+import time
+
 def analyze_sentiment(comments):
     print("Начинаем анализ настроений.")
     results = []
+    
     for i in range(0, len(comments), 50):
         batch = comments[i:i + 50]
         print(f"Анализируем батч с {i} по {i + len(batch)} комментарий: {batch}")
         
-        # Translate comments into English before sentiment analysis
-        translated_batch = [GoogleTranslator(source='auto', target="en").translate(comment) for comment in batch]
+        translated_batch = []
+        valid_comments = []  # Список комментариев с успешным переводом
+        
+        for comment in batch:
+            try:
+                translated = GoogleTranslator(source='auto', target="en").translate(comment)
+                if translated:
+                    translated_batch.append(translated)
+                    valid_comments.append(comment)
+                else:
+                    print(f"Перевод комментария не удался или вернул пустую строку: {comment}")
+            except Exception as e:
+                print(f"Ошибка при переводе комментария '{comment}': {e}")
+        
+        if not translated_batch:
+            print("Все переводы в этом батче не удались. Пропускаем этот батч.")
+            continue  # Переходим к следующему батчу, если перевод не удался
+        
         print(f"Переведённый батч: {translated_batch}")
         
-        batch_results = sentiment_model(translated_batch)
-        print(f"Результаты батча: {batch_results}")
-        results.extend(batch_results)
-        time.sleep(1)  # Задержка для предотвращения перегрузки
+        try:
+            batch_results = sentiment_model(translated_batch)
+            print(f"Результаты батча: {batch_results}")
+            results.extend(batch_results)
+        except Exception as e:
+            print(f"Ошибка при анализе настроений: {e}")
+            # Добавляем результат с неопределенным настроением для каждого переведенного комментария
+            for _ in translated_batch:
+                results.append({'label': 'UNKNOWN', 'score': 0.0})
+        
+        time.sleep(1)  # Задержка для предотвращения перегрузки API
+    
     print(f"Анализ настроений завершен. Общие результаты: {results}")
     return results
+