import gradio 
import gradio as gr
import os 
from transformers import pipeline 
summarizer = pipeline("summarization")


def my_inference_function(text,key: str):
    if len(str(text)) < 10:
        return "Error, text is too short!"
    if key != os.environ.get("key"):
        return "Error, the key is incorrect!"
    return summarizer(text)

gradio_interface = gradio.Interface(
  fn = my_inference_function,
  inputs =[gr.Textbox(
        label="prompt", placeholder="What is the capital of France"),
        gr.Textbox(label="key")],
  outputs = "text"
)
gradio_interface.launch()