Update app.py

app.py

@@ -1,38 +1,72 @@
 import gradio as gr
 import openai
 import os
+from anthropic import Anthropic
 
+# Load OpenAI and Anthropic API Keys from environment variables
 openai.api_key = os.getenv("openaiapikey")
+anthropic_api_key = os.getenv("anthropickey")
 
-def moderate_text(text):
-    text = str(text).strip() if text else ""
+# Initialize Anthropic client
+client = Anthropic(api_key=anthropic_api_key)
 
-    if not text:
-        return "Input text is empty. Please enter valid text."
+MODEL_NAME = "claude-3-haiku-20240307"
 
+def moderate_text(user_text):
+    # OpenAI Moderation
+    openai_moderation_result = "Error in OpenAI Moderation"
     try:
-        response = openai.Moderation.create(
-            input=text
-        )
-
+        response = openai.Moderation.create(input=user_text)
         moderation_categories = response["results"][0]["categories"]
         moderation_flagged = response["results"][0]["flagged"]
 
         if moderation_flagged:
-            flagged_categories = [category for category, flagged in moderation_categories.items() if flagged]
-            return f"The text is flagged for moderation due to: {', '.join(flagged_categories)}"
+            openai_moderation_result = f"OpenAI flags the text for the following categories: {', '.join([category for category, flagged in moderation_categories.items() if flagged])}"
         else:
-            return "The text is not flagged for any moderation issues."
+            openai_moderation_result = "The text is not flagged for any moderation issues by OpenAI."
+    except Exception as e:
+        openai_moderation_result = f"Error occurred with OpenAI: {e}"
+
+    # Anthropic Moderation (Using Prompt Template)
+    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:
+
+        Here is the user-generated text to categorize:
+        <user_text>{user_text}</user_text>
+
+        Based on the content, classify this text as either ALLOW or BLOCK. Return nothing else.
+        """
+
+        # Format the prompt with the user text
+        prompt = prompt_template.format(user_text=user_text)
+
+        # Send the prompt to Claude and get the response
+        response = client.messages.create(
+            model=MODEL_NAME,
+            max_tokens=10,
+            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:
-        return f"An error occurred: {e}"
+        anthropic_moderation_result = f"Error occurred with Anthropic: {e}"
+
+    return openai_moderation_result, anthropic_moderation_result
+
 
+# Create the Gradio interface
 iface = gr.Interface(
     fn=moderate_text,
     inputs=gr.Textbox(lines=2, placeholder="Enter text here..."),
-    outputs="text",
-    title="Text Moderation Tool",
-    description="Paste some text, and this tool will use OpenAI's Moderation API to determine if it contains sensitive or flagged content."
+    outputs=["text", "text"],
+    title="Content Moderation Tool",
+    description="Enter some text and get the moderation results from OpenAI and Anthropic."
 )
 
 if __name__ == "__main__":
-    iface.launch()
+    iface.launch()