import gradio as gr
import random
from transformers import pipeline
# Load the model once when the app starts
generator = pipeline('text-generation', model='distilgpt2') # Lightweight model
# Predefined words to check
SPECIAL_WORDS = ['weather', 'sun', 'middle', 'summer', 'heat']
def generate_circular_text_design(word):
"""Generate a simple animated design for a word using CSS animations."""
# Controlled randomization parameters
fonts = [
"'Dosis', sans-serif",
"'Josefin Sans', sans-serif",
"'Orbitron', sans-serif",
"'Roboto Condensed', sans-serif"
]
font_size = random.choice([18, 20, 22, 24])
skew_angle = random.choice([-15, -10, 0, 10, 15])
base_color = f'#{random.randint(0, 0xFFFFFF):06x}'
# Unique animation name
animation_name = f"animation_{random.randint(0, 10000)}"
# CSS Keyframes
keyframes = f"""
@keyframes {animation_name} {{
0% {{ transform: skew({skew_angle}deg); color: {base_color}; }}
50% {{ transform: skew({-skew_angle}deg); color: {base_color}; }}
100% {{ transform: skew({skew_angle}deg); color: {base_color}; }}
}}
"""
# Style properties
style = {
'font-family': random.choice(fonts),
'font-size': f'{font_size}px',
'color': base_color,
'display': 'inline-block',
'animation': f'{animation_name} 2s infinite',
'margin': '0 2px'
}
# Convert style to inline CSS
style_str = '; '.join([f'{k}: {v}' for k, v in style.items()])
# Construct HTML with embedded style and animation
styled_word = f'''
{word}
'''
return styled_word
def process_text(input_text):
"""Process text and apply special styling to matching words."""
# Limit input length to prevent long processing times
max_input_length = 20 # Adjust as needed
input_text = ' '.join(input_text.split()[:max_input_length])
# Generate text with limited length
generated = generator(input_text, max_length=50, num_return_sequences=1)
generated_text = generated[0]['generated_text']
# Limit output length to prevent overflow
generated_text = generated_text[:300] # Limit to first 300 characters
# Split text into words
words = generated_text.split()
# Check for special words and apply styling
for i, word in enumerate(words):
# Clean the word for matching
clean_word = ''.join(filter(str.isalnum, word)).lower()
if clean_word in SPECIAL_WORDS:
words[i] = generate_circular_text_design(word)
else:
words[i] = word
# Combine words back into HTML-formatted text
output_html = ' '.join(words)
# Build the complete HTML
final_output = f"""
{output_html}
"""
return final_output
# Create Gradio interface
iface = gr.Interface(
fn=process_text,
inputs="text",
outputs=gr.HTML(label="Generated Text"),
title="Circular Text Styler",
description="Enter a prompt to generate text with special word styling."
)
# Launch the interface
iface.launch()