import gradio as gr
from transformers import pipeline

# def greet(name):
#     return "Hello " + name + "!!"

# iface = gr.Interface(fn=greet, inputs="text", outputs="text")
# iface.launch()

chatbot_model = "facebook/blenderbot-400M-distill"

chatbot = pipeline(model=chatbot_model)

def echo(message, history):
    return message

def chatbot_response(message, chat_history):
    chat_history.append(message)
    bot_message = chatbot(chat_history)
    chat_history.append(bot_message)
    return bot_message, chat_history

app = gr.ChatInterface(fn=chatbot_response, examples=["hello", "hola", "merhaba"], title="Chat Bot")
app.launch()