Update app.py

app.py

@@ -3,7 +3,6 @@ import requests
 import json
 import os
 
-
 BASE_URL = "https://api.jigsawstack.com/v1"
 headers = {
     "x-api-key": os.getenv("JIGSAWSTACK_API_KEY")
@@ -11,7 +10,7 @@ headers = {
 
 def validate_nsfw(url):
     if not url or not url.strip():
-        return None, "Error: Image URL is required.", None, None, None
+        return None, "Error: Image URL is required.", None, None
 
     try:
         response = requests.post(
@@ -24,7 +23,7 @@ def validate_nsfw(url):
 
         if not result.get("success"):
             error_msg = f"Error: API call failed - {result.get('message', 'Unknown error')}"
-            return url, error_msg, None, None, None
+            return url, error_msg, None, None
 
         # Extract detailed information
         nsfw = result.get("nsfw", False)
@@ -34,17 +33,17 @@ def validate_nsfw(url):
         nudity_score = result.get("nudity_score", 0)
         gore_score = result.get("gore_score", 0)
 
-        # Format the output for better readability
+        # Format the output
         status_message = f"Overall NSFW: {'Yes' if nsfw else 'No'}"
         nudity_details = f"Nudity: {'Detected' if nudity else 'Not Detected'} (Score: {nudity_score:.4f})"
         gore_details = f"Gore: {'Detected' if gore else 'Not Detected'} (Score: {gore_score:.4f})"
-        
+
         return url, status_message, nudity_details, gore_details
 
     except requests.exceptions.RequestException as e:
-        return url, f"Request failed: {str(e)}", None, None, None
+        return url, f"Request failed: {str(e)}", None, None
     except Exception as e:
-        return url, f"An unexpected error occurred: {str(e)}", None, None, None
+        return url, f"An unexpected error occurred: {str(e)}", None, None
 
 with gr.Blocks() as demo:
     gr.Markdown("""
@@ -53,13 +52,14 @@ with gr.Blocks() as demo:
             <p style='font-size:1.2em; margin-top: 0;'>Quickly detect nudity, violence, and other NSFW content in images.</p>
             <p style='font-size:1em; margin-top: 0.5em;'>For more details and API usage, see the <a href='https://jigsawstack.com/docs/api-reference/validate/nsfw' target='_blank'>documentation</a>.</p>
         </div>
-        """)
+    """)
+
     with gr.Row():
         with gr.Column():
             gr.Markdown("#### Image to Analyze")
             nsfw_url = gr.Textbox(
                 label="Image URL",
-                 placeholder="Enter the URL of the image you want to validate..."
+                placeholder="Enter the URL of the image you want to validate..."
             )
             nsfw_btn = gr.Button("Validate Image", variant="primary")
 
@@ -71,8 +71,14 @@ with gr.Blocks() as demo:
             nsfw_gore_details = gr.Textbox(label="Gore Details", interactive=False)
 
     nsfw_btn.click(
-            validate_nsfw,
-            inputs=[nsfw_url],
-            outputs=[nsfw_image_display, nsfw_status_message, nsfw_nudity_details, nsfw_gore_details]
-        )
+        validate_nsfw,
+        inputs=[nsfw_url],
+        outputs=[
+            nsfw_image_display,
+            nsfw_status_message,
+            nsfw_nudity_details,
+            nsfw_gore_details
+        ]
+    )
+
 demo.launch()