Update app.py

app.py

@@ -5,8 +5,7 @@ from io import BytesIO
 import base64
 import os
 
-# Replace with your NVIDIA API key
-API_KEY = os.getenv("NVIDIA_API_KEY")  # Ensure the key is set in Space secrets
+API_KEY = os.getenv("NVIDIA_API_KEY")
 invoke_url = "https://ai.api.nvidia.com/v1/genai/stabilityai/sdxl-turbo"
 headers = {
     "Authorization": f"Bearer {API_KEY}",
@@ -14,6 +13,8 @@ headers = {
 }
 
 def generate_kindle_cover(prompt):
+    print(f"Generating cover for prompt: {prompt}")
+    
     payload = {
         "text_prompts": [{"text": prompt}],
         "seed": 0,
@@ -22,35 +23,38 @@ def generate_kindle_cover(prompt):
     }
     
     try:
+        print("Sending request to NVIDIA API...")
         response = requests.post(invoke_url, headers=headers, json=payload)
-        response.raise_for_status()  # Raise an exception for non-200 status codes
+        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)  # Debugging line to inspect the response
-        
-        # Check for 'results' key in response
-        if 'results' in response_body and len(response_body['results']) > 0:
-            result = response_body['results'][0]
-            
-            # Check for 'image' key in result
-            if 'image' in result:
-                image_data = result['image']
-                image_bytes = base64.b64decode(image_data)
-                image = Image.open(BytesIO(image_bytes))
-                return image
-            else:
-                return "No 'image' key found in the result."
-        else:
-            return "No 'results' found in the response."
+        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")
+        
+        return image
     
-    except requests.exceptions.RequestException as e:
-        return f"Error making request: {str(e)}"
-    except ValueError as e:
-        return f"Error parsing JSON response: {str(e)}"
     except Exception as e:
-        return f"Unexpected error: {str(e)}"
+        print(f"An error occurred: {str(e)}")
+        return f"Error: {str(e)}"
 
-# Create the Gradio interface
 iface = gr.Interface(
     fn=generate_kindle_cover,
     inputs="text",
@@ -59,5 +63,4 @@ iface = gr.Interface(
     description="Generate high-quality covers for Amazon Kindle books using the NVIDIA SDXL-Turbo model."
 )
 
-# Launch the Gradio interface
-iface.launch()
+iface.launch(debug=True)