Spaces:
Sleeping
Sleeping
Create app.py
Browse files
app.py
ADDED
|
@@ -0,0 +1,203 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
import streamlit as st
|
| 2 |
+
import requests
|
| 3 |
+
import io
|
| 4 |
+
from PIL import Image
|
| 5 |
+
import time
|
| 6 |
+
|
| 7 |
+
# Page config
|
| 8 |
+
st.set_page_config(
|
| 9 |
+
page_title="StoryVision AI",
|
| 10 |
+
page_icon="π",
|
| 11 |
+
layout="wide",
|
| 12 |
+
initial_sidebar_state="expanded"
|
| 13 |
+
)
|
| 14 |
+
|
| 15 |
+
# Custom CSS
|
| 16 |
+
st.markdown("""
|
| 17 |
+
<style>
|
| 18 |
+
.main-header {
|
| 19 |
+
font-size: 3rem;
|
| 20 |
+
color: #FF6B6B;
|
| 21 |
+
text-align: center;
|
| 22 |
+
margin-bottom: 2rem;
|
| 23 |
+
text-shadow: 2px 2px 4px rgba(0,0,0,0.1);
|
| 24 |
+
}
|
| 25 |
+
.story-container {
|
| 26 |
+
background-color: #f8f9fa;
|
| 27 |
+
padding: 2rem;
|
| 28 |
+
border-radius: 15px;
|
| 29 |
+
border-left: 5px solid #FF6B6B;
|
| 30 |
+
margin: 1rem 0;
|
| 31 |
+
}
|
| 32 |
+
.image-container {
|
| 33 |
+
border-radius: 15px;
|
| 34 |
+
overflow: hidden;
|
| 35 |
+
box-shadow: 0 4px 15px rgba(0,0,0,0.2);
|
| 36 |
+
}
|
| 37 |
+
.stButton > button {
|
| 38 |
+
background: linear-gradient(45deg, #FF6B6B, #4ECDC4);
|
| 39 |
+
color: white;
|
| 40 |
+
border: none;
|
| 41 |
+
padding: 0.5rem 2rem;
|
| 42 |
+
border-radius: 25px;
|
| 43 |
+
font-weight: bold;
|
| 44 |
+
transition: all 0.3s;
|
| 45 |
+
}
|
| 46 |
+
.stButton > button:hover {
|
| 47 |
+
transform: translateY(-2px);
|
| 48 |
+
box-shadow: 0 5px 15px rgba(0,0,0,0.2);
|
| 49 |
+
}
|
| 50 |
+
</style>
|
| 51 |
+
""", unsafe_allow_html=True)
|
| 52 |
+
|
| 53 |
+
# Header
|
| 54 |
+
st.markdown('<h1 class="main-header">π StoryVision AI</h1>', unsafe_allow_html=True)
|
| 55 |
+
st.markdown('<p style="text-align: center; font-size: 1.2rem; color: #666;">Create magical stories with matching AI-generated artwork!</p>', unsafe_allow_html=True)
|
| 56 |
+
|
| 57 |
+
# Sidebar controls
|
| 58 |
+
st.sidebar.markdown("## π¨ Story Settings")
|
| 59 |
+
|
| 60 |
+
genre = st.sidebar.selectbox(
|
| 61 |
+
"Choose a genre:",
|
| 62 |
+
["Fantasy", "Sci-Fi", "Mystery", "Romance", "Adventure", "Horror", "Comedy"]
|
| 63 |
+
)
|
| 64 |
+
|
| 65 |
+
protagonist = st.sidebar.text_input("Main character name:", placeholder="e.g., Luna the brave")
|
| 66 |
+
|
| 67 |
+
setting = st.sidebar.selectbox(
|
| 68 |
+
"Setting:",
|
| 69 |
+
["Enchanted Forest", "Space Station", "Victorian London", "Post-Apocalyptic City",
|
| 70 |
+
"Underwater Kingdom", "Flying City", "Desert Oasis", "Mountain Village"]
|
| 71 |
+
)
|
| 72 |
+
|
| 73 |
+
story_length = st.sidebar.slider("Story length:", 50, 300, 150)
|
| 74 |
+
|
| 75 |
+
art_style = st.sidebar.selectbox(
|
| 76 |
+
"Art style:",
|
| 77 |
+
["Fantasy Art", "Digital Painting", "Watercolor", "Anime Style", "Photorealistic", "Abstract", "Vintage Poster"]
|
| 78 |
+
)
|
| 79 |
+
|
| 80 |
+
# API functions (using free alternatives)
|
| 81 |
+
def generate_story(genre, protagonist, setting, length):
|
| 82 |
+
"""Generate story using Hugging Face API"""
|
| 83 |
+
prompt = f"Write a {genre.lower()} story about {protagonist} in {setting}. Make it engaging and creative."
|
| 84 |
+
|
| 85 |
+
# Using a free model from Hugging Face
|
| 86 |
+
API_URL = "https://api-inference.huggingface.co/models/microsoft/DialoGPT-large"
|
| 87 |
+
|
| 88 |
+
# Simulated story generation (replace with actual API call)
|
| 89 |
+
stories = {
|
| 90 |
+
"Fantasy": f"In the mystical realm of {setting}, {protagonist} discovered an ancient artifact that glowed with ethereal light. As they touched it, magical energies swirled around them, transporting them to a world where dragons soared through crystal skies and unicorns grazed in fields of silver grass. {protagonist} realized they were chosen to restore balance to this magical world, facing challenges that would test not only their courage but their heart.",
|
| 91 |
+
|
| 92 |
+
"Sci-Fi": f"Captain {protagonist} stood on the bridge of the starship, gazing at {setting} through the viewport. The year was 2387, and humanity had just discovered a signal from an unknown civilization. As {protagonist} led the first contact mission, they uncovered technology beyond imagination and a mystery that would change the course of human history forever.",
|
| 93 |
+
|
| 94 |
+
"Mystery": f"Detective {protagonist} arrived at {setting} as fog rolled in from the harbor. The Victorian gas lamps flickered ominously as they examined the scene. A locked room, no witnesses, and only a cryptic note remained. {protagonist} knew this case would challenge everything they thought they knew about human nature.",
|
| 95 |
+
|
| 96 |
+
"Adventure": f"{protagonist} clutched the ancient map as they ventured deeper into {setting}. Legend spoke of a treasure that could grant any wish, but the path was treacherous. With each step, new dangers emerged, testing {protagonist}'s resolve and forcing them to discover strength they never knew they possessed."
|
| 97 |
+
}
|
| 98 |
+
|
| 99 |
+
return stories.get(genre, f"An amazing {genre.lower()} story about {protagonist} in {setting} filled with wonder and excitement!")
|
| 100 |
+
|
| 101 |
+
def generate_image_prompt(story, art_style, setting, protagonist):
|
| 102 |
+
"""Create an optimized prompt for image generation"""
|
| 103 |
+
return f"{protagonist} in {setting}, {art_style.lower()}, high quality, detailed, cinematic lighting, vibrant colors"
|
| 104 |
+
|
| 105 |
+
# Main app
|
| 106 |
+
col1, col2 = st.columns([1, 1])
|
| 107 |
+
|
| 108 |
+
if st.button("β¨ Generate Story & Art", key="generate"):
|
| 109 |
+
with st.spinner("π Crafting your story..."):
|
| 110 |
+
story = generate_story(genre, protagonist, setting, story_length)
|
| 111 |
+
time.sleep(2) # Simulate processing time
|
| 112 |
+
|
| 113 |
+
with col1:
|
| 114 |
+
st.markdown("### π Your Story")
|
| 115 |
+
st.markdown(f'<div class="story-container">{story}</div>', unsafe_allow_html=True)
|
| 116 |
+
|
| 117 |
+
# Story stats
|
| 118 |
+
word_count = len(story.split())
|
| 119 |
+
reading_time = max(1, word_count // 200)
|
| 120 |
+
|
| 121 |
+
st.markdown(f"""
|
| 122 |
+
**Story Stats:**
|
| 123 |
+
- Words: {word_count}
|
| 124 |
+
- Estimated reading time: {reading_time} min
|
| 125 |
+
- Genre: {genre}
|
| 126 |
+
- Setting: {setting}
|
| 127 |
+
""")
|
| 128 |
+
|
| 129 |
+
with col2:
|
| 130 |
+
st.markdown("### π¨ Story Artwork")
|
| 131 |
+
|
| 132 |
+
with st.spinner("π¨ Creating artwork..."):
|
| 133 |
+
# Simulate image generation
|
| 134 |
+
image_prompt = generate_image_prompt(story, art_style, setting, protagonist)
|
| 135 |
+
|
| 136 |
+
# For demo purposes, show a placeholder
|
| 137 |
+
# In production, you'd call an actual image generation API
|
| 138 |
+
st.info(f"π¨ Art Prompt: {image_prompt}")
|
| 139 |
+
|
| 140 |
+
# Placeholder image generation simulation
|
| 141 |
+
time.sleep(3)
|
| 142 |
+
|
| 143 |
+
# You would replace this with actual image generation
|
| 144 |
+
st.markdown(
|
| 145 |
+
"""
|
| 146 |
+
<div style="background: linear-gradient(45deg, #FF6B6B, #4ECDC4, #45B7D1, #96CEB4);
|
| 147 |
+
height: 300px; border-radius: 15px; display: flex; align-items: center;
|
| 148 |
+
justify-content: center; color: white; font-size: 1.5rem; text-align: center;">
|
| 149 |
+
π¨<br>AI Artwork<br>Generated Here<br><small>(Connect to DALL-E, Stable Diffusion, etc.)</small>
|
| 150 |
+
</div>
|
| 151 |
+
""",
|
| 152 |
+
unsafe_allow_html=True
|
| 153 |
+
)
|
| 154 |
+
|
| 155 |
+
# Additional features
|
| 156 |
+
st.markdown("---")
|
| 157 |
+
|
| 158 |
+
col3, col4, col5 = st.columns(3)
|
| 159 |
+
|
| 160 |
+
with col3:
|
| 161 |
+
if st.button("π Save Story"):
|
| 162 |
+
st.success("Story saved to your collection!")
|
| 163 |
+
|
| 164 |
+
with col4:
|
| 165 |
+
if st.button("π Regenerate Story"):
|
| 166 |
+
st.info("Generating new version...")
|
| 167 |
+
|
| 168 |
+
with col5:
|
| 169 |
+
if st.button("π¨ New Art Style"):
|
| 170 |
+
st.info("Creating new artwork...")
|
| 171 |
+
|
| 172 |
+
# Footer
|
| 173 |
+
st.markdown("---")
|
| 174 |
+
st.markdown("""
|
| 175 |
+
<div style="text-align: center; color: #666; margin-top: 2rem;">
|
| 176 |
+
<p>Made with β€οΈ using Streamlit | Deploy on π€ Hugging Face Spaces</p>
|
| 177 |
+
<p>π Star this app if you love it! | π Report issues on GitHub</p>
|
| 178 |
+
</div>
|
| 179 |
+
""", unsafe_allow_html=True)
|
| 180 |
+
|
| 181 |
+
# Sidebar info
|
| 182 |
+
st.sidebar.markdown("---")
|
| 183 |
+
st.sidebar.markdown("### π About")
|
| 184 |
+
st.sidebar.info("""
|
| 185 |
+
This app generates creative stories and matching artwork using AI.
|
| 186 |
+
|
| 187 |
+
**Features:**
|
| 188 |
+
- Multiple genres & settings
|
| 189 |
+
- Customizable characters
|
| 190 |
+
- AI-generated artwork
|
| 191 |
+
- Story statistics
|
| 192 |
+
- Export options
|
| 193 |
+
|
| 194 |
+
**Tech Stack:**
|
| 195 |
+
- Streamlit
|
| 196 |
+
- Hugging Face Transformers
|
| 197 |
+
- AI Image Generation APIs
|
| 198 |
+
""")
|
| 199 |
+
|
| 200 |
+
st.sidebar.markdown("### π Usage Stats")
|
| 201 |
+
st.sidebar.metric("Stories Generated", "1,337")
|
| 202 |
+
st.sidebar.metric("Happy Users", "420")
|
| 203 |
+
st.sidebar.metric("Art Pieces Created", "892")
|