storytelling / src /lib.py
jitesh's picture
adds download button
1115bb5
raw
history blame
2.84 kB
import random
import streamlit as st
from src import StoryGenerator
import xlsxwriter
import pandas as pd
import io
def create_dowload_button(data, sheet_name='AllData', label="Download data", file_name='data.xlsx'):
buffer = io.BytesIO()
with pd.ExcelWriter(buffer, engine='xlsxwriter') as writer:
# Write each dataframe to a different worksheet.
data.to_excel(writer, sheet_name=sheet_name)
# Close the Pandas Excel writer and output the Excel file to the buffer
writer.save()
st.download_button(
label=label,
data=buffer,
file_name=file_name,
mime='application/vnd.ms-excel',
)
# @st.cache(allow_output_mutation=True)
def initialise_storytelling(gen, container_guide, container_param, container_button):
gen.initialise_models()
choices_first_sentence = [
'Custom',
'Hello, I\'m a language model,',
'So I suppose you want to ask me how I did it.',
'I always wanted to be a giraffe - until that night.',
'My first tutor was a dragon with a terrible sense of humor.',
'Doctors told her she could never diet again.',
'Memory is all around us, as well as within.',
]
cfs = st.selectbox('Choose First Sentence', choices_first_sentence)
if cfs == 'Custom':
story_till_now = st.text_input(
label='First Sentence', key='first_sentence')
else:
st.session_state.first_sentence = cfs
story_till_now = cfs
first_sentence = story_till_now
first_emotion = gen.get_emotion(first_sentence)
length = set_input(container_param,
label='Length of the sentence',
min_value=1, max_value=100, value=10, step=1,
key_slider='length_slider', key_input='length_input',)
return first_sentence, first_emotion, length
# @st.cache(allow_output_mutation=True)
def set_input(container_param,
label, key_slider, key_input,
min_value=0.,
max_value=1.,
value=.5,
step=.01,):
def slider2input():
st.session_state[key_input] = st.session_state[key_slider]
def input2slider():
st.session_state[key_slider] = st.session_state[key_input]
container_param = container_param.columns([1.1, 1])
number_input = container_param[0].number_input(
label=label,
min_value=min_value,
max_value=max_value,
value=value,
step=step,
key=key_input,
on_change=input2slider)
slider_input = container_param[1].slider(
label='',
min_value=min_value,
max_value=max_value,
value=value,
step=step,
key=key_slider,
on_change=slider2input)
return number_input