Update text2image.py

text2image.py

@@ -3,14 +3,9 @@ import os
 import json
 STABLE_API = os.getenv("STABLE_API")
 
-def generate_image(location, weather, country, description, user_input, additional=""):
+def generate_image(prompt):
   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": prompt,
@@ -36,4 +31,4 @@ def generate_image(location, weather, country, description, user_input, addition
   }
 
   response = requests.request("POST", url, headers=headers, data=payload)
-  return response.json()["output"][0]
+  return response.json()["output"][0]