Update app.py

app.py

@@ -973,4 +973,315 @@ def create_gradio_interface():
             
         except Exception as e:
             error_msg = f"❌ Generation failed: {str(e)}"
-       
+            print(error_msg)
+            return None, error_msg, None
+    
+    with gr.Blocks(title="Premium Children's Book Illustrator with Dynamic Character Consistency", theme="soft") as demo:
+        gr.Markdown("# 🎨 Premium Children's Book Illustrator")
+        gr.Markdown("Generate **studio-quality** storybook images with **dynamic character consistency**")
+        
+        # Storage info display
+        storage_info = gr.Textbox(
+            label="📊 Local Storage Information",
+            interactive=False,
+            lines=2
+        )
+        
+        def update_storage_info():
+            info = get_local_storage_info()
+            if "error" not in info:
+                return f"📁 Local Storage: {info['total_files']} images, {info['total_size_mb']} MB used"
+            return "📁 Local Storage: Unable to calculate"
+        
+        with gr.Row():
+            with gr.Column(scale=1):
+                gr.Markdown("### 🎯 Quality Settings")
+                
+                model_dropdown = gr.Dropdown(
+                    label="AI Model",
+                    choices=list(MODEL_CHOICES.keys()),
+                    value="dreamshaper-8"
+                )
+                
+                style_dropdown = gr.Dropdown(
+                    label="Art Style",
+                    choices=["childrens_book", "realistic", "fantasy", "anime"],
+                    value="childrens_book"
+                )
+                
+                # Dynamic character input for testing
+                character_names_input = gr.Textbox(
+                    label="Character Names (comma-separated)",
+                    placeholder="Enter character names: Sparkle the Star Cat, Benny the Bunny, Tilly the Turtle",
+                    info="Enter character names to test consistency features",
+                    lines=2
+                )
+                
+                prompt_input = gr.Textbox(
+                    label="Scene Description",
+                    placeholder="Describe your scene with character interactions...\nExample: Sparkle the Star Cat chasing butterflies while Benny the Bunny watches",
+                    lines=3
+                )
+                
+                generate_btn = gr.Button("✨ Generate Premium Image", variant="primary")
+                
+                # Current image management
+                current_file_path = gr.State()
+                delete_btn = gr.Button("🗑️ Delete This Image", variant="stop")
+                delete_status = gr.Textbox(label="Delete Status", interactive=False, lines=2)
+                
+                gr.Markdown("### 📚 API Usage for n8n")
+                gr.Markdown("""
+                **For complete storybooks (OCI bucket):**
+                - Endpoint: `POST /api/generate-storybook`
+                - Input: `story_title`, `scenes[]`, `characters[]`
+                - Output: Saves to OCI bucket with dynamic character consistency
+                """)
+            
+            with gr.Column(scale=2):
+                image_output = gr.Image(label="Generated Image", height=500, show_download_button=True)
+                status_output = gr.Textbox(label="Status", interactive=False, lines=4)
+        
+        # Dynamic character guidance section
+        with gr.Accordion("👥 Dynamic Character Guidance", open=False):
+            gr.Markdown("""
+            ### How to Use Dynamic Characters from n8n:
+            
+            **n8n Payload Structure:**
+            ```json
+            {
+              "story_title": "Your Story Title",
+              "characters": [
+                {
+                  "name": "Character Name",
+                  "description": "Character description...",
+                  "visual_prompt": "Detailed visual description",  // Optional
+                  "key_features": ["feature1", "feature2"]         // Optional
+                }
+              ],
+              "scenes": [
+                {
+                  "visual": "Scene description with characters...",
+                  "text": "Scene text...",
+                  "characters_present": ["Character Name"]         // Optional
+                }
+              ]
+            }
+            ```
+            
+            **Features:**
+            - ✅ Dynamic character processing from n8n
+            - ✅ Automatic visual prompt generation
+            - ✅ Key feature extraction
+            - ✅ Cross-scene consistency
+            - ✅ Flexible character numbers and types
+            """)
+        
+        # Examples section
+        with gr.Accordion("💡 Prompt Examples & Tips", open=False):
+            gr.Markdown("""
+            ## 🎨 Professional Prompt Examples with Dynamic Characters:
+            
+            **Best Results with Dynamic Characters:**
+            - "Sparkle the Star Cat chasing butterflies in a sunny meadow"
+            - "Benny the Bunny and Tilly the Turtle having a picnic"
+            - "Multiple characters discovering a magical portal together"
+            
+            ## ⚡ Dynamic Character Consistency Tips:
+            1. **Always mention character names** in your prompts
+            2. **n8n will send character details** automatically
+            3. **The system processes any number** of characters dynamically
+            4. **Consistency is maintained** across all scenes automatically
+            """)
+        
+        # Local file management section
+        with gr.Accordion("📁 Manage Local Test Images", open=True):
+            gr.Markdown("### Locally Saved Images")
+            
+            with gr.Row():
+                refresh_btn = gr.Button("🔄 Refresh List")
+                clear_all_btn = gr.Button("🗑️ Clear All Images", variant="stop")
+            
+            file_gallery = gr.Gallery(
+                label="Local Images",
+                show_label=True,
+                elem_id="gallery",
+                columns=4,
+                height="auto"
+            )
+            
+            clear_status = gr.Textbox(label="Clear Status", interactive=False)
+        
+        # Debug section
+        with gr.Accordion("🔧 Advanced Settings", open=False):
+            debug_btn = gr.Button("🔄 Check System Status", variant="secondary")
+            debug_output = gr.Textbox(label="System Info", interactive=False, lines=4)
+        
+        def check_system_status():
+            """Check system status"""
+            active_jobs = len(job_storage)
+            return f"""**System Status:**
+- Model: {current_model_name}
+- Dynamic Character Processing: ✅ Enabled
+- Fallback Templates: {len(FALLBACK_CHARACTER_TEMPLATES)} available
+- OCI API: {OCI_API_BASE_URL}
+- Local Storage: {get_local_storage_info().get('total_files', 0)} images
+- Active Jobs: {active_jobs}
+- Ready for dynamic character consistency generation!"""
+
+        # Connect buttons to functions
+        generate_btn.click(
+            fn=generate_test_image_with_characters,
+            inputs=[prompt_input, model_dropdown, style_dropdown, character_names_input],
+            outputs=[image_output, status_output, current_file_path]
+        ).then(
+            fn=refresh_local_images,
+            outputs=file_gallery
+        ).then(
+            fn=update_storage_info,
+            outputs=storage_info
+        )
+        
+        delete_btn.click(
+            fn=delete_current_image,
+            inputs=current_file_path,
+            outputs=[delete_status, image_output, status_output, file_gallery]
+        ).then(
+            fn=update_storage_info,
+            outputs=storage_info
+        )
+        
+        refresh_btn.click(
+            fn=refresh_local_images,
+            outputs=file_gallery
+        ).then(
+            fn=update_storage_info,
+            outputs=storage_info
+        )
+        
+        clear_all_btn.click(
+            fn=clear_all_images,
+            outputs=[clear_status, file_gallery]
+        ).then(
+            fn=update_storage_info,
+            outputs=storage_info
+        )
+        
+        debug_btn.click(
+            fn=check_system_status,
+            inputs=None,
+            outputs=debug_output
+        )
+        
+        # Initialize on load
+        demo.load(fn=refresh_local_images, outputs=file_gallery)
+        demo.load(fn=update_storage_info, outputs=storage_info)
+    
+    return demo
+
+# Create enhanced Gradio app
+demo = create_gradio_interface()
+
+# Enhanced root endpoint that explains the API structure
+@app.get("/")
+async def root():
+    return {
+        "message": "Storybook Generator API with Dynamic Character Consistency is running!",
+        "api_endpoints": {
+            "health_check": "GET /api/health",
+            "generate_storybook": "POST /api/generate-storybook",
+            "check_job_status": "GET /api/job-status/{job_id}",
+            "local_images": "GET /api/local-images"
+        },
+        "features": {
+            "dynamic_characters": "✅ Enabled",
+            "character_consistency": "✅ Enabled", 
+            "flexible_storytelling": "✅ Enabled",
+            "n8n_integration": "✅ Enabled"
+        },
+        "web_interface": "GET /ui",
+        "note": "Use API endpoints for programmatic access with dynamic characters from n8n"
+    }
+
+# Add a simple test endpoint
+@app.get("/api/test")
+async def test_endpoint():
+    return {
+        "status": "success", 
+        "message": "API with dynamic character consistency is working correctly",
+        "dynamic_processing": "✅ Enabled",
+        "fallback_templates": len(FALLBACK_CHARACTER_TEMPLATES),
+        "timestamp": datetime.now().isoformat()
+    }
+
+# For Hugging Face Spaces deployment
+def get_app():
+    return app
+
+if __name__ == "__main__":
+    import uvicorn
+    import os
+    
+    # Check if we're running on Hugging Face Spaces
+    HF_SPACE = os.environ.get('SPACE_ID') is not None
+    
+    if HF_SPACE:
+        print("🚀 Running on Hugging Face Spaces - Integrated Mode")
+        print("📚 API endpoints available at: /api/*")
+        print("🎨 Web interface available at: /ui")
+        print("👥 Dynamic character consistency features enabled")
+        print("🔌 Both API and UI running on same port")
+        
+        # Mount Gradio without reassigning app
+        gr.mount_gradio_app(app, demo, path="/ui")
+        
+        # Run the combined app
+        uvicorn.run(
+            app, 
+            host="0.0.0.0", 
+            port=7860, 
+            log_level="info"
+        )
+    else:
+        # Local development - run separate servers
+        print("🚀 Running locally - Separate API and UI servers")
+        print("📚 API endpoints: http://localhost:8000/api/*")
+        print("🎨 Web interface: http://localhost:7860/ui")
+        print("👥 Dynamic character consistency features enabled")
+        
+        def run_fastapi():
+            """Run FastAPI on port 8000 for API calls"""
+            uvicorn.run(
+                app, 
+                host="0.0.0.0", 
+                port=8000, 
+                log_level="info",
+                access_log=False
+            )
+        
+        def run_gradio():
+            """Run Gradio on port 7860 for web interface"""
+            demo.launch(
+                server_name="0.0.0.0",
+                server_port=7860,
+                share=False,
+                show_error=True,
+                quiet=True
+            )
+        
+        # Start both servers in separate threads
+        api_thread = threading.Thread(target=run_fastapi, daemon=True)
+        ui_thread = threading.Thread(target=run_gradio, daemon=True)
+        
+        api_thread.start()
+        print("✅ FastAPI server started on port 8000")
+        
+        ui_thread.start()
+        print("✅ Gradio server started on port 7860")
+        
+        # Keep the main thread alive
+        try:
+            while True:
+                time.sleep(1)
+        except KeyboardInterrupt:
+            print("🛑 Shutting down servers...")