Update app.py

app.py

@@ -451,15 +451,19 @@ def classify_with_bart(text, candidate_labels):
     result = classifier(text, candidate_labels=candidate_labels, multi_label=True)
     top_scores = list(zip(result['labels'][:3], result['scores'][:3]))
     return {lbl: round(score * 100, 2) for lbl, score in top_scores}
-
+def safe_round(value, digits=2):
+    try:
+        return round(float(value), digits)
+    except (TypeError, ValueError):
+        return 0.0
 def classify_with_phi(text, label_info):
     prompt = f'Classify the below text of document: {text} Given Document types: {label_info}'
     response: RunResponse = phi_agent.run(prompt)
     result = response.content
     return {
-        result.doc_type_1: round(result.confidence_1 , 2),
-        result.doc_type_2: round(result.confidence_2 , 2),
-        result.doc_type_3: round(result.confidence_3 , 2)
+        result.doc_type_1: safe_round(result.confidence_1),
+        result.doc_type_2: safe_round(result.confidence_2),
+        result.doc_type_3: safe_round(result.confidence_3),
     }
 
 def process_file(input_file, classifier_choice):