Update app.py

app.py

@@ -47,9 +47,10 @@ SPECIAL_WORDS = [
 
 # Global variables
 initial_word_design = ""
-special_words = []
+special_word = ""
 
 def generate_initial_design(word):
+    """Generate initial design for the special word in black color."""
     fonts = [
         "'VT323', monospace",
         "'Josefin Sans', sans-serif",
@@ -60,7 +61,12 @@ def generate_initial_design(word):
         "'Nothing You Could Do', cursive",
         "'Reenie Beanie', cursive",
         "'Orbitron', sans-serif",
-        "'Raleway', sans-serif"
+        "'Raleway', sans-serif",
+        "'Open Sans Condensed', sans-serif",
+        "'Poiret One', cursive",
+        "'Indie Flower', cursive",
+        "'Pacifico', cursive",
+        "'Teko', sans-serif"
     ]
     font_sizes = ["18px", "19px", "20px"]
     font_tops = ["0px", "1px", "-1px"]
@@ -110,7 +116,8 @@ def generate_initial_design(word):
         {" ".join(styled_letters)}
     </span>'''
 
-def generate_movement_design(words):
+def generate_movement_design(word):
+    """Generate a new random design for the movement animation."""
     fonts = [
         "'VT323', monospace",
         "'Josefin Sans', sans-serif",
@@ -139,9 +146,13 @@ def generate_movement_design(words):
     ]
     skew_angles = ["-25deg", "-20deg", "-15deg", "-10deg", "0deg", "10deg", "15deg", "20deg", "25deg"]
 
+    # Generate random color for the movement design
     random_color = f'#{random.randint(0, 0xFFFFFF):06x}'
+    
+    # Generate unique animation name
     animation_name = f"animate_{random.randint(0, 10000)}"
 
+    # Create keyframes for the animation sequence
     keyframes = f"""
     @keyframes {animation_name} {{
         0% {{ transform: scale(1) rotate(0deg); }}
@@ -150,35 +161,31 @@ def generate_movement_design(words):
     }}
     """
 
-    styled_words = []
-    for word in words:
-        letters = list(word)
-        styled_letters = []
-
-        for i, letter in enumerate(letters):
-            style = {
-                'font-family': random.choice(fonts),
-                'line-height': '1.6',
-                'font-size': random.choice(font_sizes),
-                'letter-spacing': random.choice(letter_spacings),
-                'text-shadow': random.choice(text_shadows),
-                'transform': f'skew({random.choice(skew_angles)})',
-                'margin-top': random.choice(["-0.02cm", "0.00cm", "0.02cm"]),
-                'position': 'relative',
-                'top': random.choice(font_tops),
-                'color': random_color,
-                'display': 'inline-block',
-                'margin': '0 1px',
-                'vertical-align': 'middle',
-                'animation': f'{animation_name} 0.5s ease-in-out',
-                'animation-delay': f'{i * 0.1}s'
-            }
-            
-            style_str = '; '.join([f'{k}: {v}' for k, v in style.items()])
-            styled_letter = f'<span class="styled-letter" style="{style_str}">{letter}</span>'
-            styled_letters.append(styled_letter)
+    letters = list(word)
+    styled_letters = []
+    
+    for i, letter in enumerate(letters):
+        style = {
+            'font-family': random.choice(fonts),
+            'line-height': '1.6',
+            'font-size': random.choice(font_sizes),
+            'letter-spacing': random.choice(letter_spacings),
+            'text-shadow': random.choice(text_shadows),
+            'transform': f'skew({random.choice(skew_angles)})',
+            'margin-top': random.choice(["-0.02cm", "0.00cm", "0.02cm"]),
+            'position': 'relative',
+            'top': random.choice(font_tops),
+            'color': random_color,
+            'display': 'inline-block',
+            'margin': '0 1px',
+            'vertical-align': 'middle',
+            'animation': f'{animation_name} 0.5s ease-in-out',
+            'animation-delay': f'{i * 0.1}s'
+        }
         
-        styled_words.append("".join(styled_letters))
+        style_str = '; '.join([f'{k}: {v}' for k, v in style.items()])
+        styled_letter = f'<span class="styled-letter" style="{style_str}">{letter}</span>'
+        styled_letters.append(styled_letter)
 
     return f'''
     <style>
@@ -191,13 +198,14 @@ def generate_movement_design(words):
                  align-items: baseline; 
                  vertical-align: middle; 
                  margin: 0 2px;">
-        {" ".join(styled_words)}
+        {" ".join(styled_letters)}
     </span>
     '''
 
 def process_text(input_text):
-    global initial_word_design, special_words
+    global initial_word_design, special_word
 
+    # Generate text with more tokens
     generated = generator(
         input_text,
         max_new_tokens=MAX_NEW_TOKENS,
@@ -208,17 +216,17 @@ def process_text(input_text):
     
     generated_text = generated[0]['generated_text']
     generated_text = generated_text[:1000]
-    words = generated_text.split()
 
-    special_words = []
-    if len(words) >= 2:
-        special_words = [words[-2], words[-1]]  # Get last two words
-
-    for word in special_words:
+    words = generated_text.split()
+    
+    for i, word in enumerate(words):
         clean_word = ''.join(filter(str.isalnum, word)).lower()
         if clean_word in SPECIAL_WORDS:
+            special_word = word
             initial_word_design = generate_initial_design(word)
-            words[words.index(word)] = initial_word_design
+            words[i] = initial_word_design
+        else:
+            words[i] = word
 
     output_html = ' '.join(words)
     
@@ -257,16 +265,17 @@ def process_text(input_text):
     return final_output
 
 def trigger_movement(input_html):
-    global initial_word_design, special_words
+    global initial_word_design, special_word
     
-    if not initial_word_design or not special_words:
+    if not initial_word_design or not special_word:
         return input_html
 
-    movement_design = generate_movement_design(special_words)
+    movement_design = generate_movement_design(special_word)
     updated_html = input_html.replace(initial_word_design, movement_design, 1)
 
     return updated_html
 
+# Create Gradio interface using Blocks
 with gr.Blocks() as demo:
     gr.Markdown("# CircularText Styler\nEnter a prompt to generate text with special word styling.")
     
@@ -281,8 +290,9 @@ with gr.Blocks() as demo:
         fn=process_text,
         inputs=input_text,
         outputs=output_html,
-        show_progress=True  # Enable a basic loading indicator
+        show_progress=True  # Enable the loading indicator
     )
     animate_button.click(trigger_movement, inputs=output_html, outputs=output_html)
 
-demo.launch()
+# Launch the app
+demo.launch()