Update app.py

app.py

@@ -31,9 +31,9 @@ if user_input and submit:
     description = response["weather"][0]["description"]
     
     if additional:
-        image_url = generate_image(city, weather, country, description, user_input, additional)
+        prompt = f"{additional} at {user_input} in {location},{country} with the weather consisting of {weather} and {description}"
     else:
-        image_url = generate_image(city, weather, country, description, user_input)
+        prompt = f"{user_input} in {location},{country} with the weather consisting of {weather} and {description}"
+    image_url = generate_image(prompt)
     st.image(image_url, caption=f"{weather} at {user_input} in {location},{country}")
 
-