added styling and tags

app/message_processing.py

@@ -258,16 +258,18 @@ def _create_safety_ratings_html(safety_ratings: List[Any]) -> str:
     summary_category = highest_rating.category.name.replace('HARM_CATEGORY_', '').replace('_', ' ').title()
     summary_probability = highest_rating.probability.name
     # Using .7f for score and .8f for severity as per example's precision
-    summary_line = f"{summary_category}: {summary_probability} (Score: {highest_rating.probability_score:.7f}, Severity: {highest_rating.severity_score:.8f})"
+    summary_score_str = f"{highest_rating.probability_score:.7f}" if highest_rating.probability_score is not None else "None"
+    summary_severity_str = f"{highest_rating.severity_score:.8f}" if highest_rating.severity_score is not None else "None"
+    summary_line = f"{summary_category}: {summary_probability} (Score: {summary_score_str}, Severity: {summary_severity_str})"
 
     # Format the list of all ratings for the <pre> block
     ratings_list = []
     for rating in safety_ratings:
         category = rating.category.name.replace('HARM_CATEGORY_', '').replace('_', ' ').title()
         probability = rating.probability.name
-        ratings_list.append(
-            f"{category}: {probability} (Score: {rating.probability_score:.7f}, Severity: {rating.severity_score:.8f})"
-        )
+        score_str = f"{rating.probability_score:.7f}" if rating.probability_score is not None else "None"
+        severity_str = f"{rating.severity_score:.8f}" if rating.severity_score is not None else "None"
+        ratings_list.append(f"{category}: {probability} (Score: {score_str}, Severity: {severity_str})")
     all_ratings_str = '\n'.join(ratings_list)
 
     # CSS Style as specified