Spaces:
Paused
Paused
rick
commited on
minors changes
Browse files- 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)
|