Update app.py

app.py

@@ -21,12 +21,14 @@ def fetch_tags(image_url):
     data = json.loads(response.text)
 
     # Extract required fields
-    character = data.get('tag_string_character', 'N/A')
-    origin = data.get('tag_string_copyright', 'N/A')
-    tags = data.get('tag_string_general', '')
+    character = data.get('tag_string_character', 'N/A').replace('_', ' ')
+    origin = data.get('tag_string_copyright', 'N/A').replace('_', ' ')
+    tags = data.get('tag_string_general', '').replace('_', ' ')
 
-    # Prepare prompt and formatted tags
+    # Prepare formatted tags (comma-separated)
     formatted_tags = tags.replace(" ", ", ")
+
+    # Prepare prompt with cleaned-up fields
     prompt = f'{character}, {origin}, {formatted_tags}'
 
     return character, origin, formatted_tags, prompt
@@ -47,4 +49,4 @@ iface = gr.Interface(
 )
 
 # Launch the interface
-iface.launch()
+iface.launch()