Update app.py

app.py

@@ -1,6 +1,7 @@
 import gradio as gr
 import requests
 import base64
+import os
 
 def image_to_base64(image):
     with open(image.name, "rb") as file:
@@ -8,6 +9,12 @@ def image_to_base64(image):
         return encoded_string.decode("utf-8")
 
 def anti_nsfw(image):
+    allowed_extensions = [".png", ".jpeg", ".jpg"]
+    file_extension = os.path.splitext(image.name)[1].lower()
+    
+    if file_extension not in allowed_extensions:
+        return "Invalid file type. Please upload a PNG or JPEG image."
+    
     encoded_image = image_to_base64(image)
     resp = requests.get(
         "https://4937ea62-c0b0-414f-afbc-53be1f5b0d06.id.repl.co/scan",
@@ -17,7 +24,7 @@ def anti_nsfw(image):
 
 iface = gr.Interface(
     fn=anti_nsfw,
-    inputs=gr.inputs.File(accept="image/png, image/jpeg"),  # Specify accepted file types
+    inputs="file",
     outputs="text",
     title="AntiNSFW",
     description="Check if an image is safe for work (SFW) or not safe for work (NSFW)."