Update templates/api_docs.html

templates/api_docs.html

@@ -112,7 +112,7 @@
             <pre>
 import requests
 
-url = 'https://imseldrith-imagine.hf.space/api/generate'
+url = 'https://imseldrith-imagine.hf.space/api/generate' 
 
 headers = {'Content-Type':'application/json'}
 payload = {
@@ -122,10 +122,18 @@ payload = {
     'model': 'REALISTIC'
 }
 
-response = requests.post(url, headers=headers, json=payload)
-result = response.json()
+response = requests.post(url, json=payload)
 
-print(result['error']) # An error occurred while generating the image.
+if response.status_code == 200:
+    # Image generation successful
+    with open('generated_image.jpg', 'wb') as image_file:
+        image_file.write(response.content)
+    print('Image saved successfully!')
+else:
+    # Error occurred during image generation
+    error_data = response.json()
+    error_message = error_data.get('error', 'Unknown error')
+    print(f'Failed to generate image: {error_message}')
             </pre>
         </div>
     </div>