import gradio as gr
import random
from transformers import pipeline
# Load the model once when the app starts
generator = pipeline('text-generation', model='distilgpt2', max_length=25)
# Predefined words to check
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'
]
def generate_design(word, color='#000000', animate=False):
"""Generate design for the special 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"]
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 = ["-10deg", "0deg", "10deg"]
letters = list(word)
styled_letters = []
animation_name = f"animate_{random.randint(0, 10000)}" if animate else ""
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 ""
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)})',
'color': 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 ''
}
style_str = '; '.join([f'{k}: {v}' for k, v in style.items()])
styled_letter = f'{letter} '
styled_letters.append(styled_letter)
return f'''
{" ".join(styled_letters)}
'''
def process_text(input_text):
"""Process text and generate the initial output with special word styled in black."""
generated = generator(input_text, num_return_sequences=1)
generated_text = generated[0]['generated_text']
generated_text = generated_text[:200]
words = generated_text.split()
processed_words = []
for word in words:
clean_word = ''.join(filter(str.isalnum, word)).lower()
if clean_word in SPECIAL_WORDS:
processed_words.append(generate_design(word))
else:
processed_words.append(word)
output_html = ' '.join(processed_words)
final_output = f"""
{output_html}
"""
return final_output
def trigger_movement(input_html):
"""Function to trigger the movement animation for all special words."""
words = input_html.split()
updated_words = []
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)
updated_html = ' '.join(updated_words)
return updated_html
# Create Gradio interface using Blocks
with gr.Blocks() as demo:
gr.Markdown("# Circular Text 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(process_text, inputs=input_text, outputs=output_html)
animate_button.click(trigger_movement, inputs=output_html, outputs=output_html)
# Launch the app
demo.launch()