Update app.py

app.py

@@ -18,6 +18,7 @@ import threading
 import uuid
 import hashlib
 from enum import Enum
+import random
 
 # External OCI API URL
 OCI_API_BASE_URL = "https://yukee1992-oci-story-book.hf.space"
@@ -202,7 +203,6 @@ def enhance_prompt(prompt, style="childrens_book"):
     ]
     
     # Add 2-3 random quality boosters
-    import random
     boosters = random.sample(quality_boosters, 2)
     enhanced += ", " + ", ".join(boosters)
     
@@ -523,76 +523,166 @@ async def list_jobs():
         }
     }
 
-
-# GRADIO INTERFACE (Optional - for manual testing)
-def create_gradio_interface():
-    """Create Gradio interface for manual testing"""
-    with gr.Blocks(title="High-Quality Storybook Generator", theme="soft") as demo:
-        gr.Markdown("# 🎨 High-Quality Storybook Generator")
-        gr.Markdown("Generate **studio-quality** storybook images with professional results")
+# SIMPLIFIED GRADIO INTERFACE FOR TESTING
+def create_test_interface():
+    """Create a simple 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
+            )
+            
+            # Create a simple save (optional)
+            try:
+                timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
+                safe_prompt = "".join(c for c in prompt[:20] if c.isalnum() or c in (' ', '-', '_')).rstrip()
+                filename = f"test_{safe_prompt}_{timestamp}.png"
+                image.save(f"/tmp/{filename}")
+                print(f"💾 Test image saved as: {filename}")
+            except Exception as e:
+                print(f"⚠️ Could not save test image: {e}")
+            
+            status_msg = f"✅ Success! Generated: {prompt}\n\nEnhanced prompt: {enhanced_prompt}"
+            return image, status_msg
+            
+        except Exception as e:
+            error_msg = f"❌ Generation failed: {str(e)}"
+            print(error_msg)
+            return None, error_msg
+    
+    # Create the interface
+    with gr.Blocks(title="High-Quality Image Generator", theme="soft") as demo:
+        gr.Markdown("# 🎨 High-Quality Image Generator")
+        gr.Markdown("Test your prompts with **studio-quality** image generation")
         
         with gr.Row():
             with gr.Column(scale=1):
-                gr.Markdown("### 🎯 Quality Settings")
+                gr.Markdown("### ⚙️ Settings")
                 
-                model_choice = gr.Dropdown(
+                model_dropdown = gr.Dropdown(
                     label="AI Model",
                     choices=list(MODEL_CHOICES.keys()),
                     value="dreamshaper-8",
-                    info="Choose the best model for your style"
+                    info="Choose the model for generation"
                 )
                 
-                style_choice = gr.Dropdown(
+                style_dropdown = gr.Dropdown(
                     label="Art Style",
                     choices=["childrens_book", "realistic", "fantasy", "anime"],
                     value="childrens_book",
                     info="Select the artistic style"
                 )
                 
-                story_title = gr.Textbox(
-                    label="Story Title",
-                    placeholder="Enter your story title..."
-                )
-                
-                scene_input = gr.Textbox(
-                    label="Scene Description",
-                    placeholder="Describe your scene in detail...",
+                prompt_input = gr.Textbox(
+                    label="Prompt",
+                    placeholder="Describe what you want to generate...\nExample: A friendly dragon reading a book under a magical tree",
                     lines=3
                 )
                 
-                generate_btn = gr.Button("✨ Generate Premium Image", variant="primary")
+                generate_btn = gr.Button("✨ Generate Image", variant="primary", size="lg")
+                
+                gr.Markdown("### 💡 Tips")
+                gr.Markdown("""
+                - Be descriptive with colors, emotions, settings
+                - Example: **"A cute baby dragon with green scales reading a giant book under a glowing mushroom"**
+                - Avoid single words: use **"friendly cartoon dragon"** instead of just **"dragon"**
+                """)
             
             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=3)
+                image_output = gr.Image(
+                    label="Generated Image", 
+                    height=500, 
+                    show_download_button=True,
+                    show_share_button=True
+                )
+                status_output = gr.Textbox(
+                    label="Generation Status", 
+                    interactive=False, 
+                    lines=4,
+                    show_copy_button=True
+                )
         
-        def generate_single_image(story_title, scene, model, style):
-            """Generate a single high-quality image for testing"""
-            try:
-                prompt, negative_prompt = enhance_prompt(scene, style)
-                image = generate_high_quality_image(prompt, model, style, negative_prompt)
-                return image, f"✅ High-quality image generated for: {story_title}"
-            except Exception as e:
-                return None, f"❌ Error: {str(e)}"
+        # Examples section
+        with gr.Accordion("🎯 Example Prompts", open=False):
+            gr.Markdown("""
+            **Try these examples:**
+            - "A friendly dragon reading a giant book under a magical tree with glowing fairies"
+            - "Cute kitten playing with yarn ball in a sunny living room"
+            - "Space astronaut exploring a colorful alien planet with strange creatures"
+            - "Magical castle in the clouds with rainbows and flying unicorns"
+            """)
         
+        # Connect the button
         generate_btn.click(
-            fn=generate_single_image,
-            inputs=[story_title, scene_input, model_choice, style_choice],
+            fn=generate_test_image,
+            inputs=[prompt_input, model_dropdown, style_dropdown],
             outputs=[image_output, status_output]
         )
         
-        return demo
+        # Quick test buttons
+        gr.Markdown("### 🚀 Quick Test Prompts")
+        with gr.Row():
+            quick_btn1 = gr.Button("🐉 Dragon Example", size="sm")
+            quick_btn2 = gr.Button("🐱 Kitten Example", size="sm")
+            quick_btn3 = gr.Button("🚀 Space Example", size="sm")
+        
+        def set_quick_prompt1():
+            return "A friendly dragon reading a giant book under a magical tree with glowing fairies"
+        
+        def set_quick_prompt2():
+            return "Cute kitten playing with yarn ball in a sunny living room with sunbeams"
+        
+        def set_quick_prompt3():
+            return "Space astronaut exploring a colorful alien planet with strange glowing plants"
+        
+        quick_btn1.click(fn=set_quick_prompt1, outputs=prompt_input)
+        quick_btn2.click(fn=set_quick_prompt2, outputs=prompt_input)
+        quick_btn3.click(fn=set_quick_prompt3, outputs=prompt_input)
+    
+    return demo
+
+# Create the Gradio app
+gradio_app = create_test_interface()
+
+# Mount Gradio interface to FastAPI
+@app.get("/")
+async def root():
+    return {"message": "Storybook Generator API is running! Visit /gradio for the testing interface."}
+
+@app.get("/gradio")
+async def gradio_redirect():
+    """Redirect to Gradio interface"""
+    import urllib.parse
+    gradio_url = "http://localhost:7860" if __name__ == "__main__" else "/gradio"
+    return {"message": "Visit the Gradio interface at the main URL"}
 
 # For Hugging Face Spaces deployment
 def get_app():
     return app
 
+# Launch both FastAPI and Gradio
 if __name__ == "__main__":
     import uvicorn
-    print("🚀 Starting High-Quality Storybook Generator API...")
+    print("🚀 Starting High-Quality Storybook Generator...")
     print("🎨 Available models:")
     for model in MODEL_CHOICES:
         print(f"   - {model}")
+    print("🌐 Web interface available at: http://localhost:7860")
+    print("🔌 API endpoints available at: http://localhost:7860/api/")
     
-    # Start both FastAPI and Gradio (if needed)
-    uvicorn.run(app, host="0.0.0.0", port=7860)
+    # Launch Gradio interface
+    gradio_app.launch(server_name="0.0.0.0", server_port=7860, share=False)