Update app.py

app.py

@@ -17,6 +17,7 @@ logger = logging.getLogger(__name__)
 
 GEMINI_API_KEY = os.getenv("GEMINI_API_KEY")
 ELEVENLABS_API_KEY = os.getenv("ELEVENLABS_API_KEY")
+FREEPIK_API_KEY = os.getenv("FREEPIK_API_KEY")
 
 MAX_IMAGE_SIZE = 1024
 RATE_LIMIT_DELAY = 3
@@ -56,13 +57,13 @@ class NanoBananaApp:
             logger.warning("No API key - demo mode")
             return
         try:
-            self.gemini_model = genai.GenerativeModel('gemini-2.5-flash-image-preview')
-            logger.info("Nano Banana (gemini-2.5-flash-image-preview) initialized")
+            self.gemini_model = genai.GenerativeModel('gemini-2.5-flash')
+            logger.info("Nano Banana (Gemini 1.5 Flash) initialized")
         except Exception as e:
             logger.error(f"Failed to initialize Gemini: {e}")
             try:
-                self.gemini_model = genai.GenerativeModel('gemini-2.5-flash-image-preview')
-                logger.info("Fallback: gemini-2.5-flash-image-preview")
+                self.gemini_model = genai.GenerativeModel('gemini-2.5-pro')
+                logger.info("Fallback: Gemini 1.5 Pro initialized")
             except Exception as e2:
                 logger.error(f"Fallback also failed: {e2}")
             
@@ -75,6 +76,56 @@ class NanoBananaApp:
         
     def _apply_rate_limiting(self):
         time.sleep(RATE_LIMIT_DELAY)
+        
+    def _generate_with_freepik(self, image, prompt, style, editing_mode):
+        if not FREEPIK_API_KEY:
+            return None, "Freepik API key not available"
+            
+        try:
+            buffered = io.BytesIO()
+            image.save(buffered, format='PNG')
+            image_b64 = base64.b64encode(buffered.getvalue()).decode('utf-8')
+            
+            style_modifiers = {
+                "realistic": "photorealistic, high-quality construction, professional architecture",
+                "futuristic": "futuristic, high-tech, modern glass and steel, sci-fi architecture",
+                "artistic": "artistic, creative design, unique architecture, colorful and innovative"
+            }
+            
+            mode_descriptions = {
+                "complete": "Complete this unfinished construction",
+                "edit": "Edit and transform this construction image",
+                "blend": "Blend and reimagine this construction"
+            }
+            
+            full_prompt = f"{mode_descriptions.get(editing_mode, 'Transform')} {prompt}. Style: {style_modifiers.get(style, '')}. Make it look professional and realistic."
+            
+            url = "https://api.freepik.com/v1/ai/gemini-2-5-flash-image-preview"
+            payload = {
+                "prompt": full_prompt,
+                "reference_images": [image_b64],
+                "webhook_url": None
+            }
+            headers = {
+                "x-freepik-api-key": FREEPIK_API_KEY,
+                "Content-Type": "application/json"
+            }
+            
+            response = requests.post(url, json=payload, headers=headers, timeout=30)
+            
+            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}"
+            
+        except Exception as e:
+            logger.error(f"Freepik generation failed: {e}")
+            return None, f"Freepik error: {str(e)}"
             
     def load_yolo_optional(self):
         if not yolo_available:
@@ -109,10 +160,20 @@ class NanoBananaApp:
         if not prompt.strip():
             return image, "Please provide a transformation prompt"
             
+        logger.info(f"Processing: {editing_mode} mode, {style} style, prompt: {prompt[:50]}...")
+            
         try:
             image = self._resize_image_if_needed(image)
             self._apply_rate_limiting()
             
+            # Try Freepik Gemini 2.5 Flash first (best quality)
+            if FREEPIK_API_KEY:
+                freepik_result, freepik_msg = self._generate_with_freepik(image, prompt, style, editing_mode)
+                if freepik_result is not None:
+                    return freepik_result, f"🍌 Nano Banana (Freepik): {freepik_msg}"
+                logger.warning(f"Freepik failed: {freepik_msg}, falling back to Gemini analysis")
+            
+            # Fallback to Gemini analysis + demo overlay
             if editing_mode == "complete":
                 base_prompt = self._get_completion_prompt(style)
                 analysis_prompt = f"Analyze this construction image and describe how to {base_prompt.lower()} User request: {prompt}. Provide detailed description of the completed construction."
