Update app.py
Browse files
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
|