Upload folder using huggingface_hub

hate_speech_demo.py

@@ -609,41 +609,67 @@ def get_contextual_rating(contextual_api, user_input):
     elif "caution" in response_text.lower() or "warning" in response_text.lower():
         safety_level = "warning"
     
-    # Parse specific parts of the response
+    # Add simple fallback in case parsing fails
+    formatted_text = response_text
+    
+    # Try to parse specific parts of the response with more robust error handling
     try:
-        # Look for policy rating
-        rating_match = re.search(r'Policy Rating: (.*?)(?:Policy|$)', response_text, re.IGNORECASE)
-        policy_rating = rating_match.group(1).strip() if rating_match else ""
+        # Simple approach: Split by key phrases and clean up
+        parts = {}
         
-        # Look for policy category
-        category_match = re.search(r'Policy Category: (.*?)(?:Target of Hate|Explanation|$)', response_text, re.IGNORECASE)
-        policy_category = category_match.group(1).strip() if category_match else ""
+        # Extract policy rating
+        if "Policy Rating:" in response_text:
+            policy_part = response_text.split("Policy Rating:")[1]
+            if "Policy Category:" in policy_part:
+                parts["rating"] = policy_part.split("Policy Category:")[0].strip()
+            elif "Target of Hate:" in policy_part:
+                parts["rating"] = policy_part.split("Target of Hate:")[0].strip()
+            elif "Explanation:" in policy_part:
+                parts["rating"] = policy_part.split("Explanation:")[0].strip()
+            else:
+                parts["rating"] = policy_part.strip()
         
-        # Look for Target of Hate (new field)
-        target_match = re.search(r'Target of Hate: (.*?)(?:Explanation|$)', response_text, re.IGNORECASE)
-        target_of_hate = target_match.group(1).strip() if target_match else ""
+        # Extract policy category
+        if "Policy Category:" in response_text:
+            category_part = response_text.split("Policy Category:")[1]
+            if "Target of Hate:" in category_part:
+                parts["category"] = category_part.split("Target of Hate:")[0].strip()
+            elif "Explanation:" in category_part:
+                parts["category"] = category_part.split("Explanation:")[0].strip()
+            else:
+                parts["category"] = category_part.strip()
         
-        # Look for explanation
-        explanation_match = re.search(r'Explanation: (.*)', response_text, re.IGNORECASE)
-        explanation = explanation_match.group(1).strip() if explanation_match else response_text
+        # Extract target of hate
+        if "Target of Hate:" in response_text:
+            target_part = response_text.split("Target of Hate:")[1]
+            if "Explanation:" in target_part:
+                parts["target"] = target_part.split("Explanation:")[0].strip()
+            else:
+                parts["target"] = target_part.strip()
         
-        # Format with HTML
-        formatted_text = ""
-        if policy_rating:
-            formatted_text += f"<strong>Policy Rating:</strong> {policy_rating}<br><br>"
-        if policy_category:
-            formatted_text += f"<strong>Policy Category:</strong> {policy_category}<br><br>"
-        if target_of_hate:
-            formatted_text += f"<strong>Target of Hate:</strong> {target_of_hate}<br><br>"
-        if explanation:
-            formatted_text += f"<strong>Explanation:</strong> {explanation}"
+        # Extract explanation
+        if "Explanation:" in response_text:
+            parts["explanation"] = response_text.split("Explanation:")[1].strip()
         
-        # If we couldn't extract anything, use the original
-        if not formatted_text:
-            formatted_text = response_text
+        # Build formatted output with HTML
+        if parts:
+            formatted_text = ""
+            if "rating" in parts:
+                formatted_text += f"<strong>Policy Rating:</strong> {parts['rating']}<br><br>"
+            if "category" in parts:
+                formatted_text += f"<strong>Policy Category:</strong> {parts['category']}<br><br>"
+            if "target" in parts:
+                formatted_text += f"<strong>Target of Hate:</strong> {parts['target']}<br><br>"
+            if "explanation" in parts:
+                formatted_text += f"<strong>Explanation:</strong> {parts['explanation']}"
             
+            # If we end up with empty formatted text, fall back to original
+            if not formatted_text.strip():
+                formatted_text = response_text
+                
     except Exception as e:
-        print(f"Error formatting rating: {e}")
+        print(f"Error formatting rating (falling back to raw text): {e}")
+        # Fall back to the original text in case of any parsing error
         formatted_text = response_text
     
     return formatted_text, retrieval_text, safety_level