Update app.py

app.py

@@ -1,6 +1,6 @@
 """
 Marketing Image Generator with Gradio MCP Server
-Professional AI image generation using Google Imagen4 with marketing review
+Professional AI image generation using Google Imagen3 with marketing review
 Deployed on HuggingFace Spaces with built-in MCP server support
 """
 
@@ -606,7 +606,16 @@ def process_generated_image_and_results(api_response_str: str) -> Tuple[Image.Im
         response_data = json.loads(api_response_str)
         
         if not response_data.get('success', False):
-            return None, f"❌ Generation failed: {response_data.get('error', 'Unknown error')}"
+            error_msg = response_data.get('error', 'Unknown error')
+            
+            # Add specific documentation links for common errors
+            doc_link = ""
+            if any(brand in error_msg.lower() for brand in ['hsbc', 'bank', 'political', 'timeout', 'stall']):
+                doc_link = "\n\n📖 See content restrictions guide: https://huggingface.co/spaces/CognizantAI/marketing-image-generator/blob/main/README.md#content-policy--brand-restrictions"
+            elif 'api' in error_msg.lower() or 'key' in error_msg.lower():
+                doc_link = "\n\n📖 See API troubleshooting: https://huggingface.co/spaces/CognizantAI/marketing-image-generator/blob/main/README.md#common-issues"
+            
+            return None, f"❌ Generation failed: {error_msg}{doc_link}"
         
         # Extract image data
         image_info = response_data.get('image', {})
@@ -720,7 +729,7 @@ def gradio_generate_marketing_image(prompt: str, style: str, max_retries: int, r
         )
         return process_generated_image_and_results(result_json)
     except Exception as e:
-        error_message = f"❌ Error: {str(e)}"
+        error_message = f"❌ Error: {str(e)}\n\n📖 For troubleshooting help, see: https://huggingface.co/spaces/CognizantAI/marketing-image-generator/blob/main/README.md#content-policy--brand-restrictions"
         logger.error(error_message)
         return None, error_message
 
@@ -765,7 +774,7 @@ with gr.Blocks(title="Marketing Image Generator MCP", theme=gr.themes.Soft()) as
             )
             
             review_guidelines = gr.Textbox(
-                label="🔍 Marketing/Style Guide Review Guidelines (Optional)",
+                label="🔍 Marketing Review Guidelines (Optional)",
                 placeholder="e.g., All text must be in English only, focus on professional appearance, ensure brand colors are prominent",
                 lines=3,
                 info="Provide specific marketing guidelines for review"
@@ -853,7 +862,7 @@ with gr.Blocks(title="Marketing Image Generator MCP", theme=gr.themes.Soft()) as
     <div style='text-align: center; color: #666; font-size: 0.9rem;'>
         <p>🎨 Marketing Image Generator | Gradio MCP Server</p>
         <p>Image Generation + Marketing Review + MCP API</p>
-        <p>MCP Endpoint: <code>/gradio_api/mcp/sse</code></p>
+        <p>📖 <a href="https://huggingface.co/spaces/CognizantAI/marketing-image-generator/blob/main/README.md" target="_blank">Full Documentation & Troubleshooting</a> | MCP Endpoint: <code>/gradio_api/mcp/sse</code></p>
     </div>
     """)