Update app.py

app.py

@@ -844,18 +844,13 @@ def _censor_me(self, p, safer=0.0005):
   resp_dict = resp_orig.model_dump()
   #
   v1 = resp_dict["results"][0]["category_scores"]
+  v1 = {key: value if value is not None else 0 for key, value in v1.items()}
   print(f'resp_dic: {resp_dict}')
   print(f'v1: {v1}')
-  try:
-    keym = vi.get
-  except Exception as e:
-    keym = 0
-  try:
-    max_key = max(v1, keym)
-  except Exception as e:
-    max_key = 0
-
-  sum_value = sum(value for value in v1.values() if value is not None)
+
+  max_key = max(v1, key=v1.get)
+  max_value = v1[max_key]
+  sum_value = sum(v1.values())
   #
   v1["is_safer_flagged"] = False
   if (max_value >= safer):