Update app.py

app.py

@@ -2,6 +2,11 @@ import gradio as gr
 import openai
 import os
 from anthropic import Anthropic
+from azure.ai.contentsafety import ContentSafetyClient
+from azure.ai.contentsafety.models import TextCategory
+from azure.core.credentials import AzureKeyCredential
+from azure.core.exceptions import HttpResponseError
+from azure.ai.contentsafety.models import AnalyzeTextOptions
 
 # Load OpenAI and Anthropic API Keys from environment variables
 openai.api_key = os.getenv("openaiapikey")
@@ -12,6 +17,42 @@ client = Anthropic(api_key=anthropic_api_key)
 
 MODEL_NAME = "claude-3-haiku-20240307"
 
+# Function for Azure Content Safety analysis
+def analyze_text_azure(user_text):
+    # Retrieve Azure keys from Hugging Face secrets (as environment variables)
+    key = os.getenv("azurekey")
+    endpoint = os.getenv("azureendpoint")
+
+    # Create Content Safety client
+    client = ContentSafetyClient(endpoint, AzureKeyCredential(key))
+
+    # Construct request
+    request = AnalyzeTextOptions(text=user_text)
+
+    # Analyze text
+    try:
+        response = client.analyze_text(request)
+    except HttpResponseError as e:
+        return f"Error occurred with Azure Content Safety: {e}"
+
+    # Extract moderation results
+    hate_result = next((item for item in response.categories_analysis if item.category == TextCategory.HATE), None)
+    self_harm_result = next((item for item in response.categories_analysis if item.category == TextCategory.SELF_HARM), None)
+    sexual_result = next((item for item in response.categories_analysis if item.category == TextCategory.SEXUAL), None)
+    violence_result = next((item for item in response.categories_analysis if item.category == TextCategory.VIOLENCE), None)
+
+    results = []
+    if hate_result:
+        results.append(f"Hate severity: {hate_result.severity}")
+    if self_harm_result:
+        results.append(f"SelfHarm severity: {self_harm_result.severity}")
+    if sexual_result:
+        results.append(f"Sexual severity: {sexual_result.severity}")
+    if violence_result:
+        results.append(f"Violence severity: {violence_result.severity}")
+
+    return "\n".join(results) if results else "No flagged content detected in Azure Content Safety."
+
 def moderate_text(user_text):
     # OpenAI Moderation
     openai_moderation_result = "Error in OpenAI Moderation"
@@ -27,12 +68,11 @@ def moderate_text(user_text):
     except Exception as e:
         openai_moderation_result = f"Error occurred with OpenAI: {e}"
 
-    # Anthropic Moderation (Using Prompt Template)
+    # Anthropic Moderation
     anthropic_moderation_result = "Error in Anthropic Moderation"
     try:
-        # Create the prompt template for Anthropic's moderation model
         prompt_template = """
-        You are a content moderation expert tasked with categorizing user-generated text based on the following guidelines:
+        You are a content moderation expert tasked with categorizing user-generated text.
 
         Here is the user-generated text to categorize:
         <user_text>{user_text}</user_text>
@@ -50,23 +90,25 @@ def moderate_text(user_text):
             messages=[{"role": "user", "content": prompt}]
         ).content[0].text
 
-        # Format the Anthropic moderation result
         anthropic_moderation_result = f"Anthropic's moderation result: {response}"
 
     except Exception as e:
         anthropic_moderation_result = f"Error occurred with Anthropic: {e}"
 
-    return openai_moderation_result, anthropic_moderation_result
+    # Azure Content Safety Moderation
+    azure_moderation_result = analyze_text_azure(user_text)
+
+    return openai_moderation_result, anthropic_moderation_result, azure_moderation_result
 
 
-# Create the Gradio interface
+# Create the Gradio interface with updated input and output labels
 iface = gr.Interface(
     fn=moderate_text,
-    inputs=gr.Textbox(lines=2, placeholder="Enter text here..."),
-    outputs=["text", "text"],
+    inputs=gr.Textbox(lines=2, placeholder="Please write your text here..."),
+    outputs=[gr.Textbox(label="OpenAI"), gr.Textbox(label="Anthropic"), gr.Textbox(label="Microsoft Azure")],
     title="Content Moderation Tool",
-    description="Enter some text and get the moderation results from OpenAI and Anthropic."
+    description="Enter some text and get the moderation results from OpenAI, Anthropic, and Azure Content Safety."
 )
 
 if __name__ == "__main__":
-    iface.launch()
+    iface.launch()