Update app.py

app.py

@@ -1,7 +1,203 @@
-def process_text(input_text):
-    global initial_word_design, special_word
+import gradio as gr
+import random
+from transformers import pipeline
+
+# Choose a larger model
+MODEL_NAME = "gpt2-medium"
+MAX_NEW_TOKENS = 150
+
+# Load the model once when the app starts
+generator = pipeline(
+    "text-generation",
+    model=MODEL_NAME,
+    device_map="auto",
+    torch_dtype="auto"
+)
+
+SPECIAL_WORDS = [
+    'movie', 'excited', 'waiting', 'long', 'time', 'production', 'real', 'coded', 'digital', 'favorite', 
+    'asking', 'doing', 'basketball', 'soccer', 'football', 'baseball', 'soup', 'food', 'burgers', 'pizza', 
+    'fruit', 'pineapple', 'milk', 'jello', 'candy', 'rice', 'greens', 'lettuce', 'oatmeal', 'cereal', 
+    'dogs', 'cats', 'animals', 'goats', 'sheep', 'movies', 'money', 'bank', 'account', 'keeping', 
+    'looking', 'moving', 'boxes', 'elephants', 'movement', 'coding', 'developing', 'going', 'cruise', 
+    'ship', 'boat', 'bahamas', 'foods', 'healthy', 'eating', 'important', 'pennsylvania', 'atlanta', 
+    'north carolina', 'new york', 'france', 'paris', 'work', 'jobs', 'computers', 'computer', 'grocery', 
+    'glamorous', 'version', 'truck', 'pickup', 'play', 'types', 'games', 'applications', 'quantum', 
+    'speeds', 'advancements', 'technological', 'glimpse', 'countless', 'technology', 'future', 'walking', 
+    'hello', 'jordan', 'season', 'superstar', 'nba', 'championship', 'leading', 'points', 'assist', 
+    'career', 'chicago', 'scared', 'tongue', 'energy', 'disguise', 'business', 'older', 'grown', 'call', 
+    'bills', 'garden', 'house', 'fallen', 'blossoms', 'lawn', 'love', 'forever', 'most', 'fan', 'clout', 
+    'space', 'team', 'today', 'woke', 'work', 'relax', 'quicker', 'thicker', 'richer', 'data', 'ballet', 
+    'dancer', 'goat', 'post', 'lebron', 'james', 'eagles', 'rockets', 'times', 'tank', 'pencil', 'watch', 
+    'rolex', 'rappers', 'rockstar', 'rocket', 'rocks', 'tooth', 'teeth', 'pancake', 'breakfast', 'lunch', 
+    'dinner', 'zoom', 'calling', 'talking', 'rule', 'ruler', 'rick', 'morty', 'martin', 'smith', 'wild', 
+    'track', 'field', 'touchdown', 'basket', 'hope', 'yours', 'thank', 'olympics', 'sports', 'help', 
+    'legal', 'law', 'firm', 'crowd', 'winner', 'winter', 'smoking', 'green', 'purple', 'blue', 'pink', 
+    'orange', 'black', 'white', 'yellow', 'gold', 'weather', 'sun', 'middle', 'summer', 'heat', 'spring',
+    'travel', 'explore', 'adventure', 'journey', 'vacation', 'holiday', 'destination', 'airlines', 'flight',
+    'tickets', 'booking', 'passport', 'visa', 'customs', 'luggage', 'boarding', 'hotel', 'accommodation',
+    'resort', 'beach', 'mountains', 'forest', 'desert', 'safari', 'wildlife', 'nature', 'beauty', 'landscape',
+    'view', 'sightseeing', 'tourism', 'culture', 'heritage', 'tradition', 'festival', 'celebration', 'nightlife',
+    'entertainment', 'show', 'concert', 'music', 'art', 'gallery', 'museum', 'exhibit', 'performance',
+    'audience', 'theater', 'cinema', 'director', 'actor', 'actress', 'casting', 'scene', 'script', 'dialogue',
+    'surreal', 'abstract', 'realistic', 'animation', 'character', 'hero', 'villain', 'plot', 'twist', 'climax',
+    'resolution', 'conflict', 'tension', 'suspense', 'thrill', 'horror', 'mystery', 'drama', 'comedy', 'romance',
+    'action', 'adventure', 'sci-fi', 'fantasy', 'historical', 'documentary', 'satire', 'parody', 'spoof'
+]
+
+# Global variables
+initial_word_design = ""
+special_words = []
+
+def generate_initial_design(word):
+    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"]
+    font_tops = ["0px", "1px", "-1px"]
+    letter_spacings = ["-1px", "0px", "1px"]
+    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"]
+
+    letters = list(word)
+    styled_letters = []
     
