Update app.py

app.py

@@ -63,10 +63,13 @@ try:
     if ELEVENLABS_API_KEY:
         set_api_key(ELEVENLABS_API_KEY)
         logger.info("ElevenLabs configured")
+        elevenlabs_available = True
     else:
         logger.info("ElevenLabs not configured - optional feature")
+        elevenlabs_available = False
 except ImportError:
     logger.info("ElevenLabs not available - optional feature")
+    elevenlabs_available = False
 
 try:
     from ultralytics import YOLO
@@ -155,30 +158,53 @@ class NanoBananaApp:
             
             if response.status_code == 200:
                 result = response.json()
-                logger.info(f"Freepik response data: {result}")
+                logger.info(f"Freepik response data keys: {list(result.keys())}")
                 
                 # Handle different possible response formats
                 if 'data' in result and len(result['data']) > 0:
-                    if 'url' in result['data'][0]:
-                        image_url = result['data'][0]['url']
-                    elif 'image_url' in result['data'][0]:
-                        image_url = result['data'][0]['image_url']
+                    data_item = result['data'][0]
+                    
+                    # Check if we have base64 data directly
+                    if 'base64' in data_item:
+                        try:
+                            image_data = base64.b64decode(data_item['base64'])
+                            generated_image = Image.open(io.BytesIO(image_data))
+                            return generated_image, "🍌 Generated with Freepik Gemini 2.5 Flash"
+                        except Exception as e:
+                            logger.error(f"Error decoding base64 image: {e}")
+                            return None, f"Error processing generated image: {str(e)}"
+                    
+                    # Check for image URL
+                    elif 'url' in data_item:
+                        image_url = data_item['url']
+                    elif 'image_url' in data_item:
+                        image_url = data_item['image_url']
                     else:
-                        return None, "Freepik API: No image URL in response"
+                        return None, "Freepik API: No image data or URL found in response"
+                        
+                    # If we have a URL, download it
+                    if 'image_url' in locals():
+                        img_response = requests.get(image_url, timeout=30)
+                        if img_response.status_code == 200:
+                            generated_image = Image.open(io.BytesIO(img_response.content))
+                            return generated_image, "🍌 Generated with Freepik API"
+                        else:
+                            return None, f"Failed to download image: {img_response.status_code}"
+                            
                 elif 'image_url' in result:
                     image_url = result['image_url']
+                    img_response = requests.get(image_url, timeout=30)
+                    if img_response.status_code == 200:
+                        generated_image = Image.open(io.BytesIO(img_response.content))
+                        return generated_image, "🍌 Generated with Freepik API"
                 elif 'url' in result:
                     image_url = result['url']
+                    img_response = requests.get(image_url, timeout=30)
+                    if img_response.status_code == 200:
+                        generated_image = Image.open(io.BytesIO(img_response.content))
+                        return generated_image, "🍌 Generated with Freepik API"
                 else:
-                    return None, f"Freepik API: Unexpected response format - {result}"
-                
-                # Download the generated image
-                img_response = requests.get(image_url, timeout=30)
-                if img_response.status_code == 200:
-                    generated_image = Image.open(io.BytesIO(img_response.content))
-                    return generated_image, "🍌 Generated with Freepik API"
-                else:
-                    return None, f"Failed to download image: {img_response.status_code}"
+                    return None, f"Freepik API: Unexpected response format - available keys: {list(result.keys())}"
                     
             elif response.status_code == 401:
                 return None, "Invalid or expired Freepik API key"
@@ -381,9 +407,9 @@ class NanoBananaApp:
             
             for attempt in range(API_RETRY_COUNT):
                 try:
-                    buffered = io.BytesIO()
+            buffered = io.BytesIO()
                     image.save(buffered, format='PNG', quality=85)
-                    image_bytes = buffered.getvalue()
+            image_bytes = buffered.getvalue()
 
                     if len(image_bytes) > 10 * 1024 * 1024:
                         return image, "Image too large. Please use a smaller image."
@@ -391,11 +417,11 @@ class NanoBananaApp:
                     try:
                         response = self.gemini_model.generate_content([
                             analysis_prompt,
-                            {
-                                'mime_type': 'image/png',
-                                'data': base64.b64encode(image_bytes).decode('utf-8')
-                            }
-                        ])
+                {
+                    'mime_type': 'image/png',
+                    'data': base64.b64encode(image_bytes).decode('utf-8')
+                }
+            ])
                     except Exception as img_error:
                         if "API_KEY_INVALID" in str(img_error) or "API key not valid" in str(img_error):
                             logger.warning("API key doesn't support image processing, using demo mode")
@@ -473,7 +499,7 @@ class NanoBananaApp:
         return prompts.get(style, prompts["realistic"])
     
     def generate_voice_optional(self, text):
-        if not ELEVENLABS_API_KEY:
+        if not elevenlabs_available or not ELEVENLABS_API_KEY:
             return None
         try:
             audio = generate(text=text, voice="Rachel", model="eleven_monolingual_v1")