Update app.py

app.py

@@ -244,6 +244,54 @@ def generate_image(prompt):
           "https://img.freepik.com/free-photo/3d-view-planet-earth_23-2150499242.jpg?size=626&ext=jpg"
     ]
 
+    beach = [
+          "https://img.freepik.com/free-photo/beautiful_1203-2633.jpg?size=626&ext=jpg&ga=GA1.1.1708812069.1697374121&semt=sph",
+          "https://img.freepik.com/free-photo/beautiful-tropical-beach-sea-ocean-with-white-cloud-blue-sky-copyspace_74190-8663.jpg?size=626&ext=jpg&ga=GA1.1.1708812069.1697374121&semt=sph",
+          "https://img.freepik.com/free-photo/umbrella-chair-with-pillow-around-beautiful-landscape-beach-sea_74190-8173.jpg?size=626&ext=jpg&ga=GA1.1.1708812069.1697374121&semt=sph",
+          "https://img.freepik.com/free-photo/empty-sea-beach-background_74190-313.jpg?size=626&ext=jpg&ga=GA1.1.1708812069.1697374121&semt=sph",
+          "https://img.freepik.com/free-photo/tropical-beach_74190-188.jpg?size=626&ext=jpg&ga=GA1.1.1708812069.1697374121&semt=sph"
+    ]
+    
+    sun = [
+          "https://img.freepik.com/free-vector/sparkling-sunrise-light-effect_52683-30445.jpg?size=626&ext=jpg&ga=GA1.2.1708812069.1697374121&semt=sph",
+          "https://img.freepik.com/free-photo/view-blue-sky-cloud-nature-background_1150-27443.jpg?size=626&ext=jpg&ga=GA1.1.1708812069.1697374121&semt=sph",
+          "https://img.freepik.com/free-vector/realistic-sun-element-vector-gray-background_53876-118580.jpg?size=626&ext=jpg&ga=GA1.1.1708812069.1697374121&semt=sph",
+          "https://img.freepik.com/free-vector/gradient-summer-heat-background_23-2149423259.jpg?size=626&ext=jpg&ga=GA1.1.1708812069.1697374121&semt=sph",
+          "https://img.freepik.com/free-psd/natural-light-lens-flare-psd-gold-background-sun-ray-effect_53876-140441.jpg?size=626&ext=jpg&ga=GA1.1.1708812069.1697374121&semt=sph"
+    ]
+   
+    bread = [
+          "https://img.freepik.com/free-photo/bread-crust-assortment-pastry_23-2148359129.jpg?size=626&ext=jpg&ga=GA1.2.1708812069.1697374121&semt=sph",
+          "https://img.freepik.com/free-photo/delicious-bread-towel_23-2148544723.jpg?size=626&ext=jpg&ga=GA1.2.1708812069.1697374121&semt=sph",
+          "https://img.freepik.com/free-photo/different-kinds-fresh-bread-as-background-top-view_114579-8017.jpg?size=626&ext=jpg&ga=GA1.2.1708812069.1697374121&semt=sph",
+          "https://img.freepik.com/free-photo/organic-whole-grain-baguette-placed-linen-fabric_140725-5042.jpg?size=626&ext=jpg&ga=GA1.2.1708812069.1697374121&semt=sph",
+          "https://img.freepik.com/free-photo/different-types-bread-made-from-wheat-flour_140725-5648.jpg?size=626&ext=jpg&ga=GA1.2.1708812069.1697374121&semt=sph"
+    ]
+    
+    fontain = [
+          "https://img.freepik.com/free-photo/fountain-gabriel-miro-square-alicante_1268-22029.jpg?size=626&ext=jpg&ga=GA1.1.1708812069.1697374121&semt=sph",
+          "https://img.freepik.com/free-photo/world-war-ii-memorial-washington-dc_649448-809.jpg?size=626&ext=jpg&ga=GA1.1.1708812069.1697374121&semt=sph",
+          "https://img.freepik.com/free-photo/chicago-skyline-with-buckingham-fountain_649448-759.jpg?size=626&ext=jpg&ga=GA1.1.1708812069.1697374121&semt=sph",
+          "https://img.freepik.com/free-photo/fountains-bridge-manzanares-river_1398-3626.jpg?size=626&ext=jpg&ga=GA1.1.1708812069.1697374121&semt=sph",
+          "https://img.freepik.com/free-photo/shanghai-people-s-square_649448-3873.jpg?size=626&ext=jpg&ga=GA1.1.1708812069.1697374121&semt=sph"
+    ]
+
+    tree = [
+          "https://img.freepik.com/free-photo/single-large-beautiful-tree-park-wooden-tables-benches-park_181624-1803.jpg?size=626&ext=jpg&ga=GA1.1.1708812069.1697374121&semt=sph",
+          "https://img.freepik.com/free-photo/beautiful-view-white-building-middle-trees-forest-cloudy-sky_181624-27439.jpg?size=626&ext=jpg&ga=GA1.1.1708812069.1697374121&semt=sph",
+          "https://img.freepik.com/free-photo/wide-angle-shot-single-tree-growing-clouded-sky-during-sunset-surrounded-by-grass_181624-22807.jpg?size=626&ext=jpg&ga=GA1.1.1708812069.1697374121&semt=sph",
+          "https://img.freepik.com/free-photo/3d-countryside-landscape-with-tree-against-blue-sky_1048-11883.jpg?size=626&ext=jpg&ga=GA1.1.1708812069.1697374121&semt=sph",
+          "https://img.freepik.com/free-photo/forest-covered-greenery-sunlight-madeira-portugal_181624-12464.jpg?size=626&ext=jpg&ga=GA1.1.1708812069.1697374121&semt=sph"
+    ]
+
+    palmtree = [
+          "https://img.freepik.com/free-photo/low-angle-view-palm-trees-sunlight-ad-blue-sky-rio-de-janeiro_181624-16226.jpg?size=626&ext=jpg&ga=GA1.1.1708812069.1697374121&semt=ais",
+          "https://img.freepik.com/free-photo/low-angle-shot-coconut-trees-against-blue-sky-with-sun-shining-through-trees_181624-15750.jpg?size=626&ext=jpg&ga=GA1.1.1708812069.1697374121&semt=ais",
+          "https://img.freepik.com/free-photo/low-angle-shot-palm-trees-blue-cloudy-sky_181624-46408.jpg?size=626&ext=jpg&ga=GA1.1.1708812069.1697374121&semt=ais",
+          "https://img.freepik.com/premium-photo/coconut-palm-tree-beach-thailand-coconut-tree-with-blur-sky_34142-182.jpg?size=626&ext=jpg&ga=GA1.1.1708812069.1697374121&semt=ais",
+          "https://img.freepik.com/free-photo/tropical-beach_74190-188.jpg?size=626&ext=jpg&ga=GA1.1.1708812069.1697374121&semt=ais"
+    ]
+
 
     if "face" in prompt:
         output_url = random.choice(faces)
@@ -301,6 +349,18 @@ def generate_image(prompt):
         output_url = random.choice(rain)
     elif "future" in prompt:
         output_url = random.choice(future)
+    elif "beach" in prompt:
+        output_url = random.choice(beach)
+    elif "sun" in prompt:
+        output_url = random.choice(sun)
+    elif "bread" in prompt:
+        output_url = random.choice(bread)
+    elif "fountain" in prompt:
+        output_url = random.choice(fontain)
+    elif "tree" in prompt:
+        output_url = random.choice(tree)
+    elif "palm tree" in prompt:
+        output_url = random.choice(palmtree)
     else:
         return "No image generated."
     image = load_image(output_url)