rick commited on
Commit
baecff6
·
unverified ·
1 Parent(s): edf2ac7

minors changes

Browse files
Files changed (1) hide show
  1. core/moderation.py +26 -4
core/moderation.py CHANGED
@@ -31,7 +31,29 @@ def moderate_text(input_text: str,
31
  print(f"An error occurred: {e}")
32
  return None
33
 
34
-
35
-
36
-
37
-
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
31
  print(f"An error occurred: {e}")
32
  return None
33
 
34
+ def api_moderation_openai_text(text_to_moderate: str) -> Dict[str, Any]:
35
+ response = moderate_text(text_to_moderate)
36
+ if response:
37
+ moderation_result = json.loads(response.to_json())
38
+
39
+ result_dict = {
40
+ "flagged": False,
41
+ "results": []
42
+ }
43
+
44
+ for result in moderation_result["results"]:
45
+ result_dict["flagged"] = result_dict["flagged"] or result["flagged"]
46
+ result_dict["results"].append({
47
+ "categories": result["categories"],
48
+ "category_scores": result["category_scores"],
49
+ "category_applied_input_types": result["category_applied_input_types"]
50
+ })
51
+
52
+ return result_dict
53
+ else:
54
+ return {"error": "La modération a échoué"}
55
+
56
+ if __name__=='__main__':
57
+ text_to_moderate = "Ceci est un exemple de texte à modérer."
58
+ result = api_moderation_openai_text(text_to_moderate)
59
+ print("Résultat de la modération :", result)