Update app.py

app.py

@@ -51,8 +51,8 @@ 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 and appropriate response to the following question: {question}"
+    # Create a structured prompt with a unique separator
+    prompt = f"Please provide a detailed and appropriate response to the following question: {question} ###"
     response = query_huggingface(prompt)
     
     generated_text = None
@@ -62,8 +62,10 @@ async def ask(ctx, *, question: str):
         generated_text = response[0]['generated_text']
     
     if generated_text:
-        # Extract only the part after the initial prompt
-        generated_text = generated_text.replace(prompt, "").strip()
+        # Remove the prompt part, using the separator to ensure we only get the response
+        if "###" in generated_text:
+            generated_text = generated_text.split("###")[1].strip()
+        
         if contains_forbidden_content(generated_text):
             await ctx.send("Sorry, the response contains inappropriate content and cannot be displayed.")
         else:
@@ -73,3 +75,4 @@ async def ask(ctx, *, question: str):
 
 # Run the bot
 bot.run(DISCORD_TOKEN)
+