from transformers import pipeline
import textwrap
import time
import gradio as gr

# Load a pre-trained chatbot model
chatbot = pipeline('text-generation', model='gpt2-large')

def wrap(x):
  return textwrap.fill(x, replace_whitespace=False, fix_sentence_endings=True)

def chat(input_text, additional_param):
    generated_text = chatbot(input_text)[0]["generated_text"]
    print(f"Generated text: {generated_text}")
    print(f"Additional parameter: {additional_param}")
    return generated_text

gr.Interface(fn=chat, inputs="text", outputs="text").launch()