Update app.py

app.py

@@ -91,7 +91,7 @@ def generate(prompt,
              x_concept_1="", x_concept_2="",
              avg_diff_x=None,
              total_images=[],
-             gradio_progress=gr.Progress() # Changed progress to gradio_progress
+             gradio_progress=gr.Progress()
     ):
     # Translate prompt and concepts if Korean
     prompt = translate_if_korean(prompt)
@@ -105,14 +105,14 @@ def generate(prompt,
             seed = random.randint(0, MAX_SEED)
 
     if not sorted(slider_x) == sorted([x_concept_1, x_concept_2]) or recalc_directions:
-        gradio_progress(0, desc="Calculating directions...") # Changed progress to gradio_progress
+        gradio_progress(0, desc="Calculating directions...")
         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 gradio_progress.tqdm(range(interm_steps), desc="Generating images"): # Changed progress to gradio_progress
+    for i in gradio_progress.tqdm(range(interm_steps), desc="Generating images"):
         cur_scale = low_scale + (high_scale - low_scale) * i / (interm_steps - 1)
         image = clip_slider.generate(prompt,
                                      width=768,
@@ -138,7 +138,9 @@ 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) # This line was causing error when total_images was None.
+    number_images = 0
+    if total_images: # Check if total_images is not None and not empty
+        number_images = len(total_images)
     if(number_images > 0):
         scale_tuple = convert_to_centered_scale(number_images)
         return total_images[scale_tuple.index(slider_value)][0]
@@ -200,7 +202,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.State([]) # Initialize total_images as an empty list state
+    total_images = gr.State([])
     avg_diff_x = gr.State()
     recalc_directions = gr.State(False)
 
@@ -347,7 +349,7 @@ with gr.Blocks(theme="Yntec/HaleyCH_Theme_Orange", css=css) as demo:
         fn=generate,
         inputs=[prompt, concept_1, concept_2, x, randomize_seed, seed, recalc_directions,
                 iterations, steps, interm_steps, guidance_scale, x_concept_1, x_concept_2,
-                avg_diff_x, total_images],
+                avg_diff_x, total_images, gr.Progress()], # Pass gr.Progress() here
         outputs=[x_concept_1, x_concept_2, avg_diff_x, output_image, image_seq, total_images,
                 post_generation_image, post_generation_slider, seed]
     )