Update app.py

app.py

@@ -29,12 +29,11 @@ temperature = st.slider("Adjust creativity of the story:", 0.7, 1.5, 1.0)
 if st.button('Generate Story Continuation'):
     with st.spinner('AI is writing the story...'):
         # Improved prompt construction with user inputs
-        prompt = f"Write a story in the style of {genre.lower()}. It is set in {setting} and revolves around {character}. Begin with: '{user_input}'"
+        prompt = f"Genre: {genre}. Main character: {character}. Setting: {setting}. Begin the story with: '{user_input}'"
         
         try:
-            generated_story = story_generator(prompt, max_length=max_length, temperature=temperature)
-            story_text = generated_story[0]['generated_text']
-            st.text_area("Here's the continuation of your story:", story_text, height=250)
+            generated_story = story_generator(prompt, max_length=max_length, temperature=temperature)[0]['generated_text']
+            st.text_area("Here's the continuation of your story:", generated_story, height=250)
         except Exception as e:
             st.error(f"Error generating story: {str(e)}")
 
@@ -42,4 +41,3 @@ if st.button('Generate Story Continuation'):
 feedback = st.text_area("Feedback on the generated story:", "")
 if st.button('Submit Feedback'):
     st.write("Thanks for your feedback!")
-