Update app.py

app.py

@@ -1,5 +1,5 @@
 import streamlit as st
-from transformers import pipeline
+from transformers import pipeline, set_seed
 
 # Create a text2text-generation pipeline with the "google/flan-t5-base" model
 text_generator = pipeline("text2text-generation", model="google/flan-t5-base")
@@ -14,16 +14,13 @@ ending = st.selectbox("Ending:", ["Happy", "Sad"])
 
 if st.button("Generate Story"):
     if title and character:
-        # 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
-        story = f"# {title}\n\nOnce upon a time, in the {era} era, there was a character named {character}. "
-        story += f"This is a {genre} story of {character}, a tale filled with {setting}, a story of {conflict}. "
-        story += f"In the end, a {resolution}."
-
+        # Prepare a structured prompt for the story
+        prompt = f"Write a {genre} story titled '{title}' set in the {era} era. The main character, {character}, faces a {genre} adventure. In the end, the story has a {ending} ending."
+        
+        # Generate the story
+        set_seed(42)  # Set a seed for reproducibility
+        story = text_generator(prompt, max_length=300, do_sample=True)[0]["generated_text"]
+        
         st.markdown(story)
     else:
         st.warning("Please provide a title and character name.")