Update app.py

app.py

@@ -4,6 +4,7 @@ from PIL import Image
 from io import BytesIO
 import base64
 import os
+import time
 
 API_KEY = os.getenv("NVIDIA_API_KEY")
 invoke_url = "https://ai.api.nvidia.com/v1/genai/stabilityai/sdxl-turbo"
@@ -49,7 +50,13 @@ def generate_kindle_cover(prompt):
         image = Image.open(BytesIO(image_bytes))
         print("Image successfully created")
         
-        return image
+        # Save the image
+        timestamp = int(time.time())
+        filename = f"kindle_cover_{timestamp}.png"
+        image.save(filename)
+        print(f"Image saved as {filename}")
+        
+        return filename
     
     except Exception as e:
         print(f"An error occurred: {str(e)}")
@@ -58,7 +65,7 @@ def generate_kindle_cover(prompt):
 iface = gr.Interface(
     fn=generate_kindle_cover,
     inputs="text",
-    outputs="image",
+    outputs="file",
     title="Kindle Cover Generator",
     description="Generate high-quality covers for Amazon Kindle books using the NVIDIA SDXL-Turbo model."
 )