Update app.py

app.py

@@ -89,19 +89,21 @@ def classify_toxicity(audio_file, selected_sounds, slider):
     class_score_dict = {class_name: score for class_name, score in zip(class_names, scores)}
     matching_label_score = {}
     exceeding_threshold = []
+    
+    # Iterate through the selected sounds
     for selected_class_name in selected_sounds:
-        print(selected_sounds)
         if selected_class_name in class_score_dict:
-            #score = class_score_dict[selected_class_name]
-            matching_label_score[class_name] = class_score_dict[class_name]
-            for label, score in matching_label_score.items():
-                # Check if the score is greater than the threshold
-                if score > threshold:
-                    # If it exceeds the threshold, append the label and score to the list
-                    exceeding_threshold.append((label, score))
-                    affirm = "Threshold Exceeded"
-                else:
-                    affirm = ""
+            score = class_score_dict[selected_class_name]
+            matching_label_score[selected_class_name] = score
+    
+    #  iterate through matching_label_score to check if any scores exceed the threshold
+    for label, score in matching_label_score.items():
+        if score > threshold:
+            exceeding_threshold.append((label, score))
+            affirm = "Threshold Exceeded"
+        else:
+            affirm = ""
+            
     # miso_label_dict = {label: score for label, score in classify_anxiety[0].items()}
 
     return class_score_dict, affirm