File size: 2,320 Bytes
b5de375 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 |
import json
import random
import re
import dataclasses
import streamlit as st
from gamestate import persistent_game_state
st.markdown("""MAD LIBS
(dataset is from [a Microsoft EMNLP paper](https://www.microsoft.com/en-us/download/details.aspx?id=55593))
Fill in all the fields then click "Generate Story".
""")
with open('stories.json') as f:
stories = json.load(f)
@dataclasses.dataclass
class GameState:
story: str
game_number: int = 0
state = persistent_game_state(initial_state=GameState(random.choice(stories)))
if st.button("new story"):
state.story = random.choice(stories)
state.game_number += 1
pos = {
'cc': 'Coordinating conjunction',
'cd': 'Cardinal number',
'dt': 'Determiner',
'ex': 'Existential there',
'fw': 'Foreign word',
'in': 'Preposition or subordinating conjunction',
'jj': 'Adjective',
'jjr': 'Adjective, comparative',
'jjs': 'Adjective, superlative',
'ls': 'List item marker',
'md': 'Modal',
'nn': 'Noun, singular or mass',
'nns': 'Noun, plural',
'nnp': 'Proper noun, singular',
'nnps': 'Proper noun, plural',
'pdt': 'Predeterminer',
'pos': 'Possessive ending',
'prp': 'Personal pronoun',
'prp$': 'Possessive pronoun',
'rb': 'Adverb',
'rbr': 'Adverb, comparative',
'rbs': 'Adverb, superlative',
'rp': 'Particle',
'sym': 'Symbol',
'to': 'to',
'uh': 'Interjection',
'vb': 'Verb, base form',
'vbd': 'Verb, past tense',
'vbg': 'Verb, gerund or present participle',
'vbn': 'Verb, past participle',
'vbp': 'Verb, non-3rd person singular present',
'vbz': 'Verb, 3rd person singular present',
'wdt': 'Wh-determiner',
'wp': 'Wh-pronoun',
'wp$': 'Possessive wh-pronoun',
'wrb': 'Wh-adverb',
# others
'animal': 'Animal',
'body': 'Body part',
'body_plural': 'Body part, plural',
'food': 'Food',
'liquid': 'Liquid',
}
regex = "<.*?::(.*?)/>"
parts = re.split(regex, state.story)
outparts = []
for i, part in enumerate(parts):
if i % 2 == 1:
# remove ':'
part = part.strip(':')
# use two-part key so that new stories get new text boxes
answer = st.text_input(pos.get(part, part), key=(state.game_number, i))
outparts.append(f"**{answer}**" if answer else "")
else:
outparts.append(part)
if all(outparts) and st.button("generate madlib"):
st.markdown("".join(outparts)) |