Update app.py

app.py

@@ -2,8 +2,9 @@ import streamlit as st
 import time
 import random
 import requests
-from PIL import Image
+from PIL import Image, ImageDraw, ImageFont
 import io
+import hashlib
 
 # Page config
 st.set_page_config(
@@ -40,12 +41,10 @@ st.markdown("""
 # Initialize session state
 if 'generated_image' not in st.session_state:
     st.session_state.generated_image = None
-if 'image_prompt' not in st.session_state:
-    st.session_state.image_prompt = ""
 
 # Header
 st.markdown('<h1 class="main-header">📚 StoryVision AI</h1>', unsafe_allow_html=True)
-st.markdown('<p style="text-align: center; font-size: 1.2rem; color: #666;">Create magical stories with REAL AI-generated artwork!</p>', unsafe_allow_html=True)
+st.markdown('<p style="text-align: center; font-size: 1.2rem; color: #666;">Create magical stories with AI-generated artwork!</p>', unsafe_allow_html=True)
 
 # Sidebar controls
 st.sidebar.markdown("## 🎨 Story Settings")
@@ -70,46 +69,124 @@ art_style = st.sidebar.selectbox(
     ["Fantasy Art", "Digital Painting", "Watercolor", "Anime Style", "Photorealistic", "Concept Art"]
 )
 
-# Image generation function
+# Working image generation using Pollinations AI (No API key needed!)
 @st.cache_data(show_spinner=False)
-def generate_image(prompt, style):
-    """Generate image using Hugging Face Stable Diffusion"""
-    
-    # Hugging Face API endpoint (free tier)
-    API_URL = "https://api-inference.huggingface.co/models/runwayml/stable-diffusion-v1-5"
-    
-    # Enhance prompt based on style
-    style_prompts = {
-        "Fantasy Art": "fantasy art, magical, ethereal, detailed, artstation, concept art",
-        "Digital Painting": "digital painting, detailed, artstation, concept art, smooth",
-        "Watercolor": "watercolor painting, soft colors, artistic, traditional art",
-        "Anime Style": "anime style, manga, detailed, vibrant colors, studio quality",
-        "Photorealistic": "photorealistic, detailed, high quality, professional photography",
-        "Concept Art": "concept art, detailed, artstation, matte painting, digital art"
-    }
-    
-    enhanced_prompt = f"{prompt}, {style_prompts.get(style, 'detailed artwork')}, high quality, masterpiece"
-    
+def generate_image_pollinations(prompt, width=512, height=512):
+    """Generate image using Pollinations.ai - completely free, no API key needed"""
     try:
-        # Make request to Hugging Face API
-        headers = {"Content-Type": "application/json"}
-        payload = {"inputs": enhanced_prompt}
+        # Clean and encode the prompt
+        clean_prompt = prompt.replace(" ", "%20").replace(",", "%2C")
+        
+        # Pollinations.ai API endpoint
+        url = f"https://image.pollinations.ai/prompt/{clean_prompt}?width={width}&height={height}&seed={random.randint(1, 10000)}"
         
-        response = requests.post(API_URL, headers=headers, json=payload, timeout=30)
+        response = requests.get(url, timeout=30)
         
         if response.status_code == 200:
-            # Convert response to image
             image = Image.open(io.BytesIO(response.content))
-            return image, enhanced_prompt
+            return image, True
         else:
-            st.warning(f"Image generation failed: {response.status_code}")
-            return None, enhanced_prompt
+            return None, False
             
     except Exception as e:
-        st.error(f"Error generating image: {str(e)}")
-        return None, enhanced_prompt
+        st.error(f"Error: {str(e)}")
+        return None, False
+
+# Alternative: Generate artistic image using code (always works)
+def generate_procedural_art(prompt, protagonist, setting, genre, style):
+    """Generate artistic image using Python - always works offline"""
+    
+    # Create a 512x512 image
+    width, height = 512, 512
+    image = Image.new('RGB', (width, height))
+    draw = ImageDraw.Draw(image)
+    
+    # Color schemes based on genre
+    color_schemes = {
+        "Fantasy": [(138, 43, 226), (75, 0, 130), (72, 61, 139), (147, 112, 219)],
+        "Sci-Fi": [(0, 191, 255), (30, 144, 255), (0, 100, 0), (50, 205, 50)],
+        "Mystery": [(25, 25, 112), (72, 61, 139), (105, 105, 105), (128, 128, 128)],
+        "Romance": [(255, 182, 193), (255, 105, 180), (219, 112, 147), (199, 21, 133)],
+        "Adventure": [(255, 140, 0), (255, 165, 0), (218, 165, 32), (184, 134, 11)],
+        "Horror": [(139, 0, 0), (128, 0, 0), (105, 105, 105), (47, 79, 79)],
+        "Comedy": [(255, 215, 0), (255, 165, 0), (255, 192, 203), (255, 20, 147)]
+    }
+    
+    colors = color_schemes.get(genre, color_schemes["Fantasy"])
+    
+    # Create gradient background
+    for y in range(height):
+        for x in range(width):
+            # Create complex gradient patterns
+            color_index = int((x + y + random.randint(-20, 20)) / 100) % len(colors)
+            base_color = colors[color_index]
+            
+            # Add some variation
+            r = max(0, min(255, base_color[0] + random.randint(-30, 30)))
+            g = max(0, min(255, base_color[1] + random.randint(-30, 30)))
+            b = max(0, min(255, base_color[2] + random.randint(-30, 30)))
+            
+            draw.point((x, y), (r, g, b))
+    
+    # Add geometric patterns based on genre
+    if genre == "Fantasy":
+        # Draw magical circles
+        for _ in range(5):
+            x = random.randint(50, width-50)
+            y = random.randint(50, height-50)
+            radius = random.randint(20, 80)
+            draw.ellipse([x-radius, y-radius, x+radius, y+radius], 
+                        outline=(255, 215, 0), width=3)
+    
+    elif genre == "Sci-Fi":
+        # Draw tech patterns
+        for _ in range(10):
+            x1 = random.randint(0, width)
+            y1 = random.randint(0, height)
+            x2 = random.randint(0, width)
+            y2 = random.randint(0, height)
+            draw.line([x1, y1, x2, y2], fill=(0, 255, 255), width=2)
+    
+    # Add title text
+    try:
+        font_size = 24
+        # Try to use a better font, fall back to default
+        try:
+            font = ImageFont.truetype("arial.ttf", font_size)
+        except:
+            font = ImageFont.load_default()
+        
+        # Add protagonist name
+        text_bbox = draw.textbbox((0, 0), protagonist, font=font)
+        text_width = text_bbox[2] - text_bbox[0]
+        text_x = (width - text_width) // 2
+        text_y = height - 60
+        
+        # Add text with outline
+        for dx in [-1, 0, 1]:
+            for dy in [-1, 0, 1]:
+                if dx != 0 or dy != 0:
+                    draw.text((text_x + dx, text_y + dy), protagonist, font=font, fill=(0, 0, 0))
+        draw.text((text_x, text_y), protagonist, font=font, fill=(255, 255, 255))
+        
+        # Add setting
+        setting_bbox = draw.textbbox((0, 0), setting, font=font)
+        setting_width = setting_bbox[2] - setting_bbox[0]
+        setting_x = (width - setting_width) // 2
+        setting_y = 30
+        
+        for dx in [-1, 0, 1]:
+            for dy in [-1, 0, 1]:
+                if dx != 0 or dy != 0:
+                    draw.text((setting_x + dx, setting_y + dy), setting, font=font, fill=(0, 0, 0))
+        draw.text((setting_x, setting_y), setting, font=font, fill=(255, 255, 255))
+        
+    except:
+        pass  # Skip text if font issues
+    
+    return image
 
-# Story templates (same as before)
+# Story templates (keeping the same as before)
 STORY_TEMPLATES = {
     "Fantasy": [
         "In the mystical realm of {setting}, {protagonist} discovered an ancient artifact that pulsed with magical energy. As whispers of forgotten spells filled the air, they realized their destiny was far greater than they ever imagined. Dragons soared overhead while mystical creatures watched from the shadows, all waiting to see if {protagonist} would embrace their newfound power or be consumed by it.",
@@ -163,17 +240,17 @@ def generate_story(genre, protagonist, setting):
 def create_image_prompt(protagonist, setting, genre, art_style):
     """Create optimized prompt for image generation"""
     prompts = {
-        "Fantasy": f"{protagonist} in {setting}, magical atmosphere, fantasy character",
-        "Sci-Fi": f"{protagonist} in {setting}, futuristic, sci-fi environment, technology",
-        "Mystery": f"{protagonist} in {setting}, mysterious atmosphere, detective scene",
-        "Romance": f"{protagonist} in {setting}, romantic atmosphere, beautiful lighting",
-        "Adventure": f"{protagonist} in {setting}, action scene, adventure",
-        "Horror": f"{setting}, dark atmosphere, horror scene, mysterious figure",
-        "Comedy": f"{protagonist} in {setting}, lighthearted scene, cheerful"
+        "Fantasy": f"fantasy art of {protagonist} in {setting}, magical, ethereal, detailed artwork",
+        "Sci-Fi": f"sci-fi art of {protagonist} in {setting}, futuristic, technology, space art",
+        "Mystery": f"mysterious scene with {protagonist} in {setting}, noir style, detective",
+        "Romance": f"romantic art of {protagonist} in {setting}, beautiful, soft lighting",
+        "Adventure": f"adventure art of {protagonist} in {setting}, action, dynamic scene",
+        "Horror": f"dark art of {setting}, horror atmosphere, mysterious figure, gothic",
+        "Comedy": f"cheerful art of {protagonist} in {setting}, bright colors, happy scene"
     }
     
     base_prompt = prompts.get(genre, f"{protagonist} in {setting}")
-    return base_prompt
+    return f"{base_prompt}, {art_style.lower()}, high quality, detailed"
 
 # Main app
 col1, col2 = st.columns([1, 1])
@@ -210,17 +287,21 @@ if st.button("✨ Generate Story & Art", type="primary"):
     with col2:
         st.markdown("### 🎨 AI Generated Artwork")
         
-        # Generate image prompt
         image_prompt = create_image_prompt(protagonist, setting, genre, art_style)
-        st.session_state.image_prompt = image_prompt
         
-        with st.spinner("🎨 Generating AI artwork... (this may take 10-30 seconds)"):
-            generated_image, full_prompt = generate_image(image_prompt, art_style)
+        with st.spinner("���� Generating AI artwork..."):
+            # Try Pollinations AI first (free, no API key)
+            generated_image, success = generate_image_pollinations(image_prompt)
+            
+            if not success or generated_image is None:
+                st.info("🎨 Using procedural art generation...")
+                generated_image = generate_procedural_art(image_prompt, protagonist, setting, genre, art_style)
+            
             st.session_state.generated_image = generated_image
         
         if generated_image:
             st.markdown('<div class="image-container">', unsafe_allow_html=True)
-            st.image(generated_image, caption=f"AI Generated: {protagonist} in {setting}", use_column_width=True)
+            st.image(generated_image, caption=f"Generated: {protagonist} in {setting}", use_column_width=True)
             st.markdown('</div>', unsafe_allow_html=True)
             
             # Image download
@@ -238,11 +319,7 @@ if st.button("✨ Generate Story & Art", type="primary"):
             with st.expander("🔍 Image Details"):
                 st.markdown(f"**Prompt:** {image_prompt}")
                 st.markdown(f"**Style:** {art_style}")
-                st.markdown(f"**Full Prompt:** {full_prompt}")
-        else:
-            # Fallback if image generation fails
-            st.error("🚫 Image generation temporarily unavailable. The Hugging Face API might be overloaded. Try again in a moment!")
-            st.info("💡 **Tip:** The free Hugging Face API can be slow or temporarily unavailable due to high demand.")
+                st.markdown(f"**Method:** {'Pollinations AI' if success else 'Procedural Art'}")
 
 # Show previous image if available
 elif st.session_state.generated_image:
@@ -260,50 +337,47 @@ with col3:
         st.rerun()
 
 with col4:
-    if st.button("🎨 Regenerate Image") and st.session_state.image_prompt:
-        with st.spinner("🎨 Creating new artwork..."):
-            generated_image, _ = generate_image(st.session_state.image_prompt, art_style)
-            if generated_image:
+    if st.button("🎨 New Artwork"):
+        if 'generated_image' in st.session_state:
+            # Regenerate with different seed
+            image_prompt = create_image_prompt(protagonist, setting, genre, art_style)
+            with st.spinner("🎨 Creating new artwork..."):
+                generated_image, success = generate_image_pollinations(image_prompt)
+                if not success:
+                    generated_image = generate_procedural_art(image_prompt, protagonist, setting, genre, art_style)
                 st.session_state.generated_image = generated_image
-                st.success("✅ New artwork generated!")
                 st.rerun()
 
 with col5:
     if st.button("🎲 Surprise Me!"):
-        st.session_state.random_generate = True
         st.rerun()
 
 # Sidebar info
 st.sidebar.markdown("---")
 st.sidebar.markdown("### 🚀 About StoryVision AI")
 st.sidebar.info("""
-🎯 **What's New:**
-- ✅ REAL AI image generation!
-- ✅ Powered by Stable Diffusion
+🎯 **Features:**
+- ✅ Real AI image generation
+- ✅ No API keys required
+- ✅ Always works offline backup
 - ✅ Download stories & images
-- ✅ Multiple art styles
 
 🛠️ **Tech Stack:**
 - Streamlit
-- Hugging Face API
-- Stable Diffusion
+- Pollinations.ai (free)
 - Python PIL
+- Procedural art generation
 
-⚡ **Note:** Image generation uses free APIs and may be slower during peak times.
+🎨 **Art Methods:**
+1. Pollinations AI (primary)
+2. Procedural art (backup)
 """)
 
-# Usage stats
-st.sidebar.markdown("### 📊 Live Stats")
-st.sidebar.metric("🖼️ Images Generated", f"{random.randint(800, 1200):,}")
-st.sidebar.metric("📚 Stories Created", f"{random.randint(1500, 2000):,}")
-st.sidebar.metric("👥 Active Users", f"{random.randint(50, 150)}")
-
 # Footer
 st.markdown("---")
 st.markdown("""
 <div style="text-align: center; color: #666; padding: 2rem;">
-    <p>🌟 <strong>StoryVision AI</strong> - Now with REAL AI artwork generation!</p>
-    <p>Made with ❤️ using Streamlit & Stable Diffusion | Perfect for 🤗 Hugging Face Spaces</p>
-    <p><small>⚡ Free tier API - Images may take 10-30 seconds to generate</small></p>
+    <p>🌟 <strong>StoryVision AI</strong> - Reliable AI artwork generation!</p>
+    <p>Made with ❤️ | Always works, no API keys needed</p>
 </div>
 """, unsafe_allow_html=True)