Update app.py

app.py

@@ -1,16 +1,16 @@
 import requests
 import json
 import gradio as gr
+import re
 
 
 def fetch_tags(image_url):
-    # Extract image ID from the provided URL
-    try:
-        image_id = image_url.split('/')[-1].split('.')[0]
-        if not image_id.isdigit():
-            return '[ERROR]: Invalid image URL format.', '', '', ''
-    except IndexError:
-        return '[ERROR]: Invalid image URL format.', '', '', ''
+    # Try to extract the image ID from the URL
+    match = re.search(r'https://danbooru.donmai.us/posts/(\d+)', image_url)
+    if not match:
+        return '[ERROR]: Invalid image URL format. Please use a valid Danbooru URL.', '', '', ''
+
+    image_id = match.group(1)  # Extract image ID from URL
 
     base_url = 'https://danbooru.donmai.us/posts'
     response = requests.get(f'{base_url}/{image_id}.json')