Update app.py

app.py

@@ -13,20 +13,27 @@ st.subheader("Generate Custom FNaF Phone Calls")
 # User inputs
 night = st.selectbox("Select the Night", ["1", "2", "3", "4", "5", "6", "7"])
 tone = st.selectbox("Select the Tone of the Call", ["Friendly", "Warning", "Ominous", "Neutral"])
-length = st.slider("Select the Length of the Call (in words)", 50, 300, 150)
+length = st.slider("Select the Length of the Call (in words)", 50, 150, 100)
 include_phrase = st.text_input("Include a Specific Phrase (Optional)")
 
 # Generate phone call
 if st.button("Generate Phone Call"):
-    # Construct the prompt based on user input
-    prompt = f"Generate Custom FNaF Phone Call for night {night}. The tone should be {tone.lower()}."
+    # Construct a more detailed prompt
+    prompt = f"You are the Phone Guy from Five Nights at Freddy's. Your job is to give a phone call to the night guard on night {night}. The tone should be {tone.lower()}."
     if include_phrase:
-        prompt += f" Make sure to include the phrase: '{include_phrase}'."
-    
+        prompt += f" Include the following phrase in the call: '{include_phrase}'."
+    prompt += " Keep the call focused on advice for the night guard to survive the night, including any warnings about the animatronics. Make sure the content is spooky and coherent."
+
     # Generate the text
     result = generator(prompt, max_length=length, num_return_sequences=1)
     generated_text = result[0]['generated_text']
-    
+
+    # Post-processing: Remove unwanted parts
+    unwanted_phrases = ["Make sure", "please let", "If you have any questions", "To use a phone call"]
+    for phrase in unwanted_phrases:
+        if phrase in generated_text:
+            generated_text = generated_text.split(phrase)[0].strip()
+
     # Display the generated phone call
     st.text(generated_text)