Changed clf_text signature and moved clf output transformation to its own function

app.py

@@ -17,13 +17,26 @@ categories = [
     "Card Payments", "Banking", "Regulations", "Account Payments"
 ]
 
-def clf_text(txt: str):
+def transform_output(res: dict) -> list:
+    return list(
+        sorted(
+            zip(res["labels"], res["scores"]),
+            key=lambda tpl: tpl[1],
+            reverse=True
+        )
+    )
+
+def clf_text(txt: str | list[str]):
     res = classifier(txt, categories, multi_label=True)
-    items = sorted(zip(res["labels"], res["scores"]), key=lambda tpl: tpl[1], reverse=True)
+    if isinstance(res, list):
+        return [ transform_output(dct) for dct in res ]
+    else:
+        return transform_output(res)
+    # items = sorted(zip(res["labels"], res["scores"]), key=lambda tpl: tpl[1], reverse=True)
     # d = dict(zip(res["labels"], res["scores"]))
     # output = [f"{lbl}:\t{score}" for lbl, score in items]
     # return "\n".join(output)
-    return list(items)
+    # return list(items)
 # classifier(sequence_to_classify, candidate_labels)
 #{'labels': ['travel', 'dancing', 'cooking'],
 # 'scores': [0.9938651323318481, 0.0032737774308770895, 0.002861034357920289],