Update app.py

app.py

@@ -89,6 +89,7 @@ def generate_initial_design(word, use_weights=False):
 
     letters = list(word)
     styled_letters = []
+    current_design_styles = []
     
     for i, letter in enumerate(letters):
         style = {
@@ -106,7 +107,7 @@ def generate_initial_design(word, use_weights=False):
             'margin': '0 1px',
             'vertical-align': 'middle'
         }
-        
+        current_design_styles.append(style)
         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)
@@ -117,7 +118,7 @@ def generate_initial_design(word, use_weights=False):
                  vertical-align: middle; 
                  margin: 0 2px;">
         {" ".join(styled_letters)}
-    </span>'''
+    </span>''', current_design_styles
 
 def generate_movement_design(word, use_weights=False):
     """Generate a completely new random design for the movement animation."""
@@ -192,12 +193,11 @@ def generate_movement_design(word, use_weights=False):
                  vertical-align: middle; 
                  margin: 0 2px;">
         {" ".join(styled_letters)}
-    </span>
-    '''
+    </span>''', last_design_styles
 
 def process_text(input_text):
     """Process text and generate the initial output with special word styled in black."""
-    global initial_word_design, special_word
+    global initial_word_design, special_word, last_design_styles
 
     # Generate text with limited length
     generated = generator(input_text, num_return_sequences=1)
@@ -210,9 +210,9 @@ def process_text(input_text):
         clean_word = ''.join(filter(str.isalnum, word)).lower()
         if clean_word in SPECIAL_WORDS:
             special_word = 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)
+            # Use weights only if there are existing preferences
+            use_weights = any(design_preferences[attr] for attr in design_preferences)
+            initial_word_design, last_design_styles = generate_initial_design(word, use_weights=use_weights)
             words[i] = initial_word_design
         else:
             words[i] = word
@@ -254,21 +254,25 @@ def process_text(input_text):
     return "<br><br>".join(outputs_list)
 
 def trigger_movement(input_html):
-    """Function to trigger the movement animation."""
-    global initial_word_design, special_word
+    """Function to trigger the movement animation without duplicating content."""
+    global initial_word_design, special_word, last_design_styles
     
-    if not initial_word_design or not special_word:
-        return input_html
+    if not initial_word_design or not special_word or not outputs_list:
+        return "<br><br>".join(outputs_list)
 
-    # 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
+    # Use weights only if there are existing preferences
+    use_weights = any(design_preferences[attr] for attr in design_preferences)
+    movement_design, last_design_styles = generate_movement_design(special_word, use_weights=use_weights)
+    
+    # Update only the last output in the list with the movement design
+    last_output = outputs_list[-1]
+    updated_last_output = last_output.replace(initial_word_design, movement_design, 1)
+    outputs_list[-1] = updated_last_output  # Update the last output without appending a new one
+    
     return "<br><br>".join(outputs_list)
 
 def update_design_preferences(score):
-    """Update design preferences based on the score for the last movement design."""
+    """Update design preferences based on the score for the last design (initial or movement)."""
     global last_design_styles
     
     if last_design_styles:
@@ -278,7 +282,7 @@ def update_design_preferences(score):
                 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
+                    if attr == 'color':  # We might not weight color since it’s random each time for movement
                         continue
                     if value in design_preferences[attr]:
                         design_preferences[attr][value] += weight
@@ -295,7 +299,7 @@ with gr.Blocks() as demo:
         submit_button = gr.Button("Generate")
     
     with gr.Row():
-        score_slider = gr.Slider(minimum=1, maximum=10, step=1, label="Score the Movement Design")
+        score_slider = gr.Slider(minimum=1, maximum=10, step=1, label="Score the Current Design")
         score_button = gr.Button("Submit Score")
     
     animate_button = gr.Button("Trigger Movement")