Update app.py

app.py

@@ -10,17 +10,19 @@ 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"])
 genre = st.selectbox("Genre:", ["Action", "Novelistic", "Love"])
-ending = st.selectbox("Ending:", ["Happy", "Sad"])
+punchline = st.text_area("Punchline (Describe what the story is about):")
+story_length = st.slider("Story Length", min_value=100, max_value=2000, step=100)
+generate_button = st.button("Generate Story")
 
-if st.button("Generate Story"):
-    if title and character:
+if generate_button:
+    if title and character and punchline:
         # 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."
-        
+        prompt = f"Write a {genre} story titled '{title}' set in the {era} era. The main character, {character}, faces a {genre} adventure. The story is about '{punchline}'."
+
         # Generate the story
         set_seed(42)  # Set a seed for reproducibility
-        full_story = text_generator(prompt, max_length=1000, do_sample=True)[0]["generated_text"]
-        
+        full_story = text_generator(prompt, max_length=story_length, do_sample=True)[0]["generated_text"]
+
         st.markdown(full_story)
     else:
-        st.warning("Please provide a title and character name.")
+        st.warning("Please provide a title, character name, and punchline.")