import streamlit as st from story_gen import StoryGenerator import plotly.figure_factory as ff import plotly.express as px st.set_page_config(page_title='Storytelling ' + u'\U0001F5BC', page_icon=u'\U0001F5BC', layout="wide") gen = StoryGenerator() container_mode = st.sidebar.container() container_param = st.sidebar.container() container_button = st.sidebar.container() mode = container_mode.radio( "Select your mode", ('Create Statistics', 'Play Storytelling'), index=0) story_till_now = container_param.text_input( label='First Sentence', value='Hello, I\'m a language model,') num_generation = container_param.slider( label='Number of generation', min_value=1, max_value=100, value=3, step=1) length = container_param.slider(label='Length of the generated sentence', min_value=1, max_value=100, value=10, step=1) if mode == 'Create Statistics': container_mode.write('You selected statistics.') num_tests = container_param.slider( label='Number of tests', min_value=1, max_value=1000, value=3, step=1) reaction_weight_mode = container_param.select_slider( "Reaction Weight w:", ["Random", "Fixed"]) if reaction_weight_mode == "Fixed": reaction_weight = container_param.slider( label='reaction_weight w', min_value=0, max_value=1, value=-1, step=.001) elif reaction_weight_mode == "Random": reaction_weight = -1 if container_button.button('Analyse'): gen.get_stats(story_till_now="Hello, I'm a language model,", num_generation=num_generation, length=length, reaction_weight=reaction_weight, num_tests=num_tests) if len(gen.stories) > 0: for si, story in enumerate(gen.stories): st.markdown(f'### Story no. {si}:', unsafe_allow_html=False) st.markdown(story, unsafe_allow_html=False) data=gen.stats_df[gen.stats_df.sentence_no==3] fig = px.violin(data_frame=data, x="reaction_weight", y="num_reactions", hover_data=data.columns) st.plotly_chart(fig, use_container_width=True) elif mode == 'Create Statistics': container_mode.write('Let\'s play storytelling.') # # , placeholder="Start writing your story...") # story_till_now = st.text_input( # label='First Sentence', value='Hello, I\'m a language model,') # num_generation = st.sidebar.slider( # label='Number of generation', min_value=1, max_value=100, value=10, step=1) # length = st.sidebar.slider(label='Length of the generated sentence', # min_value=1, max_value=100, value=20, step=1) if container_button.button('Run'): story_till_now, emotion = gen.story( story_till_now, num_generation, length) st.write('Story:') st.text(story_till_now) st.text(f'Emotion: {emotion}') else: st.write('Write the first sentence and then hit the Run button')