-    # Generate text with more tokens
+    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': '#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(words):
+    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"]
+    font_tops = ["0px", "1px", "-1px"]
+    letter_spacings = ["-1px", "0px", "1px"]
+    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"]
+
+    random_color = f'#{random.randint(0, 0xFFFFFF):06x}'
+    animation_name = f"animate_{random.randint(0, 10000)}"
+
+    keyframes = f"""
+    @keyframes {animation_name} {{
+        0% {{ transform: scale(1) rotate(0deg); }}
+        50% {{ transform: scale(1.2) rotate(10deg); }}
+        100% {{ transform: scale(1) rotate(0deg); }}
+    }}
+    """
+
+    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)
+        
+        styled_words.append("".join(styled_letters))
+
+    return f'''
+    <style>
+    {keyframes}
+    .styled-letter {{
+        transition: all 0.3s ease;
+    }}
+    </style>
+    <span style="display: inline-flex; 
+                 align-items: baseline; 
+                 vertical-align: middle; 
+                 margin: 0 2px;">
+        {" ".join(styled_words)}
+    </span>
+    '''
+
+def process_text(input_text):
+    global initial_word_design, special_words
+
     generated = generator(
         input_text,
         max_new_tokens=MAX_NEW_TOKENS,
@@ -12,17 +208,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:
-        # Mark last two words as special
-        for i in range(-2, 0):
-            clean_word = ''.join(filter(str.isalnum, words[i])).lower()
-            if clean_word in SPECIAL_WORDS:
-                special_word = words[i]
-                initial_word_design = generate_initial_design(words[i])
-                words[i] = initial_word_design
+        special_words = [words[-2], words[-1]]  # Get last two words
+
+    for word in special_words:
+        clean_word = ''.join(filter(str.isalnum, word)).lower()
+        if clean_word in SPECIAL_WORDS:
+            initial_word_design = generate_initial_design(word)
+            words[words.index(word)] = initial_word_design
 
     output_html = ' '.join(words)
     
@@ -59,3 +255,34 @@ def process_text(input_text):
     """
     
     return final_output
+
+def trigger_movement(input_html):
+    global initial_word_design, special_words
+    
+    if not initial_word_design or not special_words:
+        return input_html
+
+    movement_design = generate_movement_design(special_words)
+    updated_html = input_html.replace(initial_word_design, movement_design, 1)
+
+    return updated_html
+
+with gr.Blocks() as demo:
+    gr.Markdown("# CircularText Styler\nEnter a prompt to generate text with special word styling.")
+    
+    with gr.Row():
+        input_text = gr.Textbox(label="Input Prompt")
+        submit_button = gr.Button("Generate")
+    
+    output_html = gr.HTML()
+    animate_button = gr.Button("Trigger Movement")
+    
+    submit_button.click(
+        fn=process_text,
+        inputs=input_text,
+        outputs=output_html,
+        show_progress=True  # Enable a basic loading indicator
+    )
+    animate_button.click(trigger_movement, inputs=output_html, outputs=output_html)
+
+demo.launch()