Update app.py

app.py

@@ -10,6 +10,8 @@ import re
 import tempfile
 import time
 import base64
+import json
+from typing import Dict, List, Tuple
 
 # External OCI API URL
 OCI_API_BASE_URL = "https://yukee1992-oci-story-book.hf.space"
@@ -25,13 +27,16 @@ except ImportError:
 
 # HIGH-QUALITY MODEL SELECTION
 MODEL_CHOICES = {
-    "dreamshaper-8": "lykon/dreamshaper-8",  # Excellent for fantasy/artistic
-    "realistic-vision": "SG161222/Realistic_Vision_V5.1",  # Photorealistic
-    "anything-v5": "andite/anything-v5.0",  # Anime/illustration style
-    "openjourney": "prompthero/openjourney",  # Artistic/Midjourney style
-    "sd-2.1": "stabilityai/stable-diffusion-2-1",  # General purpose
+    "dreamshaper-8": "lykon/dreamshaper-8",
+    "realistic-vision": "SG161222/Realistic_Vision_V5.1", 
+    "anything-v5": "andite/anything-v5.0",
+    "openjourney": "prompthero/openjourney",
+    "sd-2.1": "stabilityai/stable-diffusion-2-1",
 }
 
+# Story tracking for sequence numbering
+story_registry: Dict[str, int] = {}  # {story_title: current_sequence}
+
 # Initialize the HIGH-QUALITY Stable Diffusion model
 def load_model(model_name="dreamshaper-8"):
     """Load and return a high-quality Stable Diffusion model"""
@@ -46,7 +51,6 @@ def load_model(model_name="dreamshaper-8"):
             requires_safety_checker=False
         )
         
-        # Use better scheduler for quality
         pipe.scheduler = EulerAncestralDiscreteScheduler.from_config(pipe.scheduler.config)
         pipe = pipe.to("cpu")
         
@@ -54,7 +58,6 @@ def load_model(model_name="dreamshaper-8"):
         return pipe
     except Exception as e:
         print(f"❌ Model loading failed: {e}")
-        # Fallback to SD 1.5
         return StableDiffusionPipeline.from_pretrained(
             "runwayml/stable-diffusion-v1-5", 
             torch_dtype=torch.float32,
@@ -93,25 +96,20 @@ def enhance_prompt(prompt, style="childrens_book"):
         ]
     }
     
-    # Choose style template
     templates = style_templates.get(style, style_templates["childrens_book"])
-    style_prompt = templates[0]  # Use the first template
+    style_prompt = templates[0]
     
-    # Enhanced prompt construction
     enhanced = f"{style_prompt}, {prompt}"
     
-    # Add quality boosters
     quality_boosters = [
         "intricate details", "beautiful composition", "perfect lighting",
         "professional artwork", "award winning", "trending on artstation"
     ]
     
-    # Add 2-3 random quality boosters
     import random
     boosters = random.sample(quality_boosters, 2)
     enhanced += ", " + ", ".join(boosters)
     
