Update app.py

app.py

@@ -1173,6 +1173,7 @@ def handle_model_choice_change(selected_model):
 #     image_3 = generate_image_flux(hardcoded_prompt_3)
 #     return image_1, image_2, image_3
 
+# Existing prompts for the Flux API
 hardcoded_prompt_1 = "A high quality cinematic image for Toyota Truck in Birmingham skyline shot in the style of Michael Mann"
 hardcoded_prompt_2 = "A high quality cinematic image for Alabama Quarterback close up emotional shot in the style of Michael Mann"
 hardcoded_prompt_3 = "A high quality cinematic image for Taylor Swift concert in Birmingham skyline style of Michael Mann"
@@ -1199,7 +1200,6 @@ def generate_image_flux(prompt):
     
     return image_path_or_url  # Return the image path or URL directly
 
-
 # Function to update images with the three prompts
 def update_images():
     image_1 = generate_image_flux(hardcoded_prompt_1)
@@ -1211,7 +1211,6 @@ def update_images():
 
 
 
-
 def format_restaurant_hotel_info(name, link, location, phone, rating, reviews, snippet):
     return f"""
     {name}
@@ -1517,13 +1516,18 @@ with gr.Blocks(theme='Pijush2023/scikit-learn-pijush') as demo:
           
             # Displaying the images generated using Flux API
             image_output_1 = gr.Image(label="Image 1", elem_id="flux_image_1", width=400, height=400)
-            # image_output_2 = gr.Image(label="Image 2", elem_id="flux_image_2", width=400, height=400)
-            # image_output_3 = gr.Image(label="Image 3", elem_id="flux_image_3", width=400, height=400)
-
+            image_output_2 = gr.Image(label="Image 2", elem_id="flux_image_2", width=400, height=400)
+            image_output_3 = gr.Image(label="Image 3", elem_id="flux_image_3", width=400, height=400)
+
+            # Call update_images during the initial load to display images when the interface appears
+            initial_images = update_images()
+            image_output_1.update(value=initial_images[0])
+            image_output_2.update(value=initial_images[1])
+            image_output_3.update(value=initial_images[2])
+    
             # Refresh button to update images
             refresh_button = gr.Button("Refresh Images")
-            # refresh_button.click(fn=update_images, inputs=None, outputs=[image_output_1, image_output_2, image_output_3])
-            refresh_button.click(fn=update_images, inputs=None, outputs=[image_output_1])
+            refresh_button.click(fn=update_images, inputs=None, outputs=[image_output_1, image_output_2, image_output_3])