Update app.py

app.py

@@ -1,10 +1,7 @@
 import requests
 import gradio as gr
-from PIL import Image
-from io import BytesIO
-import base64
 import os
-import time
+import json
 
 API_KEY = os.getenv("NVIDIA_API_KEY")
 invoke_url = "https://ai.api.nvidia.com/v1/genai/stabilityai/sdxl-turbo"
@@ -15,6 +12,7 @@ headers = {
 
 def generate_kindle_cover(prompt):
     print(f"Generating cover for prompt: {prompt}")
+    print(f"API Key (first 5 characters): {API_KEY[:5]}..." if API_KEY else "API Key is not set")
     
     payload = {
         "text_prompts": [{"text": prompt}],
@@ -27,36 +25,20 @@ def generate_kindle_cover(prompt):
         print("Sending request to NVIDIA API...")
         response = requests.post(invoke_url, headers=headers, json=payload)
         print(f"Received response with status code: {response.status_code}")
-        
-        if response.status_code != 200:
-            return f"Error: Received status code {response.status_code} from API"
-        
-        response_body = response.json()
-        print("Response Body:", response_body)
-        
-        if 'results' not in response_body or not response_body['results']:
-            return "Error: No results found in the response"
-        
-        result = response_body['results'][0]
-        if 'image' not in result:
-            return "Error: No image data found in the result"
-        
-        image_data = result['image']
-        print("Image data length:", len(image_data))
-        
-        image_bytes = base64.b64decode(image_data)
-        print("Decoded image bytes length:", len(image_bytes))
-        
-        image = Image.open(BytesIO(image_bytes))
-        print("Image successfully created")
-        
-        # Save the image
-        timestamp = int(time.time())
-        filename = f"kindle_cover_{timestamp}.png"
-        image.save(filename)
-        print(f"Image saved as {filename}")
-        
-        return filename
+        print(f"Response headers: {json.dumps(dict(response.headers), indent=2)}")
+        print(f"Response content: {response.text}")
+        
+        if response.status_code == 401:
+            return "Error: Unauthorized access (401). Please check your API key."
+        elif response.status_code != 200:
+            return f"Error: Received status code {response.status_code} from API. Response: {response.text}"
+        
+        try:
+            response_body = response.json()
+            print("Response Body:", json.dumps(response_body, indent=2))
+            return f"Success: API responded with status code 200. Check console for full response details."
+        except json.JSONDecodeError:
+            return f"Error: Failed to parse JSON response. Raw response: {response.text}"
     
     except Exception as e:
         print(f"An error occurred: {str(e)}")
@@ -65,7 +47,7 @@ def generate_kindle_cover(prompt):
 iface = gr.Interface(
     fn=generate_kindle_cover,
     inputs="text",
-    outputs="file",
+    outputs="text",
     title="Kindle Cover Generator",
     description="Generate high-quality covers for Amazon Kindle books using the NVIDIA SDXL-Turbo model."
 )