Update app.py

app.py

@@ -8,15 +8,15 @@ def fetch_tags(image_url):
     try:
         image_id = image_url.split('/')[-1].split('.')[0]
         if not image_id.isdigit():
-            return '[ERROR]: Invalid image URL format.'
+            return '[ERROR]: Invalid image URL format.', '', '', ''
     except IndexError:
-        return '[ERROR]: Invalid image URL format.'
+        return '[ERROR]: Invalid image URL format.', '', '', ''
 
     base_url = 'https://danbooru.donmai.us/posts'
     response = requests.get(f'{base_url}/{image_id}.json')
 
     if response.status_code != 200:
-        return f'[ERROR]: {response.status_code} - Failed to retrieve data.'
+        return f'[ERROR]: {response.status_code} - Failed to retrieve data.', '', '', ''
 
     data = json.loads(response.text)
 
@@ -26,17 +26,10 @@ def fetch_tags(image_url):
     tags = data.get('tag_string_general', '')
 
     # Prepare prompt and formatted tags
-    prompt = f'{character} {origin} {tags}'
     formatted_tags = tags.replace(" ", ", ")
-    formatted_prompt = prompt.replace(" ", ", ")
+    prompt = f'{character}, {origin}, {formatted_tags}'
 
-    # Display the results
-    return {
-        "Character": character,
-        "Origin": origin,
-        "Tags": formatted_tags,
-        "Prompt": formatted_prompt
-    }
+    return character, origin, formatted_tags, prompt
 
 
 # Create a Gradio interface
@@ -49,8 +42,8 @@ iface = gr.Interface(
         gr.Textbox(label="Tags (comma-separated)"),
         gr.Textbox(label="Prompt (comma-separated)")
     ],
-    title="Danbooru Tag Fetcher",
-    description="Enter a Danbooru image URL to fetch character, origin, tags, and prompt information."
+    title="Danbooru Tag Extractor",
+    description="Enter a Danbooru image URL to fetch and format character, origin, tags, and prompt information."
 )
 
 # Launch the interface