Update app.py

app.py

@@ -77,6 +77,9 @@ GOOGLE_API_KEY = next((key for key in GCP_KEYS if key), None)
 if GOOGLE_API_KEY and GEMINI_AVAILABLE:
     genai.configure(api_key=GOOGLE_API_KEY)
     logger.info("✅ Google AI configured successfully")
+    logger.info(f"Key source: {[key for key in ['GOOGLE_API_KEY', 'GEMINI_API_KEY', 'GCP_API_KEY'] if os.getenv(key)]}")
+else:
+    logger.warning(f"❌ Google AI NOT configured - GEMINI_AVAILABLE: {GEMINI_AVAILABLE}, GOOGLE_API_KEY: {'present' if GOOGLE_API_KEY else 'missing'}")
 
 # MCP-enabled functions for Agent1 (Image Generator)
 def enhance_prompt_with_gemini(prompt: str, style: str) -> str:
@@ -159,6 +162,7 @@ def generate_marketing_image(prompt: str, style: str = "realistic") -> str:
         if GEMINI_AVAILABLE and GOOGLE_API_KEY:
             try:
                 logger.info("🎨 Using Google Genai SDK for image generation")
+                logger.info(f"API Key available: {GOOGLE_API_KEY[:10]}...")
                 
                 # Initialize the genai SDK client  
                 client = genai_sdk.Client(api_key=GOOGLE_API_KEY)
@@ -201,6 +205,10 @@ def generate_marketing_image(prompt: str, style: str = "realistic") -> str:
                         
             except Exception as e:
                 logger.error(f"Google SDK generation failed: {e}")
+                logger.error(f"Error type: {type(e).__name__}")
+                if hasattr(e, 'response'):
+                    logger.error(f"Response status: {getattr(e.response, 'status_code', 'unknown')}")
+                    logger.error(f"Response text: {getattr(e.response, 'text', 'unknown')}")
         
         # Fallback: Generate a deterministic placeholder
         logger.info("🔄 Using placeholder URL fallback")
@@ -439,11 +447,21 @@ def process_generated_image_and_results(api_response_str: str) -> Tuple[Image.Im
         image_data_b64 = image_info.get('data', image_info.get('url', ''))
         
         image = None
-        if image_data_b64 and image_data_b64.startswith('data:image'):
+        if image_data_b64:
             try:
-                base64_data = image_data_b64.split(',')[1]
-                image_bytes = base64.b64decode(base64_data)
-                image = Image.open(io.BytesIO(image_bytes))
+                if image_data_b64.startswith('data:image'):
+                    # Handle base64 data URLs
+                    base64_data = image_data_b64.split(',')[1]
+                    image_bytes = base64.b64decode(base64_data)
+                    image = Image.open(io.BytesIO(image_bytes))
+                elif image_data_b64.startswith('http'):
+                    # Handle regular URLs (like picsum.photos)
+                    import requests
+                    response = requests.get(image_data_b64, timeout=10)
+                    if response.status_code == 200:
+                        image = Image.open(io.BytesIO(response.content))
+                    else:
+                        logger.error(f"Failed to fetch image from URL: {response.status_code}")
             except Exception as e:
                 logger.error(f"Error processing image: {str(e)}")