Update app.py

app.py

@@ -9,14 +9,15 @@ openai.api_key = st.secrets["OPENAI_API_KEY"]
 
 st.title("Children's Story and Image Panel Generator")
 
-# User inputs for the story
+# User inputs for the story and main character
 age_group = st.selectbox("Age Group", ["3-5 years", "6-8 years", "9-12 years"])
 theme = st.text_input("Story Theme", placeholder="Enter a theme for the story (e.g., adventure, friendship)")
-details = st.text_area("Additional Details", placeholder="Any specific details to include in the story (e.g., a brave rabbit, a magical forest)")
+main_character = st.text_input("Main Character", placeholder="Describe the main character (e.g., a small brave rabbit)")
+details = st.text_area("Additional Details", placeholder="Any specific details to include in the story (e.g., a magical forest)")
 
 # Function to generate the story
-def generate_story(age_group, theme, details):
-    prompt = f"Write a concise children's story suitable for age group {age_group} about {theme}. Include details such as {details}. The story should be short enough to fit within 10 panels."
+def generate_story(age_group, theme, main_character, details):
+    prompt = f"Write a concise children's story suitable for age group {age_group} about {theme} featuring a main character who is {main_character}. Include details such as {details}. The story should be short enough to fit within 10 panels."
     response = openai.ChatCompletion.create(
         model="gpt-4",
         messages=[
@@ -28,10 +29,11 @@ def generate_story(age_group, theme, details):
     return response.choices[0].message['content']
 
 # Function to generate images with DALL-E 3
-def generate_image(panel_text):
+def generate_image(panel_text, main_character):
     try:
+        prompt = f"{panel_text}. Visual style: consistent with previous panels. Main character: {main_character}."
         response = openai.Image.create(
-            prompt=panel_text,
+            prompt=prompt,
             n=1,
             size="1024x1024",  # Adjust the size if needed
             model="dall-e-3"  # Specify DALL-E 3 model
@@ -45,13 +47,13 @@ def generate_image(panel_text):
         return None
 
 if st.button('Generate Story and Images'):
-    story = generate_story(age_group, theme, details)
+    story = generate_story(age_group, theme, main_character, details)
     story_parts = story.split('.')[:10]  # Use the first 10 sentences for panels
 
     for i, part in enumerate(story_parts):
         if part.strip():  # Ensure part is not empty
             st.subheader(f'Panel {i+1}')
             st.write(part.strip())
-            image = generate_image(part)
+            image = generate_image(part, main_character)
             if image:
                 st.image(image, caption=f'Image for Panel {i+1}')