Update main.py

main.py

@@ -21,7 +21,7 @@ def generate_and_upscale_image(text_prompt, clipdrop_api_key, stability_api_key,
 
     if response.status_code != 200:
         print(f"Request failed with status code {response.status_code}")
-        return None
+        return None, f"Request failed with status code {response.status_code}"
 
     with open('generated_image.png', 'wb') as f:
         f.write(response.content)
@@ -67,9 +67,12 @@ def generate_and_upscale_image(text_prompt, clipdrop_api_key, stability_api_key,
         )
 
     response = requests.get(output)
+    if response.status_code != 200:
+        return None, f"Failed to fetch upscaled image. Status code {response.status_code}"
+
     final_img = Image.open(io.BytesIO(response.content))
     img_byte_arr = io.BytesIO()
     final_img.save(img_byte_arr, format='PNG')
     img_byte_arr = img_byte_arr.getvalue()
 
-    return img_byte_arr
+    return img_byte_arr, None