Update app.py

app.py

@@ -3,7 +3,7 @@ import random
 from transformers import pipeline
 
 # Load the model once when the app starts
-generator = pipeline('text-generation', model='distilgpt2', max_length=25)
+generator = pipeline('text-generation', model='distilgpt2', max_length=25)  # Reduced max_length for faster inference
 
 # Predefined words to check
 SPECIAL_WORDS = [
@@ -27,8 +27,12 @@ SPECIAL_WORDS = [
     'legal', 'law', 'firm', 'crowd', 'winner', 'winter', 'smoking', 'green', 'purple', 'blue', 'pink', 
     'orange', 'black', 'white', 'yellow', 'gold', 'weather', 'sun', 'middle', 'summer', 'heat', 'spring'
 ]
-def generate_design(word, color='#000000', animate=False):
-    """Generate design for the special word."""
+# Global variables
+initial_word_design = ""
+special_word = ""
+
+def generate_initial_design(word):
+    """Generate initial design for the special word in black color."""
     fonts = [
         "'VT323', monospace",
         "'Josefin Sans', sans-serif",
@@ -39,10 +43,16 @@ def generate_design(word, color='#000000', animate=False):
         "'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"]
-    letter_spacings = ["-1px", "0px", "1px"]
+    font_sizes = ["18px", "19px", "20px"]  # Narrower range
+    font_tops = ["0px", "1px", "-1px"]  # Smaller adjustments
+    letter_spacings = ["-1px", "0px", "1px"]  # Reduced range
     text_shadows = [
         "0px 0px 1px",
         "0px 0px 2px",
@@ -54,34 +64,105 @@ def generate_design(word, color='#000000', animate=False):
         "1px 1px 0px",
         "1px 0px 1px"
     ]
-    skew_angles = ["-10deg", "0deg", "10deg"]
+    skew_angles = ["-25deg", "-20deg", "-15deg", "-10deg", "0deg", "10deg", "15deg", "20deg", "25deg"]
 
     letters = list(word)
     styled_letters = []
     
-    animation_name = f"animate_{random.randint(0, 10000)}" if animate else ""
+    for i, letter in enumerate(letters):
+        style = {
+            'font-family': random.choice(fonts),
+            'line-height': '1.6',  # Consistent with body text
+            '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"]),  # Reduced range
+            'position': 'relative',
+            'top': random.choice(font_tops),
+            'color': '#000000',
+            'display': 'inline-block',
+            'margin': '0 1px',
+            'vertical-align': 'middle'
+        }
+        
+        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'''
+    <span style="display: inline-flex; 
+                 align-items: baseline; 
+                 vertical-align: middle; 
+                 margin: 0 2px;">
+        {" ".join(styled_letters)}
+    </span>'''
+
+def generate_movement_design(word):
+    """Generate a completely new random design for the movement animation."""
+    fonts = [
+        "'VT323', monospace",
+        "'Josefin Sans', sans-serif",
+        "'Rajdhani', sans-serif",
+        "'Anton', sans-serif",
+        "'Caveat', cursive",
+        "'Patrick Hand', cursive",
+        "'Nothing You Could Do', cursive",
+        "'Reenie Beanie', cursive",
+        "'Orbitron', sans-serif",
+        "'Raleway', sans-serif"
+    ]
+    font_sizes = ["18px", "19px", "20px"]  # Narrower range
+    font_tops = ["0px", "1px", "-1px"]  # Smaller adjustments
+    letter_spacings = ["-1px", "0px", "1px"]  # Reduced range
+    text_shadows = [
+        "0px 0px 1px",
+        "0px 0px 2px",
+        "1px 0px 0px",
+        "0px 0px 0px",
+        "0px 1px 0px",
+        "0px 2px 0px",
+        "0px 1px 1px",
+        "1px 1px 0px",
+        "1px 0px 1px"
+    ]
+    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); }}
         50% {{ transform: scale(1.2) rotate(10deg); }}
         100% {{ transform: scale(1) rotate(0deg); }}
     }}
-    """ if animate else ""
+    """
 
+    letters = list(word)
+    styled_letters = []
+    
     for i, letter in enumerate(letters):
         style = {
             'font-family': random.choice(fonts),
-            'line-height': '1.6',
+            'line-height': '1.6',  # Consistent with body text
             '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)})',
-            'color': color,
+            'margin-top': random.choice(["-0.02cm", "0.00cm", "0.02cm"]),  # Reduced range
+            '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' if animate else '',
-            'animation-delay': f'{i * 0.1}s' if animate else ''
+            '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()])
@@ -95,28 +176,35 @@ def generate_design(word, color='#000000', animate=False):
         transition: all 0.3s ease;
     }}
     </style>
-    <span style="display: inline-flex; align-items: baseline; vertical-align: middle; margin: 0 2px;">
+    <span style="display: inline-flex; 
+                 align-items: baseline; 
+                 vertical-align: middle; 
+                 margin: 0 2px;">
         {" ".join(styled_letters)}
     </span>
     '''
 
 def process_text(input_text):
     """Process text and generate the initial output with special word styled in black."""
+    global initial_word_design, special_word
+
+    # Generate text with limited length
     generated = generator(input_text, num_return_sequences=1)
     generated_text = generated[0]['generated_text']
-    generated_text = generated_text[:200]
+    generated_text = generated_text[:200]  # Limit output length
 
     words = generated_text.split()
-    processed_words = []
-
-    for word in words:
+    
+    for i, word in enumerate(words):
         clean_word = ''.join(filter(str.isalnum, word)).lower()
         if clean_word in SPECIAL_WORDS:
-            processed_words.append(generate_design(word))
+            special_word = word
+            initial_word_design = generate_initial_design(word)
+            words[i] = initial_word_design
         else:
-            processed_words.append(word)
+            words[i] = word
 
-    output_html = ' '.join(processed_words)
+    output_html = ' '.join(words)
     
     final_output = f"""
     <html>
@@ -153,18 +241,15 @@ def process_text(input_text):
     return final_output
 
 def trigger_movement(input_html):
-    """Function to trigger the movement animation for all special words."""
-    words = input_html.split()
-    updated_words = []
+    """Function to trigger the movement animation."""
+    global initial_word_design, special_word
+    
+    if not initial_word_design or not special_word:
+        return input_html
 
-    for word in words:
-        clean_word = ''.join(filter(str.isalnum, word)).lower()
-        if clean_word in SPECIAL_WORDS:
-            updated_words.append(generate_design(clean_word, color=f'#{random.randint(0, 0xFFFFFF):06x}', animate=True))
-        else:
-            updated_words.append(word)
+    movement_design = generate_movement_design(special_word)
+    updated_html = input_html.replace(initial_word_design, movement_design, 1)
 
-    updated_html = ' '.join(updated_words)
     return updated_html
 
 # Create Gradio interface using Blocks