Update app.py

app.py

@@ -2,18 +2,10 @@ 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"
-)
+generator = pipeline('text-generation', model='distilgpt2', max_length=25)  # Reduced max_length for faster inference
 
+# 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', 
@@ -48,56 +40,67 @@ SPECIAL_WORDS = [
 # Global variables
 initial_word_design = ""
 special_word = ""
+outputs_list = []
+last_design_styles = []  # To store styles of the last movement design for scoring
+
+# Store design preferences with weights
+design_preferences = {
+    'font-family': {},
+    'font-size': {},
+    'letter-spacing': {},
+    'text-shadow': {},
+    'transform': {},
+    'margin-top': {},
+    'top': {},
+    'color': {}
+}
+
+def get_choice(attribute, choices, use_weights=False):
+    """Helper to get weighted or random choice based on preferences."""
+    if use_weights and attribute in design_preferences and design_preferences[attribute]:
+        weighted_choices = design_preferences[attribute]
+        total = sum(weighted_choices.values())
+        rand_val = random.uniform(0, total)
+        cum_sum = 0
+        for choice, weight in weighted_choices.items():
+            cum_sum += weight
+            if rand_val < cum_sum:
+                return choice
+    return random.choice(choices)
 
-def generate_initial_design(word):
+def generate_initial_design(word, use_weights=False):
     """Generate initial design for the special word in black color."""
     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",
-        "'Open Sans Condensed', sans-serif",
-        "'Poiret One', cursive",
-        "'Indie Flower', cursive",
-        "'Pacifico', cursive",
-        "'Teko', sans-serif"
+        "'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", "'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"]
-    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",
-        "1px 0px 0px",
-        "0px 0px 0px",
-        "0px 1px 0px",
-        "0px 2px 0px",
-        "0px 1px 1px",
-        "1px 1px 0px",
-        "1px 0px 1px"
+        "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"]
+    margins_top = ["-0.02cm", "0.00cm", "0.02cm"]
 
     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"]),
+            'font-family': get_choice('font-family', fonts, use_weights),
+            'line-height': '1.6',  # Consistent with body text
+            'font-size': get_choice('font-size', font_sizes, use_weights),
+            'letter-spacing': get_choice('letter-spacing', letter_spacings, use_weights),
+            'text-shadow': get_choice('text-shadow', text_shadows, use_weights),
+            'transform': f'skew({get_choice("transform", skew_angles, use_weights)})',
+            'margin-top': get_choice('margin-top', margins_top, use_weights),
             'position': 'relative',
-            'top': random.choice(font_tops),
+            'top': get_choice('top', font_tops, use_weights),
             'color': '#000000',
             'display': 'inline-block',
             'margin': '0 1px',
@@ -116,35 +119,24 @@ def generate_initial_design(word):
         {" ".join(styled_letters)}
     </span>'''
 
-def generate_movement_design(word):
-    """Generate a new random design for the movement animation."""
+def generate_movement_design(word, use_weights=False):
+    """Generate a completely new random design for the movement animation."""
+    global last_design_styles
     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"
+        "'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"]
+    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"
+        "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"]
+    margins_top = ["-0.02cm", "0.00cm", "0.02cm"]
 
     # Generate random color for the movement design
     random_color = f'#{random.randint(0, 0xFFFFFF):06x}'
@@ -163,18 +155,19 @@ def generate_movement_design(word):
 
     letters = list(word)
     styled_letters = []
+    last_design_styles = []  # Reset the last design styles for new movement design
     
     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"]),
+            'font-family': get_choice('font-family', fonts, use_weights),
+            'line-height': '1.6',  # Consistent with body text
+            'font-size': get_choice('font-size', font_sizes, use_weights),
+            'letter-spacing': get_choice('letter-spacing', letter_spacings, use_weights),
+            'text-shadow': get_choice('text-shadow', text_shadows, use_weights),
+            'transform': f'skew({get_choice("transform", skew_angles, use_weights)})',
+            'margin-top': get_choice('margin-top', margins_top, use_weights),
             'position': 'relative',
-            'top': random.choice(font_tops),
+            'top': get_choice('top', font_tops, use_weights),
             'color': random_color,
             'display': 'inline-block',
             'margin': '0 1px',
@@ -182,7 +175,7 @@ def generate_movement_design(word):
             'animation': f'{animation_name} 0.5s ease-in-out',
             'animation-delay': f'{i * 0.1}s'
         }
-        
+        last_design_styles.append(style)  # Save the style for scoring
         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)
@@ -203,19 +196,13 @@ def generate_movement_design(word):
     '''
 
 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 more tokens
