import sys
import time
import streamlit as st
import concurrent.futures
from transformers import pipeline, set_seed
from random import randint

generator = pipeline('text-generation', model='openai-gpt')

def generate(initial_text, length=10, return_sequences=1):
    set_seed(randint(1,1000))
    result = generator(initial_text, max_length = length, num_return_sequences = return_sequences)
    return result[0]["generated_text"]

def slice(text, mak_length=10):
    return text[-mak_length:]

def type_text(text):
    for letter in text:
        sys.stdout.write(letter)
        time.sleep(0)

if __name__ == "__main__":
    text = input("Enter something to begin with... ")
    print(".\n.\n.\nGenerating\n.\n.\n.")
    
    for _ in range(50):
        result = generate(text)
        text=slice(result)
        with concurrent.futures.ThreadPoolExecutor() as executor:
            executor.submit(type_text, result.replace(text,""))

    x = st.slider('Select a value')
    st.write(x, 'squared is', x * x)