# from dotenv import load_dotenv
from langchain import LLMChain, HuggingFaceHub
from langchain.prompts import PromptTemplate
import gradio as gr

# load_dotenv()


def storyTime(input_letter):
    hub_llm = HuggingFaceHub(
        repo_id="gpt2-large",
        model_kwargs={'temperature': 0.5, 'max_length': 200}
    )
    prompt = PromptTemplate(
        input_variables=["letter"],
        template=" {letter}"
    )

    hub_chain = LLMChain(prompt=prompt, llm=hub_llm, verbose=True)
    return hub_chain.run(input_letter)


iface = gr.Interface(
    fn=storyTime,
    inputs='text',
    outputs=['text'],
    title='Story Time😃',
    description='Write a simple text you want a story on'
)

iface.launch(inline=False)

# input_letter = input("Write the first sentence of your letter:  ")
# main(input_letter)