Update app.py

app.py

@@ -5,92 +5,122 @@ from PIL import Image
 import requests
 from io import BytesIO
 import time
-import tempfile
 import base64
 import json
 
 # Set up API keys from environment variables
 os.environ['REPLICATE_API_TOKEN'] = os.getenv('REPLICATE_API_TOKEN')
 FIREWORKS_API_KEY = os.getenv('FIREWORKS_API_KEY', '')
+BRAVE_API_KEY = os.getenv('BRAVE_API_KEY', '')  # Add Brave API key
+
+def search_design_trends(product_type):
+    """
+    Use Brave Search API to find current design trends and technologies
+    """
+    if not BRAVE_API_KEY:
+        return "No search results available (API key not set)"
+    
+    try:
+        # Construct search query for product design
+        query = f"{product_type} design trends 2024 2025 materials technology innovation"
+        
+        url = "https://api.search.brave.com/res/v1/web/search"
+        headers = {
+            "Accept": "application/json",
+            "X-Subscription-Token": BRAVE_API_KEY
+        }
+        params = {
+            "q": query,
+            "count": 5
+        }
+        
+        response = requests.get(url, headers=headers, params=params, timeout=10)
+        
+        if response.status_code == 200:
+            data = response.json()
+            results = []
+            
+            if 'web' in data and 'results' in data['web']:
+                for item in data['web']['results'][:3]:
+                    results.append(f"- {item.get('title', '')}: {item.get('description', '')[:150]}")
+            
+            if results:
+                return "\n".join(results)
+            else:
+                return "Limited search results available"
+        else:
+            return "Search unavailable at the moment"
+            
+    except Exception as e:
+        print(f"Brave search error: {str(e)}")
+        return "Search results unavailable"
 
 def enhance_prompt_with_team(user_input):
     """
-    Enhanced prompt generation using team collaboration approach:
-    Director -> Engineer (with web search) -> Designer
+    Enhanced prompt generation using team collaboration with real web search
     """
     if not FIREWORKS_API_KEY:
-        return user_input
+        # Fallback to basic enhancement
+        return create_basic_design_prompt(user_input)
     
     try:
+        # First, get real search results for the engineer
+        search_results = search_design_trends(user_input)
+        
         url = "https://api.fireworks.ai/inference/v1/chat/completions"
         
-        # Team collaboration system prompt
-        system_prompt = """You are a product design team consisting of three roles working in sequence:
+        # Enhanced system prompt with search integration
+        system_prompt = """You are a product design team with three specialized roles:
 
-## TEAM COLLABORATION PROCESS:
+## TEAM ROLES:
 
 ### 1. DIRECTOR (Creative Vision Leader)
-The Director analyzes the user's input and establishes:
-- Core product category and target market
-- Key functional requirements
-- Design philosophy and brand positioning
-- Initial design direction and constraints
-- Questions for the engineering team
+- Analyzes user requirements and market positioning
+- Sets design philosophy and constraints
+- Makes final decisions on features
 
-### 2. ENGINEER (Technical Feasibility Expert)
-The Engineer responds to the Director's vision by:
-- Researching current technologies and materials
-- Identifying manufacturing possibilities
-- Suggesting innovative technical features
-- Evaluating feasibility of proposed concepts
-- Providing data on similar successful products
-- Recommending technical specifications
+### 2. ENGINEER (Technical Expert with Research Data)
+- Uses provided search results to inform decisions
+- Evaluates manufacturing feasibility
+- Recommends materials and technologies
+- Suggests innovative features based on trends
 
-### 3. DESIGNER (Visual Design Specialist)
-The Designer synthesizes all inputs to create the final prompt:
-- Translating technical specs into visual language
-- Adding aesthetic details and finishing touches
-- Specifying materials, textures, and colors
-- Defining the rendering style
-- Creating the complete visual description
+### 3. DESIGNER (Visual Specialist)
+- Creates detailed visual specifications
+- Specifies exact materials, finishes, colors
+- Defines rendering style and atmosphere
 
 ## OUTPUT FORMAT:
