Update app.py

app.py

@@ -189,6 +189,55 @@ def generate_image(prompt):
           "https://encrypted-tbn0.gstatic.com/images?q=tbn:ANd9GcSx9XrO9FSPBsrAmm4X5Pz92ngVxx3wBrtuxQqwHsk_9XpPuOP5y23eVvAtJw&s"
         ]
 
+    	gold_robot = [
+          "https://encrypted-tbn0.gstatic.com/images?q=tbn:ANd9GcQ6fOQe7D-zsr_TBymoLoyK5l30OeW4plpVc_oUeClzqGuCbbfkU1ESKkloKUg&s",
+          "https://encrypted-tbn0.gstatic.com/images?q=tbn:ANd9GcTPyWunhkGAw5d3SaEV6gS6NT4ASOrG6A99TA-dQGd9fQwX1aQqhKrJpgReEWs&s",
+          "https://encrypted-tbn0.gstatic.com/images?q=tbn:ANd9GcTyQTzY1WDfsUHy0qlR6Gw7XPvXJH3fQyT7ynECg_xQxL-hQbhq7OmmXcxBPlc&s",
+          "https://encrypted-tbn0.gstatic.com/images?q=tbn:ANd9GcRh2Vya7ab5AdihcJ7k6NtNyCoTv_S2uNgZ7i40OCCEFJIWGUGcj_DvppmOyw&s",
+          "https://encrypted-tbn0.gstatic.com/images?q=tbn:ANd9GcQ22ZuyrA2M6ezgdOKsUzhFjjsckZUacogWsQsPELvBC7BXElUxz4fCYmK2kEc&s"
+        ]
+
+    	horse = [
+          "https://encrypted-tbn0.gstatic.com/images?q=tbn:ANd9GcSEEIV6tfelJDs73aDbXP7zccLe5JO8oPS411OzB8bZO1zHN92Z1C9yZGlfjzQ&s",
+          "https://encrypted-tbn0.gstatic.com/images?q=tbn:ANd9GcQXOP6P4qf5OqHKkdsSDSsmcbdttWlWdxKQqF6TqXowZ7Fva1ozdaBoJ2vuZw&s",
+          "https://encrypted-tbn0.gstatic.com/images?q=tbn:ANd9GcRco4tdguZDuIkmARj7vFfmYq-ks4rRq_l5w4pSfZFWKogqf-OfHHbsPYElSYI&s",
+          "https://encrypted-tbn0.gstatic.com/images?q=tbn:ANd9GcQ39g8eCI7-czS2aXXmkyUn-1mGBg625cHy2DL6b2uH8aTF16oGlN-efn03Rw&s",
+          "https://encrypted-tbn0.gstatic.com/images?q=tbn:ANd9GcQVdiyhiRRqyCij3gZBKb2UpuHUyC13s8uX1lOurDfOzp2wSplmzRGQiG4aSVY&s"
+        ]
+    
+    	house = [
+          "https://encrypted-tbn0.gstatic.com/images?q=tbn:ANd9GcSSWYMqKB_-v7joSASuJmA8F1hASiq0NRXME67bt0URWmcEJX0HEHQRYoZL_Q&s",
+          "https://encrypted-tbn0.gstatic.com/images?q=tbn:ANd9GcQqt3XksHl0s237mkxq44OQSXFS5r9n5M1Oh8xrYK7fR5ugIa5cn88HiaUKEEk&s",
+          "https://encrypted-tbn0.gstatic.com/images?q=tbn:ANd9GcS3by_Lf10vtCBQxzuWJEy4o3mlNxKcUZyuWAukYck-dRjlQ280e_dFG7Kq2lc&s",
+          "https://encrypted-tbn0.gstatic.com/images?q=tbn:ANd9GcSeyiSfAmb3V5cjSudujc2eT77PHzWBK6NSoXJq98haB3lq3fkQqCaQjdH8vQ&s",
+          "https://encrypted-tbn0.gstatic.com/images?q=tbn:ANd9GcQMy2ir-gBXuMLoVsSp4_mMHRO_Rq08JGEzwDlhSvo2J59dzhWp95tYGf-KAsw&s"
+        ]
+    
+    	money = [
+          "https://encrypted-tbn0.gstatic.com/images?q=tbn:ANd9GcSwCZT1f20JD7ILOxbV2cShU-SuHUMT0oT1tjy0rJm4xYS8dwbbk-H3jkhIvg&s",
+          "https://encrypted-tbn0.gstatic.com/images?q=tbn:ANd9GcQojWsOuIE5fmrOFQsq0cXpVASRFWpMJMjJ1lB1pGMKaLnRf552mQDkfM0pha0&s",
+          "https://encrypted-tbn0.gstatic.com/images?q=tbn:ANd9GcQjUDxa2JbgapLu1lNfsULxRFW6XXR4wrkD0kDvZqDxSrWiOmwx15POh0nGdQ&s",
+          "https://encrypted-tbn0.gstatic.com/images?q=tbn:ANd9GcSPNpEwR_ANwP6kYRC65Rp-U82IhWmU3pvuIUDRtyrGyk3hagssOXfA4NT2IZ0&s",
+          "https://encrypted-tbn0.gstatic.com/images?q=tbn:ANd9GcToE7Ay3jzGO3Rz71beO326P5mljAhr5Wn_jPDv9MQ9JBh58A4EWbexZcNeMXE&s"
+        ]
+    
+    	rainbow = [
+          "https://encrypted-tbn0.gstatic.com/images?q=tbn:ANd9GcTfVYUrYDW6WSjWPnWaOePIqrkKY0RSTl3UeQ9BSZ9eV2lPau3SZwC7d8HY9w&s",
+          "https://encrypted-tbn0.gstatic.com/images?q=tbn:ANd9GcRlh_7Jt36GdZrJbh3Yo6RerkSYqj13osiaX5mbZz3tLMPoeMhzhsVLV9JW0hM&s",
+          "https://encrypted-tbn0.gstatic.com/images?q=tbn:ANd9GcScCYEWFA9lfUD_CN54UePl5Uzu3QTFJTPwgfgCl7nOACbPd9QSGfVWsYjNxfI&s",
+          "https://encrypted-tbn0.gstatic.com/images?q=tbn:ANd9GcTUjtuadZfnEkXuKUgBeevaxCTzV-Sv8kPHCybgjBBc8cVOvTvczTJvF0TGbw&s",
+          "https://encrypted-tbn0.gstatic.com/images?q=tbn:ANd9GcQkheh6kpg1Bf-rCkAp2olc962inDLHkoZeQsoSflhgy3NoF1lZB0axB_zDcA&s"
+        ]
+    
+    	rain = [
+          "https://encrypted-tbn0.gstatic.com/images?q=tbn:ANd9GcTYuXU6x9-tPwfAoPfAViD17-V405LGaL8jG07gxgKHx_CSXpf0H3u6a9pLeg&s",
+          "https://encrypted-tbn0.gstatic.com/images?q=tbn:ANd9GcQZv3dycH2a_IYBzMGZt-Y4Fkoc0AyyVyhv4V5GDWMBBB5iKYnIgAq8-29Pi1U&s",
+          "https://encrypted-tbn0.gstatic.com/images?q=tbn:ANd9GcQedxpLUE9j23dqdaD1MFeDROs4uLxV9MFC-aGNv76ck7rbltqtlmycsxagug&s",
+          "https://encrypted-tbn0.gstatic.com/images?q=tbn:ANd9GcR3jxGROSevIg4gKazU-JMH3voUJSaH4xXrm0bquGl4fxHq8V1AtfHXTEqgFA&s",
+          "https://encrypted-tbn0.gstatic.com/images?q=tbn:ANd9GcTC-JteC4ZshI0GI7fqh7WfHnglxeef9B7eaG3F9cysviki42Pv1byvpCKWfw&s"
+        ]
+
+
         if "face" in prompt:
             output_url = random.choice(faces)
         elif "landscape" in prompt:
@@ -230,7 +279,19 @@ def generate_image(prompt):
         elif "background" in prompt:
             output_url = random.choice(background)
         elif "rock" in prompt:
-            output_url = random.choice(rocks)
+            output_url = random.choice(rocks)    
+        elif "gold robot" in prompt:
+            output_url = random.choice(gold_robot)
+        elif "house" in prompt:
+            output_url = random.choice(house)
+        elif "horse" in prompt:
+            output_url = random.choice(horse)
+        elif "money" in prompt:
+            output_url = random.choice(money)
+        elif "rainbow" in prompt:
+            output_url = random.choice(rainbow)
+        elif "rain" in prompt:
+            output_url = random.choice(rain)
         else:
             return "No image generated."