Spaces:
Sleeping
Sleeping
| import streamlit as st | |
| import time | |
| import random | |
| import requests | |
| from PIL import Image, ImageDraw, ImageFont | |
| import io | |
| import hashlib | |
| # Page config | |
| st.set_page_config( | |
| page_title="StoryVision AI", | |
| page_icon="📚", | |
| layout="wide" | |
| ) | |
| # Custom CSS | |
| st.markdown(""" | |
| <style> | |
| .main-header { | |
| font-size: 3rem; | |
| color: #FF6B6B; | |
| text-align: center; | |
| margin-bottom: 2rem; | |
| } | |
| .story-container { | |
| background-color: #f8f9fa; | |
| padding: 2rem; | |
| border-radius: 15px; | |
| border-left: 5px solid #FF6B6B; | |
| margin: 1rem 0; | |
| } | |
| .image-container { | |
| border-radius: 15px; | |
| overflow: hidden; | |
| box-shadow: 0 4px 15px rgba(0,0,0,0.2); | |
| margin: 1rem 0; | |
| } | |
| </style> | |
| """, unsafe_allow_html=True) | |
| # Initialize session state | |
| if 'generated_image' not in st.session_state: | |
| st.session_state.generated_image = None | |
| # 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 AI-generated artwork!</p>', unsafe_allow_html=True) | |
| # Sidebar controls | |
| st.sidebar.markdown("## 🎨 Story Settings") | |
| genre = st.sidebar.selectbox( | |
| "Choose a genre:", | |
| ["Fantasy", "Sci-Fi", "Mystery", "Romance", "Adventure", "Horror", "Comedy"] | |
| ) | |
| protagonist = st.sidebar.text_input("Main character name:", placeholder="e.g., Luna the brave") | |
| if not protagonist: | |
| protagonist = "The Hero" | |
| setting = st.sidebar.selectbox( | |
| "Setting:", | |
| ["Enchanted Forest", "Space Station", "Victorian London", "Post-Apocalyptic City", | |
| "Underwater Kingdom", "Flying City", "Desert Oasis", "Mountain Village"] | |
| ) | |
| art_style = st.sidebar.selectbox( | |
| "Art style:", | |
| ["Fantasy Art", "Digital Painting", "Watercolor", "Anime Style", "Photorealistic", "Concept Art"] | |
| ) | |
| # Working image generation using Pollinations AI (No API key needed!) | |
| def generate_image_pollinations(prompt, width=512, height=512): | |
| """Generate image using Pollinations.ai - completely free, no API key needed""" | |
| try: | |
| # 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.get(url, timeout=30) | |
| if response.status_code == 200: | |
| image = Image.open(io.BytesIO(response.content)) | |
| return image, True | |
| else: | |
| return None, False | |
| except Exception as e: | |
| 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 (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.", | |
| "The {setting} had always been a place of wonder, but when {protagonist} stepped through the shimmering portal, everything changed. Magic flowed through their veins as they encountered beings of legend. With a prophecy to fulfill and dark forces gathering, {protagonist} must master ancient arts before the realm falls to eternal darkness." | |
| ], | |
| "Sci-Fi": [ | |
| "Captain {protagonist} gazed at the vast expanse of {setting} through the viewport. The year 2387 had brought humanity to the stars, but nothing could have prepared them for what they found. An alien signal, a mysterious technology, and a discovery that would rewrite the laws of physics. {protagonist} knew that first contact would change everything.", | |
| "In the depths of {setting}, {protagonist} uncovered technology that defied explanation. Quantum computers hummed with alien algorithms while holographic displays showed star maps of unknown galaxies. As reality itself began to shift, {protagonist} realized they were part of an experiment that spanned millennia." | |
| ], | |
| "Mystery": [ | |
| "Detective {protagonist} arrived at {setting} as fog rolled through the streets. The crime scene was unlike anything they'd encountered - a locked room, no witnesses, and only a cryptic symbol carved into the wall. As {protagonist} pieced together the clues, they discovered a conspiracy that reached the highest levels of power.", | |
| "The {setting} held secrets that {protagonist} was determined to uncover. Each clue led to more questions, each witness told a different story. In a web of lies and deception, {protagonist} raced against time to solve a mystery that threatened to destroy everything they held dear." | |
| ], | |
| "Adventure": [ | |
| "{protagonist} clutched the weathered map as they ventured into {setting}. Legend spoke of a treasure beyond imagination, but the path was fraught with danger. Ancient traps, rival treasure hunters, and the unforgiving environment tested every skill {protagonist} possessed. Victory would mean glory, but failure meant certain doom.", | |
| "The call of adventure echoed through {setting} as {protagonist} embarked on a journey that would test their limits. With only their wits and courage, they faced challenges that seemed impossible. Each step forward brought new dangers, but also the promise of discoveries that would change their life forever." | |
| ], | |
| "Romance": [ | |
| "In the enchanting {setting}, {protagonist} never expected to find love. But when fate intervened, bringing two hearts together against all odds, magic seemed to fill the air. Through misunderstandings and moments of pure joy, {protagonist} discovered that true love was worth fighting for, no matter the obstacles.", | |
| "The {setting} became the backdrop for an unexpected romance as {protagonist} met someone who challenged everything they believed about love. Through stolen glances and heartfelt conversations, a connection formed that transcended all boundaries." | |
| ], | |
| "Horror": [ | |
| "The {setting} was supposed to be abandoned, but {protagonist} could feel eyes watching from the shadows. Strange sounds echoed through empty corridors while an oppressive darkness seemed to consume all light. As reality began to blur with nightmare, {protagonist} realized some places are abandoned for very good reasons.", | |
| "Something was terribly wrong in {setting}. {protagonist} felt it the moment they arrived - a creeping dread that seemed to seep from the very walls. As night fell and the true horror revealed itself, {protagonist} discovered that some evils are better left undisturbed." | |
| ], | |
| "Comedy": [ | |
| "Everything that could go wrong did go wrong when {protagonist} found themselves in {setting}. From mistaken identities to slapstick mishaps, their simple plan had turned into a comedy of errors. With each attempt to fix things only making them worse, {protagonist} learned that sometimes you just have to laugh at life's absurdities.", | |
| "The {setting} was the perfect place for chaos, and {protagonist} seemed to be a magnet for it. Ridiculous situations, eccentric characters, and perfectly timed misunderstandings created a whirlwind of hilarity that {protagonist} would never forget." | |
| ] | |
| } | |
| def generate_story(genre, protagonist, setting): | |
| """Generate a story based on the selected parameters""" | |
| templates = STORY_TEMPLATES.get(genre, STORY_TEMPLATES["Fantasy"]) | |
| template = random.choice(templates) | |
| return template.format(protagonist=protagonist, setting=setting) | |
| def create_image_prompt(protagonist, setting, genre, art_style): | |
| """Create optimized prompt for image generation""" | |
| prompts = { | |
| "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 f"{base_prompt}, {art_style.lower()}, high quality, detailed" | |
| # Main app | |
| col1, col2 = st.columns([1, 1]) | |
| if st.button("✨ Generate Story & Art", type="primary"): | |
| with st.spinner("🎭 Crafting your story..."): | |
| story = generate_story(genre, protagonist, setting) | |
| time.sleep(1) | |
| with col1: | |
| st.markdown("### 📖 Your Story") | |
| st.markdown(f'<div class="story-container">{story}</div>', unsafe_allow_html=True) | |
| # Story stats | |
| word_count = len(story.split()) | |
| reading_time = max(1, word_count // 200) | |
| st.markdown(f""" | |
| **Story Stats:** | |
| - **Words:** {word_count} | |
| - **Reading time:** {reading_time} min | |
| - **Genre:** {genre} | |
| - **Setting:** {setting} | |
| """) | |
| # Download button | |
| st.download_button( | |
| label="📥 Download Story", | |
| data=story, | |
| file_name=f"{protagonist.replace(' ', '_')}_story.txt", | |
| mime="text/plain" | |
| ) | |
| with col2: | |
| st.markdown("### 🎨 AI Generated Artwork") | |
| image_prompt = create_image_prompt(protagonist, setting, genre, 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"Generated: {protagonist} in {setting}", use_column_width=True) | |
| st.markdown('</div>', unsafe_allow_html=True) | |
| # Image download | |
| img_buffer = io.BytesIO() | |
| generated_image.save(img_buffer, format="PNG") | |
| img_buffer.seek(0) | |
| st.download_button( | |
| label="🖼️ Download Image", | |
| data=img_buffer.getvalue(), | |
| file_name=f"{protagonist.replace(' ', '_')}_artwork.png", | |
| mime="image/png" | |
| ) | |
| with st.expander("🔍 Image Details"): | |
| st.markdown(f"**Prompt:** {image_prompt}") | |
| st.markdown(f"**Style:** {art_style}") | |
| st.markdown(f"**Method:** {'Pollinations AI' if success else 'Procedural Art'}") | |
| # Show previous image if available | |
| elif st.session_state.generated_image: | |
| col1, col2 = st.columns([1, 1]) | |
| with col2: | |
| st.markdown("### 🎨 Previous Generated Image") | |
| st.image(st.session_state.generated_image, caption="Previously Generated Artwork", use_column_width=True) | |
| # Action buttons | |
| st.markdown("---") | |
| col3, col4, col5 = st.columns(3) | |
| with col3: | |
| if st.button("🔄 New Story Only"): | |
| st.rerun() | |
| with col4: | |
| 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.rerun() | |
| with col5: | |
| if st.button("🎲 Surprise Me!"): | |
| st.rerun() | |
| # Sidebar info | |
| st.sidebar.markdown("---") | |
| st.sidebar.markdown("### 🚀 About StoryVision AI") | |
| st.sidebar.info(""" | |
| 🎯 **Features:** | |
| - ✅ Real AI image generation | |
| - ✅ No API keys required | |
| - ✅ Always works offline backup | |
| - ✅ Download stories & images | |
| 🛠️ **Tech Stack:** | |
| - Streamlit | |
| - Pollinations.ai (free) | |
| - Python PIL | |
| - Procedural art generation | |
| 🎨 **Art Methods:** | |
| 1. Pollinations AI (primary) | |
| 2. Procedural art (backup) | |
| """) | |
| # Footer | |
| st.markdown("---") | |
| st.markdown(""" | |
| <div style="text-align: center; color: #666; padding: 2rem;"> | |
| <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) |