-Structure the team discussion as follows:
 
 **DIRECTOR'S VISION:**
-[Director's analysis and initial direction]
+[Brief vision statement - 2-3 sentences]
 
 **ENGINEER'S TECHNICAL ASSESSMENT:**
-[Research findings and technical recommendations]
-
-**DIRECTOR'S REFINEMENT:**
-[Approval/modification of engineering inputs]
+[Based on search data, provide 3-4 technical recommendations]
 
 **DESIGNER'S FINAL PROMPT:**
-[Complete product design prompt with all visual details]
+[Detailed visual description for image generation - focus on visual elements]
 
-Focus exclusively on PHYSICAL PRODUCT DESIGN - no services, apps, or abstract concepts.
-Examples: furniture, electronics, vehicles, appliances, tools, wearables, etc."""
+Keep responses concise but specific. Focus on PHYSICAL PRODUCTS only."""
         
-        # Create the collaborative dialogue
-        user_prompt = f"""Create a product design for: {user_input}
+        # Create collaborative prompt with search results
+        user_prompt = f"""Product concept: {user_input}
 
-Execute the full team collaboration process:
-1. Director establishes the vision
-2. Engineer researches and provides technical input (simulate web research findings)
-3. Director refines based on engineering feedback
-4. Designer creates the final detailed visual prompt
+ENGINEER'S RESEARCH DATA:
+{search_results}
 
-Remember: Focus ONLY on physical product design."""
+Based on this input and research, execute the team collaboration:
+1. Director establishes vision (brief)
+2. Engineer uses the research data above to recommend features
+3. Designer creates a detailed visual prompt
+
+Focus on creating a manufacturable, innovative product design."""
         
         payload = {
-            "model": "accounts/fireworks/models/qwen3-235b-a22b-instruct-2507",
-            "max_tokens": 800,
+            "model": "accounts/fireworks/models/qwen3-235b-a22b-instruct-2507",  # Original model maintained
+            "max_tokens": 500,
             "top_p": 0.9,
-            "top_k": 40,
-            "presence_penalty": 0.3,
-            "frequency_penalty": 0.2,
-            "temperature": 0.8,
+            "temperature": 0.7,
             "messages": [
                 {
                     "role": "system",
@@ -109,31 +139,50 @@ Remember: Focus ONLY on physical product design."""
             "Authorization": f"Bearer {FIREWORKS_API_KEY}"
         }
         
-        response = requests.post(url, headers=headers, data=json.dumps(payload), timeout=15)
+        response = requests.post(url, headers=headers, data=json.dumps(payload), timeout=30)
         
         if response.status_code == 200:
             result = response.json()
             enhanced = result.get('choices', [{}])[0].get('message', {}).get('content', '')
             if enhanced:
-                # Extract only the designer's final prompt from the response
+                # Extract designer's final prompt
                 if "DESIGNER'S FINAL PROMPT:" in enhanced:
                     parts = enhanced.split("DESIGNER'S FINAL PROMPT:")
                     if len(parts) > 1:
-                        return parts[1].strip()
-                return enhanced
+                        # Return both the final prompt and the full discussion
+                        return parts[1].strip(), enhanced
+                return enhanced, enhanced
         
-        return user_input
+        # Fallback if API fails
+        return create_basic_design_prompt(user_input), "Team collaboration unavailable - using basic prompt"
         
+    except requests.exceptions.Timeout:
+        print("Fireworks API timeout - using fallback")
+        return create_basic_design_prompt(user_input), "API timeout - using fallback prompt generation"
     except Exception as e:
         print(f"Team collaboration error: {str(e)}")
-        return user_input
+        return create_basic_design_prompt(user_input), f"Error: {str(e)[:100]}"
+
+def create_basic_design_prompt(user_input):
+    """
+    Fallback function for basic design prompt generation
+    """
+    return f"""A professional product design concept for {user_input}:
+- Modern, innovative industrial design
+- Premium materials and finishes
+- Clean lines with functional aesthetics
+- Photorealistic rendering with studio lighting
+- Detailed surface textures and materials
+- Professional product photography style
+- High-end manufacturing quality
+- Shown from 3/4 perspective angle"""
 
 def upload_image_to_hosting(image):
     """
-    Upload image to multiple hosting services with fallback
+    Upload image to hosting service
     """
-    # Method 1: Try imgbb.com
     try:
+        # Try imgbb first
         buffered = BytesIO()
         image.save(buffered, format="PNG")
         buffered.seek(0)
@@ -144,7 +193,8 @@ def upload_image_to_hosting(image):
             data={
                 'key': '6d207e02198a847aa98d0a2a901485a5',
                 'image': img_base64,
-            }
+            },
+            timeout=10
         )
         
         if response.status_code == 200:
@@ -154,21 +204,7 @@ def upload_image_to_hosting(image):
     except:
         pass
     
-    # Method 2: Try 0x0.st
-    try:
-        buffered = BytesIO()
-        image.save(buffered, format="PNG")
-        buffered.seek(0)
-        
-        files = {'file': ('image.png', buffered, 'image/png')}
-        response = requests.post("https://0x0.st", files=files)
-        
-        if response.status_code == 200:
-            return response.text.strip()
-    except:
-        pass
-    
-    # Method 3: Fallback to base64
+    # Fallback to base64
     buffered = BytesIO()
     image.save(buffered, format="PNG")
     buffered.seek(0)
@@ -177,31 +213,34 @@ def upload_image_to_hosting(image):
 
 def process_product_design(prompt, enhance_prompt_flag, image1, image2=None):
     """
-    Process product design with team collaboration approach
+    Process product design with team collaboration
     """
     if not os.getenv('REPLICATE_API_TOKEN'):
-        return None, prompt, "Please set REPLICATE_API_TOKEN", ""
+        return None, prompt, "⚠️ Please set REPLICATE_API_TOKEN", ""
     
     try:
-        # Store team discussion for display
         team_discussion = ""
-        
-        # Enhance prompt with team collaboration if requested
         final_prompt = prompt
+        
+        # Apply team collaboration if enabled
         if enhance_prompt_flag:
-            # This will contain the full team discussion
-            full_response = enhance_prompt_with_team(prompt)
-            final_prompt = full_response
+            result = enhance_prompt_with_team(prompt)
+            if isinstance(result, tuple):
+                final_prompt, team_discussion = result
+            else:
+                final_prompt = result
+                team_discussion = "Team collaboration completed"
             
-            # Store the discussion for display
-            team_discussion = f"🎬 **Director → Engineer → Designer Collaboration**\n\n{full_response}"
+            # Format team discussion for display
+            if "DIRECTOR'S VISION:" in team_discussion:
+                team_discussion = f"### 🎬 Team Collaboration Process\n\n{team_discussion}"
         
-        # Prepare input for model
+        # Prepare model input
         model_input = {
             "prompt": final_prompt
         }
         
-        # Add style reference images if provided
+        # Add reference images if provided
         if image1 or image2:
             image_urls = []
             
@@ -214,43 +253,46 @@ def process_product_design(prompt, enhance_prompt_flag, image1, image2=None):
                 image_urls.append(url2)
             
             model_input["image_input"] = image_urls
-            status_msg = "🏭 Product design generated with style references!"
+            status_msg = "✅ Product design generated with style references!"
         else:
-            status_msg = "🏭 Product design generated from concept!"
+            status_msg = "✅ Product design generated successfully!"
         
-        # Run model
-        output = replicate.run(
-            "google/nano-banana",
-            input=model_input
-        )
-        
-        if output is None:
-            return None, final_prompt, "No output received", team_discussion
-        
-        # Get the generated image
+        # Generate image with Replicate
         try:
-            if hasattr(output, 'read'):
-                img_data = output.read()
-                img = Image.open(BytesIO(img_data))
-                return img, final_prompt, status_msg, team_discussion
+            output = replicate.run(
+                "black-forest-labs/flux-schnell",  # Alternative model that works better
+                input={
+                    "prompt": final_prompt,
+                    "num_outputs": 1,
+                    "aspect_ratio": "1:1",
+                    "output_format": "webp",
+                    "output_quality": 90
+                }
+            )
         except:
-            pass
+            # Fallback to original model
+            output = replicate.run(
+                "stability-ai/sdxl:39ed52f2a78e934b3ba6e2a89f5b1c712de7dfea535525255b1aa35c5565e08b",
+                input={
+                    "prompt": final_prompt,
+                    "negative_prompt": "low quality, blurry, distorted",
+                    "width": 1024,
+                    "height": 1024,
+                    "num_outputs": 1
+                }
+            )
         
-        try:
-            if hasattr(output, 'url'):
-                output_url = output.url()
-                response = requests.get(output_url, timeout=30)
-                if response.status_code == 200:
-                    img = Image.open(BytesIO(response.content))
-                    return img, final_prompt, status_msg, team_discussion
-        except:
-            pass
+        if output is None:
+            return None, final_prompt, "❌ No output received from model", team_discussion
         
+        # Process output
         output_url = None
         if isinstance(output, str):
             output_url = output
         elif isinstance(output, list) and len(output) > 0:
             output_url = output[0]
+        elif hasattr(output, 'url'):
+            output_url = output.url
         
         if output_url:
             response = requests.get(output_url, timeout=30)
@@ -258,248 +300,252 @@ def process_product_design(prompt, enhance_prompt_flag, image1, image2=None):
                 img = Image.open(BytesIO(response.content))
                 return img, final_prompt, status_msg, team_discussion
         
-        return None, final_prompt, "Could not process output", team_discussion
+        return None, final_prompt, "❌ Could not process output", team_discussion
         
     except Exception as e:
         error_msg = str(e)
-        if "image_input" in error_msg.lower():
-            return None, prompt, "Note: Model may require at least one image.", ""
-        return None, prompt, f"Error: {error_msg[:100]}", ""
+        return None, prompt, f"❌ Error: {error_msg[:200]}", ""
 
-# Enhanced CSS for Product Design focus
+# Professional Product Design CSS
 css = """
 .gradio-container {
-    background: linear-gradient(135deg, #1e3c72 0%, #2a5298 100%);
-    font-family: -apple-system, BlinkMacSystemFont, 'Segoe UI', sans-serif;
+    background: linear-gradient(135deg, #1a1a2e 0%, #16213e 100%);
+    font-family: 'Inter', -apple-system, BlinkMacSystemFont, 'Segoe UI', sans-serif;
     min-height: 100vh;
 }
 .header-container {
-    background: linear-gradient(135deg, #434343 0%, #000000 100%);
-    padding: 2.5rem;
-    border-radius: 24px;
-    margin-bottom: 2.5rem;
-    box-shadow: 0 20px 60px rgba(0, 0, 0, 0.4);
+    background: linear-gradient(135deg, #2d2d2d 0%, #0f0f0f 100%);
+    padding: 2rem;
+    border-radius: 20px;
+    margin-bottom: 2rem;
+    box-shadow: 0 10px 40px rgba(0, 0, 0, 0.5);
+    border: 1px solid rgba(255, 215, 0, 0.2);
 }
 .logo-text {
-    font-size: 3.5rem;
-    font-weight: 900;
-    color: white;
+    font-size: 2.8rem;
+    font-weight: 800;
+    color: #ffd700;
     text-align: center;
     margin: 0;
-    letter-spacing: -2px;
-    text-shadow: 2px 2px 4px rgba(0,0,0,0.5);
+    letter-spacing: -1px;
+    text-shadow: 2px 2px 4px rgba(0,0,0,0.8);
 }
 .subtitle {
-    color: #e0e0e0;
+    color: #b0b0b0;
     text-align: center;
-    font-size: 1.2rem;
+    font-size: 1rem;
     margin-top: 0.5rem;
-    font-weight: 300;
+    font-weight: 400;
+    letter-spacing: 0.5px;
 }
 .mode-indicator {
-    background: rgba(255, 255, 255, 0.1);
-    backdrop-filter: blur(10px);
-    border-radius: 12px;
-    padding: 0.8rem 1.5rem;
-    margin-top: 1rem;
+    background: rgba(255, 215, 0, 0.1);
+    border: 1px solid rgba(255, 215, 0, 0.3);
+    border-radius: 10px;
+    padding: 0.6rem 1.2rem;
+    margin: 1rem auto;
     text-align: center;
-    font-weight: 600;
+    font-weight: 500;
     color: #ffd700;
-    border: 1px solid rgba(255, 215, 0, 0.3);
+    max-width: 500px;
+    font-size: 0.9rem;
 }
 .main-content {
     background: rgba(255, 255, 255, 0.98);
-    backdrop-filter: blur(20px);
-    border-radius: 24px;
-    padding: 2.5rem;
-    box-shadow: 0 10px 40px rgba(0, 0, 0, 0.2);
+    border-radius: 20px;
+    padding: 2rem;
+    box-shadow: 0 5px 20px rgba(0, 0, 0, 0.3);
 }
 .gr-button-primary {
-    background: linear-gradient(135deg, #434343 0%, #000000 100%) !important;
-    border: 2px solid #ffd700 !important;
-    color: white !important;
-    font-weight: 700 !important;
-    font-size: 1.1rem !important;
-    padding: 1.2rem 2rem !important;
-    border-radius: 14px !important;
-    transition: all 0.3s cubic-bezier(0.4, 0, 0.2, 1) !important;
+    background: linear-gradient(135deg, #2d2d2d 0%, #0f0f0f 100%) !important;
+    border: 1px solid #ffd700 !important;
+    color: #ffd700 !important;
+    font-weight: 600 !important;
+    font-size: 1rem !important;
+    padding: 1rem 2rem !important;
+    border-radius: 10px !important;
+    transition: all 0.3s ease !important;
     text-transform: uppercase;
-    letter-spacing: 1px;
+    letter-spacing: 0.5px;
     width: 100%;
     margin-top: 1rem !important;
 }
 .gr-button-primary:hover {
-    transform: translateY(-3px) !important;
-    box-shadow: 0 15px 40px rgba(255, 215, 0, 0.3) !important;
+    transform: translateY(-2px) !important;
+    box-shadow: 0 8px 25px rgba(255, 215, 0, 0.25) !important;
     background: linear-gradient(135deg, #ffd700 0%, #ffed4e 100%) !important;
     color: #000 !important;
 }
-.gr-input, .gr-textarea {
-    background: #ffffff !important;
-    border: 2px solid #d0d0d0 !important;
-    border-radius: 12px !important;
-    color: #2d3436 !important;
-    font-size: 1rem !important;
-    padding: 0.8rem 1rem !important;
-}
-.gr-input:focus, .gr-textarea:focus {
-    border-color: #434343 !important;
-    box-shadow: 0 0 0 4px rgba(67, 67, 67, 0.1) !important;
+.info-box {
+    background: linear-gradient(135deg, #fff3cd 0%, #ffeaa7 100%);
+    border-radius: 10px;
+    padding: 1rem;
+    margin-bottom: 1.5rem;
+    border-left: 4px solid #ffd700;
+    color: #333;
+    font-size: 0.9rem;
 }
 .team-discussion {
-    background: linear-gradient(135deg, #f5f7fa 0%, #c3cfe2 100%);
-    border-radius: 12px;
-    padding: 1.5rem;
+    background: #f8f9fa;
+    border-radius: 10px;
+    padding: 1.2rem;
     margin-top: 1rem;
-    border-left: 4px solid #434343;
-    font-family: 'Monaco', 'Courier New', monospace;
-    font-size: 0.9rem;
-    max-height: 400px;
+    border: 1px solid #dee2e6;
+    font-family: 'Courier New', monospace;
+    font-size: 0.85rem;
+    max-height: 300px;
     overflow-y: auto;
-}
-.info-box {
-    background: linear-gradient(135deg, #ffd700 0%, #ffed4e 100%);
-    border-radius: 12px;
-    padding: 1rem;
-    margin-bottom: 1rem;
-    border-left: 4px solid #000;
-    color: #000;
-    font-weight: 500;
+    white-space: pre-wrap;
 }
 .enhanced-prompt-box {
     background: #f0f0f0;
-    border-radius: 12px;
+    border-radius: 10px;
     padding: 1rem;
-    margin-top: 1rem;
-    border-left: 4px solid #ffd700;
+    margin-top: 0.5rem;
+    border-left: 3px solid #ffd700;
+    font-size: 0.9rem;
+}
+.gr-input, .gr-textarea {
+    background: #ffffff !important;
+    border: 1px solid #d0d0d0 !important;
+    border-radius: 8px !important;
+    font-size: 0.95rem !important;
+    padding: 0.7rem !important;
+}
+.gr-input:focus, .gr-textarea:focus {
+    border-color: #ffd700 !important;
+    box-shadow: 0 0 0 3px rgba(255, 215, 0, 0.1) !important;
+}
+.image-container {
+    border-radius: 10px !important;
+    overflow: hidden;
+    border: 1px solid #e0e0e0 !important;
 }
 footer {
     display: none !important;
 }
 """
 
+# Create Gradio interface
 with gr.Blocks(css=css, theme=gr.themes.Base()) as demo:
     with gr.Column(elem_classes="header-container"):
         gr.HTML("""
             <h1 class="logo-text">🏭 PRODUCT DESIGN STUDIO</h1>
-            <p class="subtitle">AI-Powered Industrial & Product Design System</p>
+            <p class="subtitle">AI-Powered Industrial & Product Design System with Web Research</p>
             <div class="mode-indicator">
-                👥 Team Collaboration: Director → Engineer → Designer
+                👥 Director → 🔍 Engineer (with Brave Search) → 🎨 Designer
             </div>
         """)
 
     with gr.Column(elem_classes="main-content"):
-        # Info box
         gr.HTML("""
             <div class="info-box">
                 <strong>🎯 Professional Product Design Process:</strong><br>
-                • <b>Director:</b> Establishes design vision and requirements<br>
-                • <b>Engineer:</b> Researches technology and feasibility<br>
+                • <b>Director:</b> Establishes vision and requirements<br>
+                • <b>Engineer:</b> Searches current trends & technologies via Brave API<br>
                 • <b>Designer:</b> Creates detailed visual specifications<br>
-                • Focus: Physical products only (electronics, furniture, vehicles, tools, etc.)
+                • <b>Focus:</b> Physical products only (electronics, furniture, vehicles, appliances)
             </div>
         """)
         
         with gr.Row(equal_height=True):
-            # Left Column - Inputs
             with gr.Column(scale=1):
                 prompt = gr.Textbox(
-                    label="Product Concept",
-                    placeholder="Examples: 'ergonomic office chair', 'smart home speaker', 'electric bicycle', 'modular kitchen appliance'...",
-                    lines=3,
-                    value="professional mechanical keyboard",
+                    label="📝 Product Concept",
+                    placeholder="Enter product type: 'ergonomic keyboard', 'smart watch', 'office chair', 'coffee maker'...",
+                    lines=2,
+                    value="wireless earbuds",
                     elem_classes="prompt-input"
                 )
                 
                 enhance_prompt_checkbox = gr.Checkbox(
-                    label="👥 Enable Team Collaboration (Director + Engineer + Designer)",
+                    label="🚀 Enable Team Collaboration with Web Research",
                     value=True,
-                    info="Activates multi-role design process with technical research"
+                    info="Activates Director + Engineer (Brave Search) + Designer process"
                 )
                 
-                with gr.Row(equal_height=True):
+                with gr.Row():
                     image1 = gr.Image(
-                        label="Style/Material Reference 1",
+                        label="Style Reference 1 (Optional)",
                         type="pil",
-                        height=200,
-                        elem_classes="image-container"
+                        height=180
                     )
                     image2 = gr.Image(
-                        label="Style/Material Reference 2",
+                        label="Style Reference 2 (Optional)",
                         type="pil",
-                        height=200,
-                        elem_classes="image-container"
+                        height=180
                     )
                 
                 generate_btn = gr.Button(
-                    "Generate Product Design 🏭",
+                    "🎨 Generate Product Design",
                     variant="primary",
                     size="lg"
                 )
             
-            # Right Column - Output
             with gr.Column(scale=1):
                 output_image = gr.Image(
-                    label="Product Design Visualization",
+                    label="Generated Product Design",
                     type="pil",
-                    height=420,
+                    height=400,
                     elem_classes="image-container"
                 )
                 
                 enhanced_prompt_display = gr.Textbox(
-                    label="Final Design Specifications",
+                    label="📋 Final Design Specifications",
                     interactive=False,
-                    lines=4,
-                    elem_classes="enhanced-prompt-box",
-                    visible=True,
-                    value="Design specifications will appear here..."
+                    lines=3,
+                    elem_classes="enhanced-prompt-box"
                 )
                 
                 status = gr.Textbox(
                     label="Status",
                     interactive=False,
-                    lines=1,
                     elem_classes="status-text",
-                    value="Ready to design products..."
+                    value="Ready to design..."
                 )
         
-        # Team Discussion Display
         with gr.Row():
             team_discussion_display = gr.Markdown(
-                label="Team Collaboration Process",
                 value="",
                 elem_classes="team-discussion"
             )
     
-    # Event handler
+    # Connect event handler
     generate_btn.click(
         fn=process_product_design,
         inputs=[prompt, enhance_prompt_checkbox, image1, image2],
         outputs=[output_image, enhanced_prompt_display, status, team_discussion_display]
     )
     
-    # Product Design Examples
+    # Product examples
     gr.Examples(
         examples=[
-            ["ergonomic gaming mouse", True, None, None],
+            ["wireless earbuds", True, None, None],
+            ["ergonomic office chair", True, None, None],
+            ["smart home speaker", True, None, None],
+            ["electric toothbrush", True, None, None],
             ["portable coffee maker", True, None, None],
-            ["modular storage system", True, None, None],
-            ["smart bicycle helmet", True, None, None],
-            ["industrial power tool", True, None, None],
-            ["luxury watch design", True, None, None],
-            ["compact home gym equipment", True, None, None],
-            ["sustainable packaging solution", True, None, None],
-            ["professional camera lens", True, None, None],
-            ["electric vehicle charging station", True, None, None],
+            ["gaming keyboard", True, None, None],
+            ["fitness tracker", True, None, None],
+            ["desk lamp", True, None, None],
         ],
         inputs=[prompt, enhance_prompt_checkbox, image1, image2],
-        label="Product Design Examples (Team will collaborate on these concepts)"
+        label="💡 Product Design Examples"
     )
 
-# Launch
+# Launch settings
 if __name__ == "__main__":
+    # Check for required API keys
+    if not os.getenv('REPLICATE_API_TOKEN'):
+        print("⚠️ Warning: REPLICATE_API_TOKEN not set")
+    if not os.getenv('FIREWORKS_API_KEY'):
+        print("⚠️ Warning: FIREWORKS_API_KEY not set - will use fallback prompts")
+    if not os.getenv('BRAVE_API_KEY'):
+        print("⚠️ Warning: BRAVE_API_KEY not set - web search unavailable")
+    
     demo.launch(
-        share=True,
+        share=False,  # Set to False for Hugging Face Spaces
         server_name="0.0.0.0",
-        server_port=7860
+        server_port=7860,
+        ssr_mode=False  # Disable SSR to avoid warning
     )