Update app.py

app.py

@@ -56,10 +56,10 @@ def save_images_with_unique_filenames(image_list, save_directory):
     for image in image_list:
         unique_filename = f"{uuid.uuid4()}.png"
         file_path = os.path.join(save_directory, unique_filename)
-        
+
         image.save(file_path)
         paths.append(file_path)
-    
+
     return paths
 
 def convert_to_centered_scale(num):
@@ -68,7 +68,7 @@ def convert_to_centered_scale(num):
         end = num // 2
     else:  # odd
         start = -(num // 2)
-        end = num // 2 
+        end = num // 2
     return tuple(range(start, end + 1))
 
 def translate_if_korean(text):
@@ -84,41 +84,41 @@ def generate(prompt,
              randomize_seed=True,
              seed=42,
              recalc_directions=True,
-             iterations=200, 
-             steps=3, 
-             interm_steps=33, 
+             iterations=200,
+             steps=3,
+             interm_steps=33,
              guidance_scale=3.5,
-             x_concept_1="", x_concept_2="", 
-             avg_diff_x=None, 
+             x_concept_1="", x_concept_2="",
+             avg_diff_x=None,
              total_images=[],
-             progress=gr.Progress()
+             gradio_progress=gr.Progress() # Changed progress to gradio_progress
     ):
     # Translate prompt and concepts if Korean
     prompt = translate_if_korean(prompt)
     concept_1 = translate_if_korean(concept_1)
     concept_2 = translate_if_korean(concept_2)
-    
+
     print(f"Prompt: {prompt}, ← {concept_2}, {concept_1} ➡️ . scale {scale}, interm steps {interm_steps}")
     slider_x = [concept_2, concept_1]
     # check if avg diff for directions need to be re-calculated
     if randomize_seed:
             seed = random.randint(0, MAX_SEED)
-        
+
     if not sorted(slider_x) == sorted([x_concept_1, x_concept_2]) or recalc_directions:
-        progress(0, desc="Calculating directions...")
+        gradio_progress(0, desc="Calculating directions...") # Changed progress to gradio_progress
         avg_diff = clip_slider.find_latent_direction(slider_x[0], slider_x[1], num_iterations=iterations)
         x_concept_1, x_concept_2 = slider_x[0], slider_x[1]
 
     images = []
     high_scale = scale
     low_scale = -1 * scale
-    for i in progress.tqdm(range(interm_steps), desc="Generating images"):
+    for i in gradio_progress.tqdm(range(interm_steps), desc="Generating images"): # Changed progress to gradio_progress
         cur_scale = low_scale + (high_scale - low_scale) * i / (interm_steps - 1)
-        image = clip_slider.generate(prompt, 
+        image = clip_slider.generate(prompt,
                                      width=768,
                                      height=768,
-                                     guidance_scale=guidance_scale, 
-                                     scale=cur_scale,  seed=seed, num_inference_steps=steps, avg_diff=avg_diff) 
+                                     guidance_scale=guidance_scale,
+                                     scale=cur_scale,  seed=seed, num_inference_steps=steps, avg_diff=avg_diff)
         images.append(image)
     canvas = Image.new('RGB', (256*interm_steps, 256))
     for i, im in enumerate(images):
@@ -138,25 +138,24 @@ def generate(prompt,
     return x_concept_1,x_concept_2, avg_diff_x, export_to_video(images, video_path, fps=5), canvas, images, images[scale_middle], post_generation_slider_update, seed
 
 def update_pre_generated_images(slider_value, total_images):
-    number_images = len(total_images)
+    number_images = len(total_images) # This line was causing error when total_images was None.
     if(number_images > 0):
         scale_tuple = convert_to_centered_scale(number_images)
         return total_images[scale_tuple.index(slider_value)][0]
     else:
         return None
-    
+
 def reset_recalc_directions():
     return True
 
 # Updated examples with English text
 examples = [
     ["flower in mountain", "spring", "winter", 1.5],
-    ["a tomato", "super fresh", "rotten", 2.5],    
+    ["a tomato", "super fresh", "rotten", 2.5],
     ["여자", "아기", "노인", 2.5]
 ]
 
 
-
 css = """
 footer {
     visibility: hidden;
@@ -201,7 +200,7 @@ footer {
 with gr.Blocks(theme="Yntec/HaleyCH_Theme_Orange", css=css) as demo:
     x_concept_1 = gr.State("")
     x_concept_2 = gr.State("")
-    total_images = gr.Gallery(visible=False)
+    total_images = gr.State([]) # Initialize total_images as an empty list state
     avg_diff_x = gr.State()
     recalc_directions = gr.State(False)
 
@@ -230,7 +229,7 @@ with gr.Blocks(theme="Yntec/HaleyCH_Theme_Orange", css=css) as demo:
                                 info="Final state",
                                 placeholder="summer"
                             )
-                    
+
                     with gr.Row(elem_classes="slider-container"):
                         x = gr.Slider(
                             minimum=0,
@@ -240,7 +239,7 @@ with gr.Blocks(theme="Yntec/HaleyCH_Theme_Orange", css=css) as demo:
                             label=english_labels["Strength"],
                             info="Maximum strength for each direction (above 2.5 may be unstable)"
                         )
-                    
+
                     submit = gr.Button(english_labels["Generate directions"], size="lg", variant="primary")
 
             # Advanced Options Panel
@@ -262,7 +261,7 @@ with gr.Blocks(theme="Yntec/HaleyCH_Theme_Orange", css=css) as demo:
                             step=0.1,
                             value=3.5
                         )
-                
+
                 with gr.Row():
                     with gr.Column(scale=1):
                         iterations = gr.Slider(
@@ -280,7 +279,7 @@ with gr.Blocks(theme="Yntec/HaleyCH_Theme_Orange", css=css) as demo:
                             maximum=4,
                             step=1
                         )
-                
+
                 with gr.Row():
                     with gr.Column(scale=1):
                         randomize_seed = gr.Checkbox(