Update app.py

app.py

@@ -19,33 +19,54 @@ class SafetyChecker:
         self.ENDPOINT_URL = ENDPOINT_URL
         self.HF_API_TOKEN = HF_API_TOKEN
 
-    def extract_and_parse_json(self, response: str):
-        match = re.search(r'```(?:json)?\s*(.*?)\s*```', response, re.DOTALL)
-        content = match.group(1).strip() if match else response.strip()
+    # def extract_and_parse_json(self, response: str):
+    #     match = re.search(r'```(?:json)?\s*(.*?)\s*```', response, re.DOTALL)
+    #     content = match.group(1).strip() if match else response.strip()
         
-        if not content.startswith("{") and ":" in content:
-            content = "{" + content + "}"
+    #     if not content.startswith("{") and ":" in content:
+    #         content = "{" + content + "}"
         
+    #     try:
+    #         parsed = json.loads(content)
+    #     except json.JSONDecodeError:
+    #         cleaned = content.replace(""", "\"").replace(""", "\"").replace("'", "\"")
+    #         cleaned = re.sub(r',\s*}', '}', cleaned)
+    #         cleaned = re.sub(r',\s*]', ']', cleaned)
+    #         try:
+    #             parsed = json.loads(cleaned)
+    #         except Exception:
+    #             pairs = re.findall(r'"([^"]+)":\s*"?([^",\{\}\[\]]+)"?', content)
+    #             if pairs:
+    #                 parsed = {k.strip(): v.strip() for k, v in pairs}
+    #             else:
+    #                 parsed = {
+    #                     "Safety": "",
+    #                     "Score": "",
+    #                     "Unsafe Categories": "",
+    #                 }
+    #     return parsed
+
+    def extract_and_parse_json(self, response: str):
+        if response.startswith("```"):
+            response = response.strip("`").strip()
+            if response.startswith("json"):
+                response = response[4:].strip()
+    
+        # Now response should be clean JSON text
         try:
-            parsed = json.loads(content)
-        except json.JSONDecodeError:
-            cleaned = content.replace(""", "\"").replace(""", "\"").replace("'", "\"")
-            cleaned = re.sub(r',\s*}', '}', cleaned)
-            cleaned = re.sub(r',\s*]', ']', cleaned)
-            try:
-                parsed = json.loads(cleaned)
-            except Exception:
-                pairs = re.findall(r'"([^"]+)":\s*"?([^",\{\}\[\]]+)"?', content)
-                if pairs:
-                    parsed = {k.strip(): v.strip() for k, v in pairs}
-                else:
-                    parsed = {
-                        "Safety": "",
-                        "Score": "",
-                        "Unsafe Categories": "",
-                    }
+            parsed = json.loads(response)
+        except Exception:
+            # If somehow still not JSON, return default empty
+            parsed = {
+                "Safety": "",
+                "Score": "",
+                "Unsafe Categories": [],
+            }
+        
         return parsed
 
+    
+
     def check_safety(self, input_text):
         if not input_text.strip():
             return "⚠️ Please enter some text to check."