Update src/streamlit_app.py

src/streamlit_app.py

@@ -137,43 +137,37 @@ def adjust_syllables(word: str, target_count: int) -> List[str]:
 
 def process_text(text: str) -> Tuple[List[Tuple[str, str]], List[Tuple[str, List[Tuple[str, str]]]]]:
     """Process text and return colored words and syllable breakdown"""
-    words = re.findall(r'\S+|\s+', text)  # Keep spaces
+    # Split into words while preserving spaces
+    words = re.findall(r'\b\w+\b', text)  # Only actual words, no spaces
     
     colored_words = []
     syllable_breakdown = []
     
-    for word in words:
-        if word.strip():  # If it's a word (not just whitespace)
-            syllables_list = syllabify_word(word.strip())
-            
-            # Generate colors for this word's syllables
-            colors = generate_color_palette(len(syllables_list))
-            
-            # For original text coloring, we need to map syllables back to original word
-            word_colored = []
-            syllable_colored = []
-            
-            # Handle punctuation
-            clean_word = re.sub(r'[^\w]', '', word.strip())
-            prefix_punct = re.match(r'^[^\w]*', word.strip()).group()
-            suffix_punct = re.search(r'[^\w]*$', word.strip()).group()
-            
-            if clean_word:
-                # Color each syllable
-                for i, syl in enumerate(syllables_list):
-                    color = colors[i % len(colors)]
-                    syllable_colored.append((syl, color))
+    # Process original text with spaces preserved
+    text_parts = re.findall(r'\S+|\s+', text)
+    
+    word_index = 0
+    for part in text_parts:
+        if part.strip() and re.match(r'\w', part):  # If it's a word
+            if word_index < len(words):
+                word = words[word_index]
+                syllables_list = syllabify_word(word)
+                
+                # Generate colors for this word's syllables
+                colors = generate_color_palette(len(syllables_list))
                 
-                # For the original word, we'll color it uniformly with the first syllable color
-                # or create a gradient effect
-                word_with_punct = prefix_punct + clean_word + suffix_punct
-                colored_words.append((word_with_punct, colors[0] if colors else "#000000"))
+                # For original text, color with first syllable color
+                colored_words.append((part, colors[0] if colors else "#000000"))
                 
-                syllable_breakdown.append((word.strip(), syllable_colored))
+                # Add to syllable breakdown (no duplicates)
+                syllable_colored = [(syl, colors[i % len(colors)]) for i, syl in enumerate(syllables_list)]
+                syllable_breakdown.append((word, syllable_colored))
+                
+                word_index += 1
             else:
-                colored_words.append((word, "#000000"))
+                colored_words.append((part, "#000000"))
         else:
-            colored_words.append((word, "#000000"))  # Spaces remain uncolored
+            colored_words.append((part, "#000000"))  # Spaces and punctuation
     
     return colored_words, syllable_breakdown
 
@@ -194,22 +188,23 @@ def display_syllable_breakdown(syllable_breakdown: List[Tuple[str, List[Tuple[st
     html_parts = []
     
     for word, syllables in syllable_breakdown:
-        if syllables:
-            word_html = []
+        if syllables and len(syllables) > 0:
+            # Create hyphenated syllable display
+            syllable_spans = []
             for syl, color in syllables:
-                word_html.append(f'<span style="color: {color}; font-weight: bold; margin-right: 2px;">{syl}</span>')
+                syllable_spans.append(f'<span style="color: {color}; font-weight: bold;">{syl}</span>')
             
-            # Add separator between syllables
-            syllable_display = '<span style="color: #666;">|</span>'.join([
-                f'<span style="color: {color}; font-weight: bold;">{syl}</span>' 
-                for syl, color in syllables
-            ])
+            # Join with hyphens for multi-syllable words
+            if len(syllables) > 1:
+                syllable_display = '<span style="color: #666;">-</span>'.join(syllable_spans)
+            else:
+                syllable_display = syllable_spans[0]
             
-            html_parts.append(f'<div style="margin-bottom: 10px;"><strong>{word}:</strong> {syllable_display}</div>')
+            html_parts.append(f'<span style="margin-right: 20px; display: inline-block; margin-bottom: 8px;">{syllable_display}</span>')
     
     if html_parts:
         html_content = ''.join(html_parts)
-        st.markdown(f'<div style="font-size: 16px; line-height: 1.8; padding: 15px; border: 2px solid #ddd; border-radius: 10px; background-color: #f0f8ff;">{html_content}</div>', unsafe_allow_html=True)
+        st.markdown(f'<div style="font-size: 18px; line-height: 2.0; padding: 15px; border: 2px solid #ddd; border-radius: 10px; background-color: #f0f8ff;">{html_content}</div>', unsafe_allow_html=True)
 
 # Main app
 def main():