| import gradio as gr | |
| import random | |
| import markovify | |
| def gen(text): | |
| if text in ["1091393561192304661","848565511834566677", "1077278396393996420"]: | |
| size = random.randint(1, 650) | |
| file_name = "./premium.txt" | |
| with open(file_name, encoding="utf-8") as f: | |
| db = f.read() | |
| db = db.strip().lower() | |
| text_model = markovify.NewlineText(input_text=db, state_size=1, well_formed=False) | |
| symbolsplus = [".","~~","!","?"] | |
| sentence: str = text_model.make_short_sentence(size) or random.choice( | |
| db.splitlines())[:size] + random.choice(symbolsplus) | |
| return sentence | |
| else: | |
| size = random.randint(1, 128) | |
| file_name = "./free.txt" | |
| with open(file_name, encoding="utf-8") as f: | |
| db = f.read() | |
| db = db.strip().lower() | |
| text_model = markovify.NewlineText(input_text=db, state_size=1, well_formed=False) | |
| symbolsplus = [".","~~","!","?"] | |
| sentence: str = text_model.make_short_sentence(size) or random.choice( | |
| db.splitlines())[:size] + random.choice(symbolsplus) | |
| return sentence | |
| interface = gr.Interface(fn=gen,inputs="text",outputs="text",examples=[["ID!"], ["Как дела?"]]) | |
| interface.launch() |