Update app.py

app.py

@@ -13,7 +13,7 @@ def generate_image(prompt, negative_prompt, scheduler, steps, width, height, res
     # Define the request headers
     headers = {
         "Content-Type": "application/json",
-        "token": "token"
+        "token": token
     }
 
     # Define the request body
@@ -24,7 +24,7 @@ def generate_image(prompt, negative_prompt, scheduler, steps, width, height, res
         "batch_size": 1,
         "prompt": prompt,
         "negative_prompt": negative_prompt,
-        "seed": seed if seed else random.randint(0, 999999999),
+        "seed": seed if seed is not None else random.randint(0, 999999999),
         "scheduler": scheduler,
         "n_iter": 1,
         "steps": steps,
@@ -69,13 +69,17 @@ def generate_image(prompt, negative_prompt, scheduler, steps, width, height, res
     # Check the response status
     if response.status_code == 200:
         # Get the image URL from the response
-        image_url = json.loads(response.text)['results'][0]
+        response_json = response.json()
+        if 'results' in response_json and isinstance(response_json['results'], list) and len(response_json['results']) > 0:
+            image_url = response_json['results'][0]
 
-        # Get the image from the URL
-        image_response = requests.get(image_url)
-        image = PIL.Image.open(BytesIO(image_response.content))
+            # Get the image from the URL
+            image_response = requests.get(image_url)
+            image = PIL.Image.open(BytesIO(image_response.content))
 
-        return image
+            return image
+        else:
+            raise Exception("Unexpected API response format")
     else:
         raise Exception("API request failed with status code " + str(response.status_code))