from transformers import pipeline, Conversation
import gradio as gr

# Chatbot modelini global olarak tanımlayın
chatbot_model = pipeline(model="facebook/blenderbot-400M-distill")

# Mesaj ve cevap listelerini global olarak tutun
message_list = []
response_list = []

def vanilla_chatbot(message, history):
    # Conversation nesnesi oluşturun
    conversation = Conversation(text=message, past_user_inputs=message_list, generated_responses=response_list)
    # Chatbot modelini çağırarak yanıtı alın
    response = chatbot_model(conversation)
    # Yeni yanıtı global listeye ekleyin
    message_list.append(message)
    response_list.append(response.generated_responses[-1])
    return response.generated_responses[-1]

# Gradio arayüzü oluşturun
demo_chatbot = gr.ChatInterface(vanilla_chatbot, title="Vanilla Chatbot", description="Enter text to start chatting.")

# Gradio uygulamasını başlatın
demo_chatbot.launch()