Update app.py

app.py

@@ -587,9 +587,112 @@ async def delete_local_image_api(filename: str):
     except Exception as e:
         return {"status": "error", "message": str(e)}
 
-# GRADIO INTERFACE WITH LOCAL FILE MANAGEMENT
-def create_gradio_interface():
-    """Create Gradio interface with complete local file management"""
+# Create the Gradio interface with QUALITY OPTIONS and LOCAL FILE MANAGEMENT
+with gr.Blocks(title="Premium Children's Book Illustrator", theme="soft") as demo:
+    gr.Markdown("# 🎨 Premium Children's Book Illustrator")
+    gr.Markdown("Generate **studio-quality** storybook images with professional results")
+    
+    # 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_choice = gr.Dropdown(
+                label="AI Model",
+                choices=list(MODEL_CHOICES.keys()),
+                value="dreamshaper-8",
+                info="Choose the best model for your style"
+            )
+            
+            style_choice = gr.Dropdown(
+                label="Art Style",
+                choices=["childrens_book", "realistic", "fantasy", "anime"],
+                value="childrens_book",
+                info="Select the artistic style"
+            )
+            
+            prompt_input = gr.Textbox(
+                label="Scene Description",
+                placeholder="Describe your scene in detail...\nExample: A friendly dragon reading a giant book under a magical tree with glowing fairies",
+                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 automatically
+            """)
+        
+        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)
+    
+    # Examples section
+    with gr.Accordion("💡 Prompt Examples & Tips", open=False):
+        gr.Markdown("""
+        ## 🎨 Professional Prompt Examples:
+        
+        **Best Results:**
+        - "A cute baby dragon learning to read with an old wise turtle in a magical forest"
+        - "Group of animal friends having a picnic under a giant glowing mushroom"
+        - "Little girl exploring a castle made of candy with talking animals"
+        - "Space adventure with friendly robots and colorful aliens"
+        
+        ## 🚫 Avoid:
+        - "dragon" → **"friendly cartoon dragon reading book"**
+        - "cat" → **"cute kitten playing with yarn ball"**
+        - "tree" → **"magical tree with glowing fairies"**
+        
+        ## ⚡ Pro Tips:
+        1. **Be descriptive** - Add colors, emotions, settings
+        2. **Specify style** - "watercolor", "cartoon", "realistic"
+        3. **Add details** - "with glowing effects", "in a magical forest"
+        4. **Use positive language** - "happy", "friendly", "colorful"
+        """)
+    
+    # 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 generate_test_image(prompt, model_choice, style_choice):
         """Generate a single image for testing and save locally"""
@@ -640,7 +743,6 @@ def create_gradio_interface():
         
         success, message = delete_local_image(filepath)
         updated_files = refresh_local_images()
-        storage_info = get_local_storage_info()
         
         if success:
             status_msg = f"✅ {message}"
@@ -665,151 +767,62 @@ def create_gradio_interface():
         except Exception as e:
             return f"❌ Error: {str(e)}", refresh_local_images()
     
-    with gr.Blocks(title="Storybook Generator - Complete Interface", theme="soft") as demo:
-        gr.Markdown("# 🎨 Storybook Generator - Complete Interface")
-        gr.Markdown("**Test images (local) • Storybooks (OCI) • File management**")
-        
-        # 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("### ⚙️ Test Image Generation")
-                
-                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"
-                )
-                
-                prompt_input = gr.Textbox(
-                    label="Test Prompt",
-                    placeholder="Enter a prompt to test image generation...",
-                    lines=3
-                )
-                
-                generate_btn = gr.Button("✨ Generate Test 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 automatically
-                """)
-            
-            with gr.Column(scale=2):
-                image_output = gr.Image(
-                    label="Generated Image", 
-                    height=400, 
-                    show_download_button=True
-                )
-                status_output = gr.Textbox(
-                    label="Status", 
-                    lines=4,
-                    show_copy_button=True
-                )
-        
-        # Quick examples
-        with gr.Row():
-            gr.Button("🐉 Dragon Example").click(
-                lambda: "A friendly dragon reading a giant book under a magical tree",
-                outputs=prompt_input
-            )
-            gr.Button("🐱 Kitten Example").click(
-                lambda: "Cute kitten playing with yarn ball in a sunny living room",
-                outputs=prompt_input
-            )
-            gr.Button("🚀 Space Example").click(
-                lambda: "Space astronaut exploring a colorful alien planet",
-                outputs=prompt_input
-            )
-        
-        # 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)
-        
-        # Connect events
-        generate_btn.click(
-            fn=generate_test_image,
-            inputs=[prompt_input, model_dropdown, style_dropdown],
-            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
-        )
-        
-        # Initialize on load
-        demo.load(fn=refresh_local_images, outputs=file_gallery)
-        demo.load(fn=update_storage_info, outputs=storage_info)
+    def check_system_status():
+        """Check system status"""
+        return f"""**System Status:**
+- Model: {current_model_name}
+- OCI API: {OCI_API_BASE_URL}
+- Local Storage: {get_local_storage_info().get('total_files', 0)} images
+- Active Jobs: {len(job_storage)}
+- Ready for premium image generation!"""
+
+    # Connect buttons to functions
+    generate_btn.click(
+        fn=generate_test_image,
+        inputs=[prompt_input, model_choice, style_choice],
+        outputs=[image_output, status_output, current_file_path]
+    ).then(
+        fn=refresh_local_images,
+        outputs=file_gallery
+    ).then(
+        fn=update_storage_info,
+        outputs=storage_info
+    )
     
-    return demo
-
-# Create Gradio app
-gradio_app = create_gradio_interface()
+    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)
 
 # Enhanced root endpoint that explains the API structure
 @app.get("/")
@@ -853,7 +866,7 @@ if __name__ == "__main__":
         print("🔌 Both API and UI running on same port")
         
         # FIX: Properly mount Gradio without reassigning app
-        gr.mount_gradio_app(app, gradio_app, path="/ui")
+        gr.mount_gradio_app(app, demo, path="/ui")
         
         # Run the combined app
         uvicorn.run(
@@ -880,7 +893,7 @@ if __name__ == "__main__":
         
         def run_gradio():
             """Run Gradio on port 7860 for web interface"""
-            gradio_app.launch(
+            demo.launch(
                 server_name="0.0.0.0",
                 server_port=7860,
                 share=False,