Update app.py

app.py

@@ -32,7 +32,16 @@ def verificar_politica(message):
             stream=False
         )
         answer = response['choices'][0]['message']['content']
-        if answer.startswith("Yes"):
+        
+        # Calcula las probabilidades de "Yes" y "No" basándote en la respuesta generada
+        yes_probability = answer.lower().count("yes") / (answer.lower().count("yes") + answer.lower().count("no"))
+        no_probability = 1 - yes_probability
+
+        print(f"Policy: {policy}")
+        print(f"Yes probability: {yes_probability}")
+        print(f"No probability: {no_probability}")
+
+        if yes_probability > no_probability:
             return f"Your question violates the following policy: {policy.split(':')[0]}"
     return None
 
@@ -70,7 +79,26 @@ def chat_stream_completion(message, history, system_prompt):
             message_repl += chunk['choices'][0]["delta"]["content"]
             yield message_repl
 
-gr.ChatInterface(
-    chat_stream_completion,
-    additional_inputs=[gr.Textbox("You are a helpful AI.", label="System Prompt")]
-).queue().launch(server_name="0.0.0.0")
+# Interfaz de Gradio con la descripción y configuración adicional
+with gr.Blocks() as demo:
+    gr.Markdown("# Child-Safe-Chatbot (Experimental)")
+    gr.Markdown("""
+    ### Description
+    This chatbot is designed to assist users while ensuring that all interactions comply with defined safety policies. It checks whether user inputs violate any of the following categories:
+    
+    - Dangerous Content
+    - Harassment
+    - Hate Speech
+    - Sexually Explicit Information
+    
+    The chatbot will inform the user if any violation occurs and, if not, will proceed to respond to the user's message in a friendly manner.
+    
+    I'm Norberto Martín Afonso. Follow me on [Twitter](https://twitter.com/norbertomartnaf) and [GitHub](https://github.com/nmarafo) for more updates and projects!
+    """)
+    
+    gr.ChatInterface(
+        chat_stream_completion,
+        additional_inputs=[gr.Textbox("You are a helpful AI.", label="System Prompt")]
+    ).queue().launch(server_name="0.0.0.0")
+
+demo.launch(debug=True)