Update app.py

app.py

@@ -7,7 +7,18 @@ import re
 from gradio_client import Client
 
 is_shared_ui = True if "fffiloni/consistent-character" in os.environ['SPACE_ID'] else False
-
+def safety_check(user_prompt):
+    
+    client = Client("fffiloni/safety-checker-bot", hf_token=hf_token)
+    response = client.predict(
+            source_space="consistent-character space",
+            user_prompt=user_prompt,
+            api_name="/infer"
+    )
+    print(response)
+    
+    return response
+    
 from utils.gradio_helpers import parse_outputs, process_outputs
 
 names = ['prompt', 'negative_prompt', 'subject', 'number_of_outputs', 'number_of_images_per_pose', 'randomise_poses', 'output_format', 'output_quality', 'seed']
@@ -23,31 +34,21 @@ def predict(request: gr.Request, *args, progress=gr.Progress(track_tqdm=True)):
     try:
         if is_shared_ui:
             hf_token = os.environ.get("HF_TOKEN")
-            
-            def safety_check(user_prompt):
-    
-                client = Client("fffiloni/safety-checker-bot", hf_token=hf_token)
-                response = client.predict(
-                		source_space="consistent-character space",
-                		user_prompt=user_prompt,
-                		api_name="/infer"
-                )
-                print(response)
-                
-                return response
-            
+
             is_safe = safety_check(args[0])
             print(is_safe)
-        
+    
             match = re.search(r'\bYes\b', is_safe)
-        
+    
             if match:
                 status = 'Yes'
             else:
                 status = None
-        
-            if status == "Yes" :
-                raise gr.Error("Do not ask for such things.")
+        else:
+            status = None
+    
+        if status == "Yes" :
+            raise gr.Error("Do not ask for such things.")
         else:
     
             headers = {'Content-Type': 'application/json'}