Update text2image.py

text2image.py

@@ -3,12 +3,17 @@ import os
 import json
 STABLE_API = os.getenv("STABLE_API")
 
-def generate_image(location, weather, country, description):
+def generate_image(location, weather, country, description, user_input, additional=""):
   url = "https://stablediffusionapi.com/api/v3/text2img"
 
+  if additional:
+    prompt = f"{additional} at {user_input} in {location},{country} with the weather consisting of {weather} and {description}"
+  else:
+    prompt = f"{user_input} in {location},{country} with the weather consisting of {weather} and {description}"
+
   payload = json.dumps({
     "key": STABLE_API,
-    "prompt": f"scenary or famous landmark in {location},{country} with the weather consisting of {weather} and {description}",
+    "prompt": prompt,
     "negative_prompt": None,
     "width": "512",
     "height": "512",