Option to replace underscores with spaces.

app/app.py

@@ -523,7 +523,6 @@ def image_tagger_app():
     windows_system = is_windows()
     flash_attn_installed = check_flash_attention()
 
-    # Initialize session state settings
     if 'settings' not in st.session_state:
         st.session_state.settings = {
             'show_all_tags': False,
@@ -532,7 +531,8 @@ def image_tagger_app():
             'threshold_profile': "Balanced",
             'active_threshold': 0.35,
             'active_category_thresholds': None,
-            'selected_categories': {}
+            'selected_categories': {},
+            'replace_underscores': False  # Added new setting
         }
         # Initialize show_profile_help state
         st.session_state.show_profile_help = False
@@ -1687,6 +1687,10 @@ def image_tagger_app():
                                         value=st.session_state.settings['show_all_tags'])
                 compact_view = st.checkbox("Compact view (hide progress bars)", 
                                         value=st.session_state.settings['compact_view'])
+                
+                # Add the new checkbox for replacing underscores with spaces
+                replace_underscores = st.checkbox("Replace underscores with spaces", 
+                                            value=st.session_state.settings.get('replace_underscores', False))
             
             with col2:
                 min_confidence = st.slider("Minimum confidence to display", 0.0, 0.5, 
@@ -1696,6 +1700,7 @@ def image_tagger_app():
             st.session_state.settings['show_all_tags'] = show_all_tags
             st.session_state.settings['compact_view'] = compact_view
             st.session_state.settings['min_confidence'] = min_confidence
+            st.session_state.settings['replace_underscores'] = replace_underscores
             
             # Category selection for the "All Tags" section
             st.write("Categories to include in 'All Tags' section:")
@@ -1931,26 +1936,38 @@ def image_tagger_app():
                         if compact_view:
                             # Compact view - show tags in a comma-separated list with %
                             tag_list = []
+
+                            replace_underscores = st.session_state.settings.get('replace_underscores', False)
                             for tag, prob in limited_tags_to_display:
                                 # Format: tag (percent%)
                                 percentage = int(prob * 100)
-                                tag_list.append(f"{tag} ({percentage}%)")
+                                
+                                # Apply underscore replacement for display if enabled
+                                display_tag = tag.replace('_', ' ') if replace_underscores else tag
+                                tag_list.append(f"{display_tag} ({percentage}%)")
                                 
                                 # Add tag to all_tags list if it passes threshold and category is selected
+                                # Note: always use original tag (with underscores) for all_tags collection
                                 if prob >= threshold and selected_categories.get(category, True):
                                     all_tags.append(tag)
-                            
+
                             # Join with commas and display
                             st.markdown(", ".join(tag_list))
                         else:
                             # Expanded view with progress bars
                             for tag, prob in limited_tags_to_display:
+                                # Get the replacement setting 
+                                replace_underscores = st.session_state.settings.get('replace_underscores', False)
+                                
+                                # Apply underscore replacement for display if enabled
+                                display_tag = tag.replace('_', ' ') if replace_underscores else tag
+                                
                                 # Add tag to all_tags list if it passes threshold and category is selected
                                 if prob >= threshold and selected_categories.get(category, True):
-                                    all_tags.append(tag)
-                                    tag_display = f"**{tag}**"  # Bold for tags above threshold
+                                    all_tags.append(tag)  # Add original tag to all_tags
+                                    tag_display = f"**{display_tag}**"  # Bold for tags above threshold
                                 else:
-                                    tag_display = tag  # Regular for tags below threshold
+                                    tag_display = display_tag  # Regular for tags below threshold
                                 
                                 # Display tag with progress bar
                                 st.write(tag_display)
@@ -1964,7 +1981,16 @@ def image_tagger_app():
             st.markdown("---")
             st.subheader(f"All Tags ({len(all_tags)} total)")
             if all_tags:
-                st.write(", ".join(all_tags))
+                # Check if underscore replacement is enabled
+                replace_underscores = st.session_state.settings.get('replace_underscores', False)
+                
+                if replace_underscores:
+                    # Create a new list with underscores replaced by spaces for display only
+                    display_tags = [tag.replace('_', ' ') for tag in all_tags]
+                    st.write(", ".join(display_tags))
+                else:
+                    # Display original tags
+                    st.write(", ".join(all_tags))
             else:
                 st.info("No tags detected above the threshold.")