Update app.py

app.py

@@ -1,12 +1,11 @@
 import streamlit as st
-from transformers import pipeline, set_seed
+from transformers import pipeline
 
-# Create a text generation pipeline with the "google/flan-t5-base" model
-text_generator = pipeline("text-generation", model="google/flan-t5-base")
+# Create a text2text-generation pipeline with the "google/flan-t5-base" model
+text_generator = pipeline("text2text-generation", model="google/flan-t5-base")
 
 st.title("Story Generator")
 
-# User inputs
 title = st.text_input("Title of the Story:")
 character = st.text_input("Character (Hero) Name:")
 era = st.selectbox("Era of the Story:", ["Medieval", "Modern", "Future"])
@@ -15,16 +14,16 @@ ending = st.selectbox("Ending:", ["Happy", "Sad"])
 
 if st.button("Generate Story"):
     if title and character:
-        # Prepare the prompt based on user inputs
-        prompt = f"In the {era} era, there was a character named {character}. This is the {genre} story of {character}, a {genre} story with a {ending} ending."
-        
-        # Generate the story
-        set_seed(42)  # Set a seed for reproducibility
-        story = text_generator(prompt, max_length=500, do_sample=True)[0]["generated_text"]
-        
-        # Set the title
-        full_story = f"# {title}\n\n{story}"
-        
+        # Generate story elements using the model
+        setting = text_generator(f"Create the setting for a story set in the {era} era.")
+        conflict = text_generator(f"Describe the main conflict faced by the character {character}.")
+        resolution = text_generator(f"Provide a resolution for the {ending} ending.")
+
+        # Combine the elements into a full story
+        full_story = f"# {title}\n\nOnce upon a time, in the {era} era, there was a character named {character}. "
+        full_story += f"This is a {genre} story of {character}, a tale filled with {setting}, a story of {conflict}. "
+        full_story += f"In the end, a {resolution}."
+
         st.markdown(full_story)
     else:
         st.warning("Please provide a title and character name.")