import gradio as gr
from transformers import BlenderbotTokenizer, BlenderbotForConditionalGeneration

# Model Load
model_name = "facebook/blenderbot-3B"
tokenizer = BlenderbotTokenizer.from_pretrained(model_name)
model = BlenderbotForConditionalGeneration.from_pretrained(model_name)

# Chatbot Function
def chatbot_response(user_input):
    inputs = tokenizer(user_input, return_tensors="pt")
    reply_ids = model.generate(**inputs)
    response = tokenizer.decode(reply_ids[0], skip_special_tokens=True)
    return response

# Gradio Interface
iface = gr.Interface(
    fn=chatbot_response,
    inputs="text",
    outputs="text",
    title="Celebrity AI Chatbot",
    description="Apne favourite celebrity se baat karein!",
)

iface.launch()