Update app.py

app.py

@@ -13,6 +13,52 @@ def is_token_configured():
     if not HF_API_TOKEN:
         return "⚠️ Warning: HF_API_TOKEN is not configured. The app won't work until you add this secret in your Space settings."
     return "✅ API token is configured"
+import requests
+
+def check_safety(input_text):
+    if not input_text.strip():
+        return "⚠️ Please enter some text to check."
+    
+    payload = {
+        "inputs": input_text
+    }
+    
+    headers = {
+        "Content-Type": "application/json",
+        "Authorization": f"Bearer {HF_API_TOKEN}"  # Assuming you have token ready
+    }
+    
+    try:
+        response = requests.post(ENDPOINT_URL, json=payload, headers=headers, timeout=30)
+        
+        if response.status_code == 200:
+            result = response.json()
+            
+            is_safe = result.get("is_safe", False)
+            safety_result = result.get("safety_result", {})
+
+            # Extract the fields
+            safety = safety_result.get("Safety", "Unknown")
+            score = safety_result.get("Score", "")
+            categories = safety_result.get("Unsafe Categories", "")
+
+            # Format nicely
+            if is_safe:
+                return f"✅ Safe\n\nSafety: {safety}\nScore: {score}\nUnsafe Categories: {categories}"
+            else:
+                return f"❌ Unsafe\n\nSafety: {safety}\nScore: {score}\nUnsafe Categories: {categories}"
+        
+        else:
+            return f"❗ Error: Request failed with status code {response.status_code}.\nDetails: {response.text}"
+    
+    except requests.exceptions.Timeout:
+        return "❗ Error: Request timed out. The endpoint may be overloaded or unavailable."
+    
+    except requests.exceptions.ConnectionError:
+        return "❗ Error: Failed to connect to the endpoint. Please check the endpoint URL."
+    
+    except Exception as e:
+        return f"❗ Error: {str(e)}"
 
 
 with gr.Blocks(title="Safety Content Classifier", css="footer {display: none !important}") as demo: