Upload folder using huggingface_hub

hate_speech_demo.py

@@ -513,15 +513,25 @@ def get_contextual_rating(contextual_api, user_input):
     if error:
         return f"Error: {error}", "", "unsafe"
     
+    # Store the original response for safety level detection
+    original_response = response_text
+    
     # Determine safety level based on response content
     safety_level = "safe"
-    if "out of policy" in response_text.lower() or "unsafe" in response_text.lower():
+    if "out of policy" in original_response.lower() or "unsafe" in original_response.lower():
         safety_level = "unsafe"
-    elif "caution" in response_text.lower() or "warning" in response_text.lower():
+    elif "caution" in original_response.lower() or "warning" in original_response.lower():
         safety_level = "warning"
     
-    # Format the response text to separate rating, category, and explanation
-    formatted_response = format_contextual_rating(response_text)
+    # Try to format the response (but keep the original if anything goes wrong)
+    try:
+        # Simple formatting approach - just add line breaks between sentences
+        formatted_response = response_text.replace(". ", ".<br><br>")
+        # Ensure we don't have too many breaks
+        formatted_response = formatted_response.replace("<br><br><br>", "<br><br>")
+    except Exception as e:
+        print(f"Error formatting response: {e}")
+        formatted_response = response_text
     
     return formatted_response, retrieval_text, safety_level