@@ -132,13 +193,21 @@ class NanoBananaApp:
                     if len(image_bytes) > 10 * 1024 * 1024:
                         return image, "Image too large. Please use a smaller image."
                     
-                    response = self.gemini_model.generate_content([
-                        analysis_prompt,
-                        {
-                            'mime_type': 'image/png',
-                            'data': base64.b64encode(image_bytes).decode('utf-8')
-                        }
-                    ])
+                    try:
+                        response = self.gemini_model.generate_content([
+                            analysis_prompt,
+                            {
+                                '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")
+                            processed_image = self._create_nano_banana_demo(image, f"Demo mode: {prompt}", style, editing_mode)
+                            return processed_image, f"🍌 Nano Banana Demo: {editing_mode} mode with {style} style (Image analysis unavailable)"
+                        else:
+                            raise img_error
                     
                     if hasattr(response, 'text') and response.text:
                         processed_image = self._create_nano_banana_demo(image, response.text, style, editing_mode)
@@ -317,12 +386,21 @@ custom_css = """
 """
 
 demo_mode_notice = ""
-if not GEMINI_API_KEY:
+if not FREEPIK_API_KEY and not GEMINI_API_KEY:
     demo_mode_notice = """
     <div style="background: #ffebee; border: 1px solid #f44336; border-radius: 8px; padding: 15px; margin: 10px 0;">
-        <h3>🔑 API Key Required</h3>
-        <p>To use Nano Banana features, add your <strong>GEMINI_API_KEY</strong> in the Space settings.</p>
-        <p>Get your free API key from <a href="https://makersuite.google.com/app/apikey" target="_blank">Google AI Studio</a></p>
+        <h3>🔑 API Keys Required</h3>
+        <p><strong>For best results:</strong> Add <strong>FREEPIK_API_KEY</strong> for real image generation</p>
+        <p><strong>For analysis:</strong> Add <strong>GEMINI_API_KEY</strong> from <a href="https://makersuite.google.com/app/apikey" target="_blank">Google AI Studio</a></p>
+        <p><em>Demo mode available without API keys</em></p>
+    </div>
+    """
+elif not FREEPIK_API_KEY:
+    demo_mode_notice = """
+    <div style="background: #e3f2fd; border: 1px solid #2196f3; border-radius: 8px; padding: 15px; margin: 10px 0;">
+        <h3>🍌 Enhanced Mode Available</h3>
+        <p>Add <strong>FREEPIK_API_KEY</strong> for real Gemini 2.5 Flash image generation!</p>
+        <p><em>Currently using Gemini analysis + demo mode</em></p>
     </div>
     """
 
@@ -412,11 +490,12 @@ with gr.Blocks(title="🍌 Nano Banana - Dynamic Image Creation", theme=gr.theme
     
     gr.Markdown("""
     ### 🏆 Competition Features
-    - **Nano Banana Core**: Gemini 2.5 Flash Image for dynamic creation
-    - **Word-Based Editing**: Transform images with natural language
-    - **Reality Blending**: Seamlessly fuse different visual elements  
-    - **Optional Enhancements**: Structure detection and voice narration
-    - **Real-time Processing**: Fast image transformations
+    - **🍌 Nano Banana Core**: Freepik's Gemini 2.5 Flash Image Preview for real image generation
+    - **🎨 Word-Based Editing**: Transform images with natural language prompts
+    - **🌟 Reality Blending**: Seamlessly fuse different visual elements  
+    - **⚡ Real-time Processing**: Fast image transformations and generation
+    - **🛠️ Optional Enhancements**: Structure detection (YOLO) and voice narration (ElevenLabs)
+    - **🔄 Smart Fallbacks**: Multiple processing modes for reliability
     """)
     
     process_btn.click(