Update app.py

app.py

@@ -9,11 +9,13 @@ openai.api_key = os.getenv("OPENAI_API_KEY")
 
 
 # Define the function that generates the image
-response = openai.Image.create(
-  prompt=prompt,
-  n=1,
-  size="1024x1024"
-)
+def generate_image(prompt):
+    response = openai.Image.create(
+      prompt=prompt,
+      n=1,
+      size="1024x1024"
+    )    
+
 image_url = response['data'][0]['url']