cleanup

app.py

@@ -303,53 +303,13 @@ def audio_zero_shot_classifier(audio, candidate_labels, model_name):
         result = classifier(audio, candidate_labels=labels)
         output = "Результаты классификации:\n"
         
-        # Обрабатываем разные форматы результатов
-        if isinstance(result, dict):
-            # Формат: {'labels': [...], 'scores': [...]}
-            if 'labels' in result and 'scores' in result:
-                for label, score in zip(result['labels'], result['scores']):
-                    output += f"{label}: {score:.4f}\n"
-            else:
-                # Пытаемся найти любые ключи, которые могут содержать результаты
-                for key, value in result.items():
-                    if isinstance(value, (list, tuple)) and len(value) > 0:
-                        if isinstance(value[0], dict):
-                            for item in value:
-                                if isinstance(item, dict):
-                                    label = item.get('label', item.get('labels', 'unknown'))
-                                    score = item.get('score', item.get('scores', 0.0))
-                                    if isinstance(score, (list, tuple)) and len(score) > 0:
-                                        score = score[0]
-                                    if isinstance(label, (list, tuple)) and len(label) > 0:
-                                        label = label[0]
-                                    output += f"{label}: {score:.4f}\n"
-        elif isinstance(result, list):
-            # Формат: [{'label': '...', 'score': ...}, ...] или список словарей
+        # Парсинг результата: [{'score': ..., 'label': ...}, ...]
+        if isinstance(result, list):
             for item in result:
-                if isinstance(item, dict):
-                    # Проверяем разные возможные форматы
-                    if 'label' in item and 'score' in item:
-                        label = item['label']
-                        score = item['score']
-                    elif 'labels' in item and 'scores' in item:
-                        # Вложенный формат
-                        for label, score in zip(item['labels'], item['scores']):
-                            output += f"{label}: {score:.4f}\n"
-                        continue
-                    else:
-                        # Пытаемся извлечь любым способом
-                        label = item.get('label', item.get('labels', 'unknown'))
-                        score = item.get('score', item.get('scores', 0.0))
-                        if isinstance(score, (list, tuple)) and len(score) > 0:
-                            score = score[0]
-                        if isinstance(label, (list, tuple)) and len(label) > 0:
-                            label = label[0]
-                    output += f"{label}: {score:.4f}\n"
-                else:
-                    # Если элемент не словарь, возможно это вложенная структура
-                    output += f"Элемент: {str(item)}\n"
+                if isinstance(item, dict) and 'label' in item and 'score' in item:
+                    output += f"{item['label']}: {item['score']:.4f}\n"
         else:
-            return f"Ошибка: Неожиданный формат результата от pipeline: {type(result)}. Ожидался словарь с ключами 'labels' и 'scores' или список словарей. Получен: {result}"
+            return f"Ошибка: Неожиданный формат результата от pipeline: {type(result)}. Ожидался список словарей с ключами 'label' и 'score'."
         
         return output
     except Exception as e:
@@ -934,37 +894,21 @@ def image_zero_shot_classification(image, candidate_labels, model_name):
             result = classifier(image, candidate_labels=labels)
             output = "Результаты классификации:\n"
             
-            # Обрабатываем разные форматы результатов
-            if isinstance(result, dict) and 'labels' in result and 'scores' in result:
-                # Формат: {'labels': [...], 'scores': [...]}
-                for label, score in zip(result['labels'], result['scores']):
-                    output += f"{label}: {score:.4f}\n"
-            elif isinstance(result, list):
-                # Формат может быть: [{'label': '...', 'score': ...}, ...] или просто список словарей
-                # Также может быть список, где каждый элемент - это словарь с ключами 'label' и 'score'
+            # Парсим результат
+            if isinstance(result, list):
+                # Результат - ��писок словарей с 'score' и 'label'
                 for item in result:
-                    if isinstance(item, dict):
-                        # Проверяем разные возможные форматы
-                        if 'label' in item and 'score' in item:
-                            label = item['label']
-                            score = item['score']
-                        elif 'labels' in item and 'scores' in item:
-                            # Вложенный формат
-                            for label, score in zip(item['labels'], item['scores']):
-                                output += f"{label}: {score:.4f}\n"
-                            continue
-                        else:
-                            # Пытаемся извлечь любым способом
-                            label = item.get('label', item.get('labels', 'unknown'))
-                            score = item.get('score', item.get('scores', 0.0))
-                            if isinstance(score, (list, tuple)) and len(score) > 0:
-                                score = score[0]
-                            if isinstance(label, (list, tuple)) and len(label) > 0:
-                                label = label[0]
-                        output += f"{label}: {score:.4f}\n"
+                    if isinstance(item, dict) and 'label' in item and 'score' in item:
+                        output += f"{item['label']}: {item['score']:.4f}\n"
                     else:
-                        # Если элемент не словарь, пропускаем или обрабатываем иначе
-                        output += f"Элемент: {str(item)}\n"
+                        return f"Ошибка: Неожиданный формат элемента в результате: {item}"
+            elif isinstance(result, dict):
+                # Результат - словарь с 'labels' и 'scores'
+                if 'labels' in result and 'scores' in result:
+                    for label, score in zip(result['labels'], result['scores']):
+                        output += f"{label}: {score:.4f}\n"
+                else:
+                    return f"Ошибка: Неожиданный формат результата от pipeline: {result}. Ожидался словарь с ключами 'labels' и 'scores' или список словарей."
             else:
                 return f"Ошибка: Неожиданный формат результата от pipeline: {type(result)}. Ожидался словарь с ключами 'labels' и 'scores' или список словарей. Получен: {result}"