Update app.py

app.py

@@ -2,7 +2,7 @@ import gradio as gr
 import requests
 import json
 import time
-from better_profanity import profanity  
+from better_profanity import profanity
 
 url = "https://nexra.aryahcr.cc/api/image/complements"
 headers = {
@@ -10,13 +10,14 @@ headers = {
 }
 
 profanity.load_censor_words()
+no_nsfw_image_path = "no_nsfw.png"  
 
 def is_nsfw(text):
     return profanity.contains_profanity(text)
 
 def generate_image(prompt, model):
     if is_nsfw(prompt):
-        return "Error: Prompt contains inappropriate content."
+        return no_nsfw_image_path  
 
     data = {
         "prompt": prompt,
@@ -30,7 +31,7 @@ def generate_image(prompt, model):
             image_id = response_data.get("id")
             
             if not image_id:
-                return "Error: No image ID returned in the response."
+                return no_nsfw_image_path 
 
             while True:
                 status_response = requests.get(f"{url}/{image_id}")
@@ -45,23 +46,23 @@ def generate_image(prompt, model):
                             image_url = images[0]
                             return image_url
                         else:
-                            return "Error: No images found in the response."
+                            return no_nsfw_image_path  
                     elif status == "error":
-                        return "Error: Image generation failed."
+                        return no_nsfw_image_path
                     elif status == "not_found":
-                        return "Error: Image ID not found."
+                        return no_nsfw_image_path
                     elif status == "pending":
                         time.sleep(1)
                     else:
-                        return f"Error: Unexpected status '{status}'."
+                        return no_nsfw_image_path
                 else:
-                    return f"Error: Status check failed with code {status_response.status_code}."
+                    return no_nsfw_image_path
         else:
-            return f"Error: Initial request failed with code {response.status_code}."
+            return no_nsfw_image_path
     except json.JSONDecodeError:
-        return "Error: Invalid JSON response."
+        return no_nsfw_image_path
     except Exception as e:
-        return f"Exception occurred: {str(e)}"
+        return no_nsfw_image_path
 
 iface = gr.Interface(
     fn=generate_image,