Update app.py

app.py

@@ -497,15 +497,159 @@ async def root():
         "oci_bucket_api": OCI_API_BASE_URL
     }
 
-# For Hugging Face Spaces
+# GRADIO INTERFACE FOR TESTING
+def create_gradio_interface():
+    """Create Gradio interface for testing image generation"""
+    
+    def generate_test_image(prompt, model_choice, style_choice):
+        """Generate a single image for testing"""
+        try:
+            if not prompt.strip():
+                return None, "❌ Please enter a prompt"
+            
+            print(f"🎨 Generating test image with prompt: {prompt}")
+            
+            # Enhance the prompt
+            enhanced_prompt, negative_prompt = enhance_prompt(prompt, style_choice)
+            
+            # Generate the image
+            image = generate_high_quality_image(
+                enhanced_prompt, 
+                model_choice, 
+                style_choice,
+                negative_prompt
+            )
+            
+            status_msg = f"""✅ Success! Generated: {prompt}
+
+🎨 Enhanced prompt: {enhanced_prompt}
+
+💡 This is a test image. For complete storybook generation, use the API endpoint:
+POST /api/generate-storybook"""
+            
+            return image, status_msg
+            
+        except Exception as e:
+            error_msg = f"❌ Generation failed: {str(e)}"
+            print(error_msg)
+            return None, error_msg
+    
+    with gr.Blocks(title="Storybook Generator - Test Interface", theme="soft") as demo:
+        gr.Markdown("# 🎨 Storybook Generator - Test Interface")
+        gr.Markdown("**Test image generation • Full storybooks via API**")
+        
+        with gr.Row():
+            with gr.Column(scale=1):
+                gr.Markdown("### ⚙️ Test 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"
+                )
+                
+                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")
+                
+                gr.Markdown("### 📚 API Usage")
+                gr.Markdown("""
+                **For complete storybooks with multiple images:**
+                - Use: `POST /api/generate-storybook`
+                - Send: story_title, scenes[], characters[]
+                - Returns: job_id for status tracking
+                """)
+            
+            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
+            )
+        
+        # API info section
+        with gr.Accordion("🔧 API Documentation", open=False):
+            gr.Markdown("""
+            ### API Endpoints
+            
+            **Generate Storybook**
+            ```bash
+            POST /api/generate-storybook
+            {
+              "story_title": "Your Story",
+              "scenes": [
+                {
+                  "visual": "Scene description",
+                  "text": "Story text"
+                }
+              ],
+              "callback_url": "your-webhook-url"
+            }
+            ```
+            
+            **Check Status**
+            ```bash
+            GET /api/job-status/{job_id}
+            ```
+            
+            **Health Check**
+            ```bash
+            GET /api/health
+            ```
+            """)
+        
+        generate_btn.click(
+            fn=generate_test_image,
+            inputs=[prompt_input, model_dropdown, style_dropdown],
+            outputs=[image_output, status_output]
+        )
+    
+    return demo
+
+# Create Gradio app
+gradio_app = create_gradio_interface()
+
+# For Hugging Face Spaces deployment
 def get_app():
     return app
 
 if __name__ == "__main__":
     import uvicorn
-    print("🚀 Starting Storybook Generator API...")
+    print("🚀 Starting Storybook Generator API + Gradio Interface...")
     print("📚 Features: Multi-image generation + Text saving + OCI bucket storage")
     print("🔗 OCI API:", OCI_API_BASE_URL)
-    print("🌐 API ready at: http://localhost:7860")
+    print("🌐 Web Interface: http://localhost:7860")
+    print("🔌 API Endpoints: http://localhost:7860/api/")
     
-    uvicorn.run(app, host="0.0.0.0", port=7860)
+    # Launch both FastAPI and Gradio
+    gradio_app.launch(server_name="0.0.0.0", server_port=7860)