Update app.py

app.py

@@ -15,6 +15,9 @@ def is_token_configured():
     return "✅ API token is configured"
 import requests
 
+import json
+import requests
+
 def check_safety(input_text):
     if not input_text.strip():
         return "⚠️ Please enter some text to check."
@@ -25,37 +28,44 @@ def check_safety(input_text):
     
     headers = {
         "Content-Type": "application/json",
-        "Authorization": f"Bearer {HF_API_TOKEN}"  # Assuming you have token ready
+        "Authorization": f"Bearer {HF_API_TOKEN}"
     }
     
     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", "")
+        if response.headers.get("content-type", "").startswith("application/json"):
+            result = response.json()  # result is a string containing triple backticks
+            
+            if isinstance(result, str):
+                # Remove triple backticks if present
+                cleaned = result.strip()
+                if cleaned.startswith("```"):
+                    cleaned = cleaned.strip("```").strip()
+                    if cleaned.startswith("json"):
+                        cleaned = cleaned[4:].strip()  # remove 'json' label if there
+                
+                # Now parse cleaned string
+                result = json.loads(cleaned)
+            
+            # Now safely access fields
+            is_safe = result.get("Safety", "").lower() == "safe"
+            score = result.get("Score", "")
+            categories = result.get("Unsafe Categories", "")
 
-            # Format nicely
             if is_safe:
-                return f"✅ Safe\n\nSafety: {safety}\nScore: {score}\nUnsafe Categories: {categories}"
+                return f"✅ Safe\n\nSafety: safe\nScore: {score}\nUnsafe Categories: {categories}"
             else:
-                return f"❌ Unsafe\n\nSafety: {safety}\nScore: {score}\nUnsafe Categories: {categories}"
+                return f"❌ Unsafe\n\nSafety: unsafe\nScore: {score}\nUnsafe Categories: {categories}"
         
         else:
-            return f"❗ Error: Request failed with status code {response.status_code}.\nDetails: {response.text}"
-    
+            return f"❗ Error: Server returned non-JSON response:\n\n{response.text}"
+
     except requests.exceptions.Timeout:
-        return "❗ Error: Request timed out. The endpoint may be overloaded or unavailable."
+        return "❗ Error: Request timed out."
     
     except requests.exceptions.ConnectionError:
-        return "❗ Error: Failed to connect to the endpoint. Please check the endpoint URL."
+        return "❗ Error: Failed to connect to the endpoint."
     
     except Exception as e:
         return f"❗ Error: {str(e)}"