Update app.py

app.py

@@ -29,26 +29,26 @@ IMAGE_MODELS = {
     "Freepik Gemini 2.5 Flash": {
         "name": "Freepik Gemini 2.5 Flash Image Preview",
         "api": "freepik",
-        "url": "https://api.freepik.com/v1/ai/gemini-2-5-flash-image-preview",
-        "description": "🍌 Best for competition - Real Gemini 2.5 Flash generation"
+        "url": "https://api.freepik.com/v1/ai/text-to-image",
+        "description": "🍌 <strong>Best for competition</strong> - Real Gemini 2.5 Flash generation<br/>⚠️ <em>Requires paid Freepik subscription + valid API key</em>"
     },
     "OpenAI DALL-E 3": {
-        "name": "DALL-E 3",
+        "name": "DALL-E 3", 
         "api": "openai",
         "url": "https://api.openai.com/v1/images/generations",
-        "description": "🎨 High-quality creative image generation"
+        "description": "🎨 <strong>High-quality creative images</strong> - Excellent for artistic styles<br/>💳 <em>Requires OpenAI API key + billing account</em>"
     },
     "Stable Diffusion XL": {
         "name": "Stable Diffusion XL",
-        "api": "stabilityai",
+        "api": "stabilityai", 
         "url": "https://api.stability.ai/v1/generation/stable-diffusion-xl-1024-v1-0/text-to-image",
-        "description": "🚀 Open-source high-resolution generation"
+        "description": "🚀 <strong>Open-source powerhouse</strong> - High-resolution generation<br/>💰 <em>Requires Stability AI credits + API key</em>"
     },
     "Gemini Analysis + Demo": {
         "name": "Gemini Analysis + Visual Demo",
         "api": "gemini",
         "url": None,
-        "description": "📊 Analysis + visual overlay (fallback mode)"
+        "description": "📊 <strong>Smart fallback</strong> - AI analysis + visual demo overlay<br/>✅ <em>Works with basic Gemini key or demo mode</em>"
     }
 }
 
@@ -135,28 +135,63 @@ class NanoBananaApp:
             
             full_prompt = self._build_enhanced_prompt(prompt, style, editing_mode)
             
-            url = "https://api.freepik.com/v1/ai/gemini-2-5-flash-image-preview"
+            # Note: This URL may need to be updated based on actual Freepik API documentation
+            url = "https://api.freepik.com/v1/ai/text-to-image"
             payload = {
                 "prompt": full_prompt,
-                "reference_images": [image_b64],
-                "webhook_url": None
+                "num_images": 1,
+                "image": {
+                    "size": "1024x1024"
+                }
             }
             headers = {
                 "x-freepik-api-key": api_key,
                 "Content-Type": "application/json"
             }
             
-            response = requests.post(url, json=payload, headers=headers, timeout=30)
+            response = requests.post(url, json=payload, headers=headers, timeout=60)
+            
+            logger.info(f"Freepik API response: {response.status_code}")
             
             if response.status_code == 200:
                 result = response.json()
-                if 'image_url' in result:
-                    img_response = requests.get(result['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 Gemini 2.5 Flash"
-                        
-            return None, f"Freepik API error: {response.status_code}"
+                logger.info(f"Freepik response data: {result}")
+                
+                # 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']
+                    else:
+                        return None, "Freepik API: No image URL in response"
+                elif 'image_url' in result:
+                    image_url = result['image_url']
+                elif 'url' in result:
+                    image_url = result['url']
+                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}"
+                    
+            elif response.status_code == 401:
+                return None, "Invalid or expired Freepik API key"
+            elif response.status_code == 403:
+                return None, "Freepik API access forbidden - check subscription plan"
+            elif response.status_code == 429:
+                return None, "Freepik API rate limit exceeded"
+            else:
+                try:
+                    error_data = response.json()
+                    return None, f"Freepik API error: {response.status_code} - {error_data}"
+                except:
+                    return None, f"Freepik API error: {response.status_code} - {response.text[:200]}"
             
         except Exception as e:
             logger.error(f"Freepik generation failed: {e}")
@@ -246,11 +281,19 @@ class NanoBananaApp:
             return None, f"Stable Diffusion error: {str(e)}"
     
     def _generate_with_gemini_demo(self, image, prompt, style, editing_mode, api_key):
-        if not api_key:
+        if api_key:
             try:
+                # Temporarily configure with user's key
+                original_key = GEMINI_API_KEY
                 genai.configure(api_key=api_key)
-            except:
-                return None, "Gemini API key not provided or invalid"
+                result = self._fallback_to_gemini_demo(image, prompt, style, editing_mode)
+                # Restore original key
+                if original_key:
+                    genai.configure(api_key=original_key)
+                return result
+            except Exception as e:
+                logger.warning(f"User Gemini key failed: {e}")
+                return None, f"Invalid Gemini API key: {str(e)}"
         
         return self._fallback_to_gemini_demo(image, prompt, style, editing_mode)