Update app.py

app.py

@@ -89,6 +89,7 @@ def enhance_prompt_with_team(user_input):
 - Creates detailed visual specifications
 - Specifies exact materials, finishes, colors
 - Defines rendering style and atmosphere
+- MUST include text labels and annotations in the design
 
 ## OUTPUT FORMAT:
 
@@ -99,9 +100,14 @@ def enhance_prompt_with_team(user_input):
 [Based on search data, provide 3-4 technical recommendations]
 
 **DESIGNER'S FINAL PROMPT:**
-[Detailed visual description for image generation - focus on visual elements]
+[Detailed visual description INCLUDING text labels and annotations. MUST specify:
+- Product name/model as title text
+- Key feature callouts with arrow annotations
+- Technical specifications as overlay text
+- Material/component labels
+Example: "with text overlay showing 'MODEL X-500' at top, feature callouts pointing to 'Titanium Frame', 'OLED Display', '5000mAh Battery' with arrows"]
 
-Keep responses concise but specific. Focus on PHYSICAL PRODUCTS only."""
+Keep responses concise but specific. Focus on PHYSICAL PRODUCTS only with TEXT ANNOTATIONS."""
         
         # Create collaborative prompt with search results
         user_prompt = f"""Product concept: {user_input}
@@ -112,9 +118,9 @@ ENGINEER'S RESEARCH DATA:
 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
+3. Designer creates a detailed visual prompt WITH TEXT LABELS AND ANNOTATIONS
 
-Focus on creating a manufacturable, innovative product design."""
+Focus on creating a manufacturable, innovative product design with clear text overlays showing key features, specifications, and component labels."""
         
         payload = {
             "model": "accounts/fireworks/models/qwen3-235b-a22b-instruct-2507",  # Original model maintained
@@ -165,17 +171,24 @@ Focus on creating a manufacturable, innovative product design."""
 
 def create_basic_design_prompt(user_input):
     """
-    Fallback function for basic design prompt generation
+    Fallback function for basic design prompt generation with text annotations
     """
-    return f"""A professional product design concept for {user_input}:
-- Modern, innovative industrial design
-- Premium materials and finishes
-- Clean lines with functional aesthetics
+    product_name = user_input.upper().replace(" ", "-")
+    return f"""A professional product design concept for {user_input} with detailed text annotations:
+- Product title text overlay: "{product_name} PRO 2025" in bold modern font at top
+- Feature callout labels with arrows pointing to key components:
+  * "Premium Materials" → body
+  * "Ergonomic Design" → form factor  
+  * "Smart Integration" → control panel
+  * "Sustainable Manufacturing" → eco-friendly elements
+- Technical specifications text overlay in corner:
+  * Dimensions, weight, power specs
+- Clean industrial design with annotations
 - Photorealistic rendering with studio lighting
-- Detailed surface textures and materials
-- Professional product photography style
-- High-end manufacturing quality
-- Shown from 3/4 perspective angle"""
+- Professional product photography style with infographic elements
+- Text labels in clean sans-serif font (Helvetica/Arial style)
+- High contrast text for readability
+- Shown from 3/4 perspective with all labels clearly visible"""
 
 def upload_image_to_hosting(image):
     """
@@ -259,10 +272,13 @@ def process_product_design(prompt, enhance_prompt_flag, image1, image2=None):
         
         # Generate image with Replicate
         try:
+            # Add text annotation emphasis to the prompt
+            annotated_prompt = f"{final_prompt}. Include clear text labels, feature callouts with arrows, product name/model as title, technical specifications overlay, component annotations in clean modern typography"
+            
             output = replicate.run(
                 "black-forest-labs/flux-schnell",  # Alternative model that works better
                 input={
-                    "prompt": final_prompt,
+                    "prompt": annotated_prompt,
                     "num_outputs": 1,
                     "aspect_ratio": "1:1",
                     "output_format": "webp",
@@ -270,12 +286,14 @@ def process_product_design(prompt, enhance_prompt_flag, image1, image2=None):
                 }
             )
         except:
-            # Fallback to original model
+            # Fallback to original model with text annotations
+            annotated_prompt = f"{final_prompt}. With text overlays showing product name, key features with arrow callouts, specifications, and component labels in professional typography"
+            
             output = replicate.run(
                 "stability-ai/sdxl:39ed52f2a78e934b3ba6e2a89f5b1c712de7dfea535525255b1aa35c5565e08b",
                 input={
-                    "prompt": final_prompt,
-                    "negative_prompt": "low quality, blurry, distorted",
+                    "prompt": annotated_prompt,
+                    "negative_prompt": "low quality, blurry, distorted, unreadable text, messy labels",
                     "width": 1024,
                     "height": 1024,
                     "num_outputs": 1
@@ -431,20 +449,21 @@ 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 with Web Research</p>
+            <p class="subtitle">AI-Powered Industrial Design with Annotated Feature Guides</p>
             <div class="mode-indicator">
-                👥 Director → 🔍 Engineer (with Brave Search) → 🎨 Designer
+                👥 Director → 🔍 Engineer (Research) → 🎨 Designer (with Text Annotations)
             </div>
         """)
 
     with gr.Column(elem_classes="main-content"):
         gr.HTML("""
             <div class="info-box">
-                <strong>🎯 Professional Product Design Process:</strong><br>
+                <strong>🎯 Professional Product Design Process with Annotations:</strong><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>
-                • <b>Focus:</b> Physical products only (electronics, furniture, vehicles, appliances)
+                • <b>Designer:</b> Creates visual specs with <b>text labels & feature callouts</b><br>
+                • <b>Output:</b> Annotated product designs with key features highlighted<br>
+                • <b>Focus:</b> Physical products with infographic-style annotations
             </div>
         """)