Update app.py

app.py

@@ -293,10 +293,18 @@ def get_ai_suggestion(field_name, existing_values=None):
 # Story generation function
 def generate_story(story_details):
     try:
-        system_prompt = system_dislecxic(story_details.get('target_age_group', '7-9'), 150)
+        word_limits = {
+            "1-3": 50,
+            "4-6": 100,
+            "7-9": 150,
+            "10-12": 200
+        }
+        word_limit = word_limits[story_details.get('target_age_group', '7-9')]
+        
+        system_prompt = system_dislecxic(story_details.get('target_age_group', '7-9'), word_limit)
         user_prompt = user_dislecxic(
             story_details.get('target_age_group', '7-9'),
-            150,
+            word_limit,
             story_details.get('story_style', ''),
             story_details.get('theme', ''),
             story_details.get('language', 'English'),
@@ -347,14 +355,24 @@ def generate_images(image_prompts, art_style):
     try:
         styled_prompts = []
         style_descriptions = {
-            "Abstract Art": "Digital art in Abstract style with vibrant shapes and patterns, ",
-            "Art Deco": "Digital art in Art Deco style with symmetrical designs and bold geometric forms, ",
-            "Art Nouveau": "Digital illustration in Art Nouveau style with intricate floral motifs and curvilinear designs, ",
-            "Bauhaus": "Digital art inspired by Bauhaus movement with geometric shapes and functional form, ",
-            "Celtic Art": "Digital art in Celtic style with intricate knotwork and spiral designs, ",
-            "Chinese Brush Painting": "Digital art in Chinese Brush Painting style with fluid brush strokes, ",
-            "Concept Art": "Detailed concept art illustration with rich details and atmosphere, ",
-            "Cyber Folk": "Digital art blending traditional folk patterns with futuristic cyber aesthetics, "
+            "Abstract Art": "Digital art in Abstract style with vibrant shapes and patterns, dynamic composition, non-representational elements, ",
+            "Art Deco": "Digital art in Art Deco style with symmetrical designs, bold geometric forms, radiant colors, 1920s aesthetic, ",
+            "Art Nouveau": "Digital illustration in Art Nouveau style with intricate floral motifs, curvilinear designs, flowing organic patterns, ",
+            "Bauhaus": "Digital art inspired by Bauhaus movement with geometric shapes, functional form, modernist approach, ",
+            "Celtic Art": "Digital art in Celtic style with intricate knotwork, spiral designs, medieval influence, ",
+            "Chinese Brush Painting": "Digital art in Chinese Brush Painting style with fluid brush strokes, serene composition, nature-inspired, ",
+            "Concept Art": "Detailed concept art illustration with rich details, atmospheric lighting, cinematic composition, ",
+            "Cyber Folk": "Digital art blending traditional folk patterns with futuristic cyber aesthetics, neon accents, ",
+            "Fairy Tale": "Classic storybook illustration with whimsical elements, soft colors, enchanted atmosphere, ",
+            "Fantasy": "Magical and imaginative digital art with ethereal lighting, fantastical elements, ",
+            "Pixar": "3D animated style with emotional depth, vibrant colors, expressive characters, cinematic quality, ",
+            "Watercolor": "Digital watercolor style with soft edges, flowing colors, artistic textures, delicate details, ",
+            "Abstract Geometry": "Digital art with clean geometric shapes, interlocking triangles and circles, modern minimal style, ",
+            "Bokeh Art": "Digital art with soft focus effects, luminous spots, dreamy atmosphere, ethereal lighting, ",
+            "Brutalism": "Digital art with raw textures, bold shapes, monolithic structures, stark contrasts, ",
+            "Byzantine": "Digital art with rich gold tones, mosaic-like textures, religious icon style, ornate details, ",
+            "Charcoal": "Digital art emulating charcoal drawing with bold strokes, rich textures, dramatic shadows, ",
+            "Chiptune": "Digital art in retro pixel art style, 8-bit aesthetic, vibrant colors, nostalgic gaming feel, "
         }
         
         style_prefix = style_descriptions.get(art_style, "Digital illustration in a ")
@@ -378,7 +396,7 @@ def generate_images(image_prompts, art_style):
         return None
 
 # Main UI
-st.title("✨ StoryMii - AI Story Generator")
+st.title("✨ StoryMii")
 
 # Sidebar navigation
 page = st.sidebar.radio("Navigation", ["Create Story", "Story Library"])
@@ -437,22 +455,52 @@ if page == "Create Story":
         
         story_details["target_age_group"] = st.select_slider(
             "Target Age Group",
-            options=["5-7", "7-9", "9-11"],
-            value="7-9"
+            options=["1-3", "4-6", "7-9", "10-12"],
+            value="7-9",
+            help="Select the target age group for your story"
         )
-    
-        with col2:
-            story_details["main_character"] = st.text_input(
-                "Main Character",
-                help="Who is your story's protagonist?",
-                value=st.session_state.story_details.get("main_character", "")
-            )
-            suggest_character = st.form_submit_button("🦸 Suggest Character")
-            if suggest_character:
-                suggestion = get_ai_suggestion("main character")
-                if suggestion:
-                    st.session_state.story_details["main_character"] = suggestion
-                    story_details["main_character"] = suggestion
+
+        # Word limit based on age group
+        word_limits = {
+            "1-3": 50,   # Simpler, shorter stories for toddlers
+            "4-6": 100,  # Slightly longer for preschoolers
+            "7-9": 150,  # More complex for early readers
+            "10-12": 200 # Longer stories for confident readers
+        }
+        word_limit = word_limits[story_details["target_age_group"]]
+
+        # Updated story tones
+        story_details["story_tone"] = st.selectbox(
+            "Story Tone",
+            ["Whimsical", "Educational", "Neutral", "Gentle", "Encouraging"],
+            help="Choose the overall tone of your story"
+        )
+
+        # Updated story styles
+        story_details["story_style"] = st.radio(
+            "Story Style",
+            [
+                "Picture Book", 
+                "Early Reader", 
+                "Educational Story",
+                "Bedtime Story",
+                "Interactive Story",
+                "Simple Chapter Story"
+            ],
+            help="Select the style that best fits your story"
+        )
+
+        story_details["main_character"] = st.text_input(
+            "Main Character",
+            help="Who is your story's protagonist?",
+            value=st.session_state.story_details.get("main_character", "")
+        )
+        suggest_character = st.form_submit_button("🦸 Suggest Character")
+        if suggest_character:
+            suggestion = get_ai_suggestion("main character")
+            if suggestion:
+                st.session_state.story_details["main_character"] = suggestion
+                story_details["main_character"] = suggestion
             
             story_details["main_character_attributes"] = st.text_area(
                 "Character Attributes",