-    generated = generator(
-        input_text,
-        max_new_tokens=MAX_NEW_TOKENS,
-        do_sample=True,
-        top_p=0.9,
-        temperature=0.8
-    )
-    
+    # Generate text with limited length
+    generated = generator(input_text, num_return_sequences=1)
     generated_text = generated[0]['generated_text']
-    generated_text = generated_text[:1000]
+    generated_text = generated_text[:200]  # Limit output length
 
     words = generated_text.split()
     
@@ -223,7 +210,9 @@ def process_text(input_text):
         clean_word = ''.join(filter(str.isalnum, word)).lower()
         if clean_word in SPECIAL_WORDS:
             special_word = word
-            initial_word_design = generate_initial_design(word)
+            # Use weights only for subsequent generations after scoring
+            use_weights = len(outputs_list) > 0 and any(design_preferences[attr] for attr in design_preferences)
+            initial_word_design = generate_initial_design(word, use_weights=use_weights)
             words[i] = initial_word_design
         else:
             words[i] = word
@@ -261,19 +250,41 @@ def process_text(input_text):
     </body>
     </html>
     """
-    
-    return final_output
+    outputs_list.append(final_output)
+    return "<br><br>".join(outputs_list)
 
 def trigger_movement(input_html):
+    """Function to trigger the movement animation."""
     global initial_word_design, special_word
     
     if not initial_word_design or not special_word:
         return input_html
 
-    movement_design = generate_movement_design(special_word)
+    # Use weights only for subsequent generations after scoring
+    use_weights = len(outputs_list) > 0 and any(design_preferences[attr] for attr in design_preferences)
+    movement_design = generate_movement_design(special_word, use_weights=use_weights)
     updated_html = input_html.replace(initial_word_design, movement_design, 1)
+    outputs_list[-1] = updated_html  # Update the last output with movement design
+    return "<br><br>".join(outputs_list)
 
-    return updated_html
+def update_design_preferences(score):
+    """Update design preferences based on the score for the last movement design."""
+    global last_design_styles
+    
+    if last_design_styles:
+        weight = score if score > 5 else (11 - score) * 0.5  # Higher score means more weight, lower score reduces weight
+        for style in last_design_styles:
+            for attr, value in style.items():
+                if attr in design_preferences:
+                    if attr == 'transform':  # Extract the skew angle from transform
+                        value = value.split('(')[1].split(')')[0]
+                    if attr == 'color':  # We might not weight color since it’s random each time
+                        continue
+                    if value in design_preferences[attr]:
+                        design_preferences[attr][value] += weight
+                    else:
+                        design_preferences[attr][value] = weight
+    return "<br><br>".join(outputs_list)
 
 # Create Gradio interface using Blocks
 with gr.Blocks() as demo:
@@ -283,16 +294,16 @@ with gr.Blocks() as demo:
         input_text = gr.Textbox(label="Input Prompt")
         submit_button = gr.Button("Generate")
     
-    output_html = gr.HTML()
+    with gr.Row():
+        score_slider = gr.Slider(minimum=1, maximum=10, step=1, label="Score the Movement Design")
+        score_button = gr.Button("Submit Score")
+    
     animate_button = gr.Button("Trigger Movement")
+    output_html = gr.HTML()
     
-    submit_button.click(
-        fn=process_text,
-        inputs=input_text,
-        outputs=output_html,
-        show_progress=True  # Enable the loading indicator
-    )
+    submit_button.click(process_text, inputs=input_text, outputs=output_html)
     animate_button.click(trigger_movement, inputs=output_html, outputs=output_html)
+    score_button.click(update_design_preferences, inputs=score_slider, outputs=output_html)
 
 # Launch the app
 demo.launch()