-    # Negative prompt to avoid bad quality
     negative_prompt = (
         "blurry, low quality, low resolution, ugly, deformed, poorly drawn, "
         "bad anatomy, wrong anatomy, extra limb, missing limb, floating limbs, "
@@ -121,23 +119,37 @@ def enhance_prompt(prompt, style="childrens_book"):
     
     return enhanced, negative_prompt
 
-def save_to_oci_via_api(image, prompt):
-    """Save image using the EXTERNAL OCI API endpoint"""
+def get_next_sequence_number(story_title: str) -> int:
+    """Get the next sequence number for a story"""
+    # Clean the story title for use in filenames
+    clean_title = re.sub(r'[^a-zA-Z0-9_\-]', '', story_title.strip().replace(' ', '_'))
+    
+    if clean_title not in story_registry:
+        story_registry[clean_title] = 0
+    
+    story_registry[clean_title] += 1
+    return story_registry[clean_title], clean_title
+
+def save_to_oci_via_api(image, prompt, story_title, sequence_number):
+    """Save image with organized storybook structure"""
     try:
         # Convert image to bytes
         img_bytes = io.BytesIO()
         image.save(img_bytes, format='PNG')
         img_data = img_bytes.getvalue()
         
-        # Create filename
-        timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
-        safe_prompt = "".join(c for c in prompt[:30] if c.isalnum() or c in (' ', '-', '_')).rstrip()
-        filename = f"hq_story_{timestamp}_{safe_prompt}.png"
+        # Get sequence number and clean title
+        seq_num, clean_title = get_next_sequence_number(story_title)
+        if sequence_number is not None:
+            seq_num = sequence_number  # Use provided sequence number if available
+        
+        # Create organized filename with sequence
+        filename = f"page_{seq_num:03d}_{clean_title}.png"
         
         # Use the EXTERNAL API URL
         api_url = f"{OCI_API_BASE_URL}/api/upload"
         
-        print(f"🌐 Calling EXTERNAL OCI API: {api_url}")
+        print(f"📁 Saving to: stories/{clean_title}/{filename}")
         
         # Prepare form data for API request
         files = {
@@ -145,162 +157,204 @@ def save_to_oci_via_api(image, prompt):
         }
         
         data = {
-            'project_id': 'storybook-generator',
-            'subfolder': 'high-quality-books'
+            'project_id': 'storybook-library',
+            'subfolder': f'stories/{clean_title}'
         }
         
-        # Make the API request with timeout
+        # Make the API request
         response = requests.post(api_url, files=files, data=data, timeout=30)
         
-        print(f"📨 API Response: {response.status_code}")
-        
         if response.status_code == 200:
             result = response.json()
             if result['status'] == 'success':
-                return f"✅ {result['message']}"
+                return f"✅ Saved: {filename} | Story: {clean_title}"
             else:
                 return f"❌ API Error: {result.get('message', 'Unknown error')}"
         else:
-            return f"❌ HTTP Error: {response.status_code} - {response.text}"
+            return f"❌ HTTP Error: {response.status_code}"
             
     except Exception as e:
-        return f"❌ API upload failed: {str(e)}"
+        return f"❌ Upload failed: {str(e)}"
 
-def generate_high_quality_image(prompt, model_choice="dreamshaper-8", style="childrens_book"):
-    """Generate HIGH-QUALITY image from text prompt"""
+def generate_storybook_page(prompt, story_title, sequence_number=None, model_choice="dreamshaper-8", style="childrens_book"):
+    """Generate a storybook page with organized naming"""
     try:
         if not prompt or not prompt.strip():
             return None, "❌ Please enter a scene description"
         
+        if not story_title or not story_title.strip():
+            return None, "❌ Please enter a story title"
+        
         # Reload model if different choice
         global pipe
-        if model_choice != "dreamshaper-8":  # Default already loaded
+        if model_choice != "dreamshaper-8":
             pipe = load_model(model_choice)
         
-        # PROFESSIONAL PROMPT ENHANCEMENT
+        # Professional prompt enhancement
         enhanced_prompt, negative_prompt = enhance_prompt(prompt, style)
         
-        print(f"🎨 Generating HQ image for prompt: {enhanced_prompt}")
-        print(f"🚫 Negative prompt: {negative_prompt[:100]}...")
+        print(f"📖 Generating page for: {story_title}")
+        print(f"🎨 Prompt: {enhanced_prompt}")
         
-        # Generate HIGH-QUALITY image with better settings
+        # Generate high-quality image
         image = pipe(
             prompt=enhanced_prompt,
             negative_prompt=negative_prompt,
-            num_inference_steps=30,  # More steps = better quality
-            guidance_scale=8.5,      # Higher guidance = better prompt following
-            width=768,               # Higher resolution
+            num_inference_steps=30,
+            guidance_scale=8.5,
+            width=768,
             height=768,
-            generator=torch.Generator(device="cpu").manual_seed(int(time.time()))  # Random seed
+            generator=torch.Generator(device="cpu").manual_seed(int(time.time()))
         ).images[0]
         
-        print("✅ HQ Image generated successfully!")
+        print("✅ Page generated successfully!")
         
-        # Save to OCI
-        print("💾 Saving HQ image to OCI storage...")
-        save_status = save_to_oci_via_api(image, prompt)
-        print(f"Save status: {save_status}")
+        # Save with organized structure
+        save_status = save_to_oci_via_api(image, prompt, story_title, sequence_number)
+        print(f"💾 {save_status}")
         
         return image, save_status
         
     except Exception as e:
-        error_msg = f"❌ HQ Generation failed: {str(e)}"
+        error_msg = f"❌ Generation failed: {str(e)}"
         print(error_msg)
-        import traceback
-        print(f"Traceback: {traceback.format_exc()}")
         return None, error_msg
 
-# Create the Gradio interface with QUALITY OPTIONS
-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")
+def batch_generate_storybook(story_title, scenes_text, model_choice="dreamshaper-8", style="childrens_book"):
+    """Generate multiple pages for a complete storybook"""
+    if not story_title or not scenes_text:
+        return [], "❌ Please provide story title and scenes"
+    
+    scenes = [scene.strip() for scene in scenes_text.split('\n') if scene.strip()]
+    results = []
+    status_messages = []
+    
+    for i, scene in enumerate(scenes, 1):
+        print(f"📖 Generating page {i}/{len(scenes)} for: {story_title}")
+        image, status = generate_storybook_page(
+            scene, story_title, i, model_choice, style
+        )
+        
+        if image:
+            results.append((f"Page {i}: {scene}", image))
+        status_messages.append(f"Page {i}: {status}")
+    
+    return results, "\n".join(status_messages)
+
+# Create the enhanced Gradio interface
+with gr.Blocks(title="Professional Storybook Generator", theme="soft") as demo:
+    gr.Markdown("# 📚 Professional Storybook Generator")
+    gr.Markdown("Create organized storybooks with sequential page numbering")
     
     with gr.Row():
         with gr.Column(scale=1):
+            gr.Markdown("### 📖 Story Information")
+            
+            story_title = gr.Textbox(
+                label="Story Title",
+                placeholder="Enter your story title...\nExample: The Dragon's Reading Adventure",
+                lines=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"
+                value="dreamshaper-8"
             )
             
             style_choice = gr.Dropdown(
                 label="Art Style",
                 choices=["childrens_book", "realistic", "fantasy", "anime"],
-                value="childrens_book",
-                info="Select the artistic style"
+                value="childrens_book"
             )
+        
+        with gr.Column(scale=2):
+            gr.Markdown("### 🎨 Single Page Generation")
             
             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
+                label="Page Description",
+                placeholder="Describe this page's scene...\nExample: The dragon discovers a magical library in the forest",
+                lines=2
+            )
+            
+            generate_btn = gr.Button("✨ Generate Single Page", variant="primary")
+            image_output = gr.Image(label="Generated Page", height=400, show_download_button=True)
+            status_output = gr.Textbox(label="Status", interactive=False, lines=2)
+    
+    with gr.Row():
+        gr.Markdown("### 📚 Complete Storybook Generation")
+    
+    with gr.Row():
+        with gr.Column():
+            scenes_input = gr.Textbox(
+                label="Story Scenes (One per line = One per page)",
+                placeholder="Enter each page's scene on a separate line...\nExample:\nThe dragon finds a mysterious book\nHe learns to read with owl friend\nThey discover hidden treasure map\nFriends celebrate with magical feast",
+                lines=6
             )
             
