Update app.py

app.py

@@ -15,6 +15,9 @@ HF_API_KEY = os.getenv('HFREAD')
 API_URL = "https://api-inference.huggingface.co/models/mistralai/Mistral-7B-Instruct-v0.3"
 headers = {"Authorization": f"Bearer {HF_API_KEY}"}
 
+# List of forbidden words/phrases
+forbidden_words = ["badword1", "badword2", "inappropriate_phrase"]  # Extend this list as needed
+
 # Function to query the Hugging Face model with a structured prompt
 def query_huggingface(prompt):
     try:
@@ -25,6 +28,13 @@ def query_huggingface(prompt):
         print(f"Error querying the API: {e}")
         return {"error": str(e)}
 
+# Function to check if the response contains forbidden words/phrases
+def contains_forbidden_content(text):
+    for word in forbidden_words:
+        if word in text.lower():
+            return True
+    return False
+
 # Initialize the Discord bot
 intents = discord.Intents.default()
 intents.messages = True
@@ -42,12 +52,22 @@ async def ask(ctx, *, question: str):
     Command to ask a question to the Hugging Face model with an instructive prompt.
     """
     # Create a structured prompt
-    prompt = f"From this point forward, I require that all responses adhere to strict content guidelines. Please avoid any content that is explicit, offensive, harmful, or inappropriate. This includes, but is not limited to, content that is violent, sexually explicit, discriminatory, or offensive to any group or individual. Ensure that all information is presented in a respectful and professional manner. If a topic is deemed sensitive or inappropriate, please redirect the conversation to a more suitable subject. Please provide a detailed response to the following question: {question}"
+    prompt = f"From this point forward, I require that all responses adhere to strict content guidelines. Please avoid any content that is explicit, offensive, harmful, or inappropriate. This includes, but is not limited to, content that is violent, sexually explicit, discriminatory, or offensive to any group or individual. Ensure that all information is presented in a respectful and professional manner. If a topic is deemed sensitive or inappropriate, please redirect the conversation to a more suitable subject. Please provide a detailed and appropriate response to the following question: {question}"
     response = query_huggingface(prompt)
-    if 'generated_text' in response:
-        await ctx.send(f"Response: {response['generated_text']}")
+    
+    generated_text = None
+    if isinstance(response, dict) and 'generated_text' in response:
+        generated_text = response['generated_text']
     elif isinstance(response, list) and len(response) > 0 and 'generated_text' in response[0]:
-        await ctx.send(f"Response: {response[0]['generated_text']}")
+        generated_text = response[0]['generated_text']
+    
+    if generated_text:
+        # Extract only the part after the initial prompt
+        generated_text = generated_text.replace(prompt, "").strip()
+        if contains_forbidden_content(generated_text):
+            await ctx.send("Sorry, the response contains inappropriate content and cannot be displayed.")
+        else:
+            await ctx.send(generated_text)
     else:
         await ctx.send("Sorry, I couldn't generate a response.")