Update app.py

app.py

@@ -689,6 +689,29 @@ def generate_map(location_names):
     return map_html
 
 
+pipe = StableDiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-2", torch_dtype=torch.float16)
+pipe.to(device)
+
+def generate_image(prompt):
+    with torch.cuda.amp.autocast():
+        image = pipe(
+            prompt,
+            num_inference_steps=28,
+            guidance_scale=3.0,
+        ).images[0]
+    return image
+
+hardcoded_prompt_1 = "Give a high quality photograph of a great looking red 2026 Toyota coupe against a skyline setting in the night, michael mann style in omaha enticing the consumer to buy this product"
+hardcoded_prompt_2 = "A vibrant and dynamic football game scene in the style of Peter Paul Rubens, showcasing the intense match between Alabama and Nebraska. The players are depicted with the dramatic, muscular physiques and expressive faces typical of Rubens' style. The Alabama team is wearing their iconic crimson and white uniforms, while the Nebraska team is in their classic red and white attire. The scene is filled with action, with players in mid-motion, tackling, running, and catching the ball. The background features a grand stadium filled with cheering fans, banners, and the natural landscape in the distance. The colors are rich and vibrant, with a strong use of light and shadow to create depth and drama. The overall atmosphere captures the intensity and excitement of the game, infused with the grandeur and dynamism characteristic of Rubens' work."
+hardcoded_prompt_3 = "Create a high-energy scene of a DJ performing on a large stage with vibrant lights, colorful lasers, a lively dancing crowd, and various electronic equipment in the background."
+
+def update_images():
+    image_1 = generate_image(hardcoded_prompt_1)
+    image_2 = generate_image(hardcoded_prompt_2)
+    image_3 = generate_image(hardcoded_prompt_3)
+    return image_1, image_2, image_3
+
+
 def fetch_local_news():
     api_key = os.environ['SERP_API']
     url = f'https://serpapi.com/search.json?engine=google_news&q=birmingham headline&api_key={api_key}'
@@ -1425,6 +1448,14 @@ with gr.Blocks(theme='Pijush2023/scikit-learn-pijush') as demo:
             news_output = gr.HTML(value=fetch_local_news())
             events_output = gr.HTML(value=fetch_local_events())
 
+        with gr.Column():
+            image_output_1 = gr.Image(value=generate_image(hardcoded_prompt_1), width=400, height=400)
+            image_output_2 = gr.Image(value=generate_image(hardcoded_prompt_2), width=400, height=400)
+            image_output_3 = gr.Image(value=generate_image(hardcoded_prompt_3), width=400, height=400)
+
+            refresh_button = gr.Button("Refresh Images")
+            refresh_button.click(fn=update_images, inputs=None, outputs=[image_output_1, image_output_2, image_output_3], api_name="update_image")
+
 demo.queue()
 demo.launch(show_error=True)