-            generate_btn = gr.Button("✨ Generate Premium Image", variant="primary")
+            batch_btn = gr.Button("📖 Generate Complete Storybook", variant="primary")
         
-        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)
+        with gr.Column():
+            batch_status = gr.Textbox(label="Generation Status", interactive=False, lines=8)
+            batch_gallery = gr.Gallery(label="Storybook Pages", columns=2, height=600)
     
-    # Examples section
-    with gr.Accordion("💡 Prompt Examples & Tips", open=False):
+    with gr.Accordion("📁 Folder Structure Preview", open=True):
         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"
+        **Your story will be organized as:**
+        ```
+        storybook-library/
+        └── stories/
+            └── The_Dragons_Reading_Adventure/
+                ├── page_001_The_Dragons_Reading_Adventure.png
+                ├── page_002_The_Dragons_Reading_Adventure.png
+                ├── page_003_The_Dragons_Reading_Adventure.png
+                └── page_004_The_Dragons_Reading_Adventure.png
+        ```
         
-        ## 🚫 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"
+        **Perfect for:** 
+        - ✅ Easy PDF compilation later
+        - ✅ Automatic sequencing
+        - ✅ Organized by story title
+        - ✅ Ready for n8n automation
         """)
     
-    # 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"""
-        return f"""**System Status:**
-- Model: {model_choice.value}
-- Style: {style_choice.value}
-- OCI API: {OCI_API_BASE_URL}
-- Ready for premium image generation!"""
+    with gr.Accordion("💡 API Usage for n8n", open=False):
+        gr.Markdown("""
+        **For n8n automation, use this JSON format:**
+        ```json
+        {
+          "story_title": "The Dragon's Reading Adventure",
+          "scenes": [
+            "The dragon finds a mysterious old book in the forest",
+            "He meets an owl who teaches him to read",
+            "They discover a hidden treasure map in the book",
+            "All animal friends celebrate with a magical feast"
+          ],
+          "model_choice": "dreamshaper-8",
+          "style": "childrens_book"
+        }
+        ```
+        """)
 
     # Connect buttons to functions
     generate_btn.click(
-        fn=generate_high_quality_image,
-        inputs=[prompt_input, model_choice, style_choice],
+        fn=generate_storybook_page,
+        inputs=[prompt_input, story_title, gr.Number(visible=False), model_choice, style_choice],
         outputs=[image_output, status_output]
     )
     
-    debug_btn.click(
-        fn=check_system_status,
-        inputs=None,
-        outputs=debug_output
+    batch_btn.click(
+        fn=batch_generate_storybook,
+        inputs=[story_title, scenes_input, model_choice, style_choice],
+        outputs=[batch_gallery, batch_status]
     )
 
-# For Hugging Face Spaces deployment
 def get_app():
     return demo
 
 if __name__ == "__main__":
-    print("🚀 Starting Premium Children's Book Illustrator...")
-    print("🎨 Loaded with HIGH-QUALITY models:")
-    for model in MODEL_CHOICES:
-        print(f"   - {model}")
+    print("🚀 Starting Professional Storybook Generator...")
+    print("📁 Feature: Organized story sequencing enabled")
     demo.launch(server_name="0.0.0.0", server_port=7860)