adi-123 commited on
Commit
8ab2cd0
1 Parent(s): 5cb9f07

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -2
app.py CHANGED
@@ -68,7 +68,6 @@ def get_user_preferences() -> Dict[str, str]:
68
  preferences['tone'] = st.selectbox("Tone", ["Serious", "Light-hearted", "Humorous", "Dark"])
69
  preferences['theme'] = st.selectbox("Theme", ["Self-discovery", "Redemption", "Love", "Justice"])
70
  preferences['conflict'] = st.selectbox("Conflict Type", ["Person vs. Society", "Internal struggle", "Person vs. Nature", "Person vs. Person"])
71
- preferences['magic_tech'] = st.selectbox("Magic/Technology", ["Advanced technology", "Magic system", "Supernatural abilities", "Alien technology"])
72
  preferences['twist'] = st.selectbox("Mystery/Twist", ["Plot twist", "Hidden identity", "Unexpected ally/enemy", "Time paradox"])
73
  preferences['ending'] = st.selectbox("Ending", ["Happy", "Bittersweet", "Open-ended", "Tragic"])
74
 
@@ -108,7 +107,6 @@ def main():
108
  prompt = f"Based on the image description: '{scenario}', create a {preferences['genre']} story set in {preferences['setting']} in {preferences['continent']}. " \
109
  f"The story should have a {preferences['tone']} tone and explore the theme of {preferences['theme']}. " \
110
  f"The main conflict should be {preferences['conflict']}. " \
111
- f"Include {preferences['magic_tech']} as a key element. " \
112
  f"The story should have a {preferences['twist']} and end with a {preferences['ending']} ending."
113
 
114
  story = txt2story(prompt, top_k, top_p, temperature) # Generates a story based on the image text, LLM params, and user preferences
 
68
  preferences['tone'] = st.selectbox("Tone", ["Serious", "Light-hearted", "Humorous", "Dark"])
69
  preferences['theme'] = st.selectbox("Theme", ["Self-discovery", "Redemption", "Love", "Justice"])
70
  preferences['conflict'] = st.selectbox("Conflict Type", ["Person vs. Society", "Internal struggle", "Person vs. Nature", "Person vs. Person"])
 
71
  preferences['twist'] = st.selectbox("Mystery/Twist", ["Plot twist", "Hidden identity", "Unexpected ally/enemy", "Time paradox"])
72
  preferences['ending'] = st.selectbox("Ending", ["Happy", "Bittersweet", "Open-ended", "Tragic"])
73
 
 
107
  prompt = f"Based on the image description: '{scenario}', create a {preferences['genre']} story set in {preferences['setting']} in {preferences['continent']}. " \
108
  f"The story should have a {preferences['tone']} tone and explore the theme of {preferences['theme']}. " \
109
  f"The main conflict should be {preferences['conflict']}. " \
 
110
  f"The story should have a {preferences['twist']} and end with a {preferences['ending']} ending."
111
 
112
  story = txt2story(prompt, top_k, top_p, temperature) # Generates a story based on the image text, LLM params, and user preferences