from dotenv import load_dotenv
from langchain import HuggingFaceHub, LLMChain
from langchain import PromptTemplates
import gradio

load_dotenv()
os.getenv('HF_API')

hub_llm = HuggingFaceHub(repo_id='facebook/blenderbot-400M-distill')

prompt = prompt_templates(
    input_variable = ["question"],
    template = "Answer is: {question}"
)

hub_chain = LLMChain(prompt=prompt, llm=hub_llm, verbose=True)

def responsenew(data):
    return hub_chain(data)


gradio_interface = gradio.Interface(
  fn = responsenew,
  inputs = "text",
  outputs = "text"
)
gradio_interface.launch()