import gradio as gr
from langchain.chains.conversation import ConversationChain
from transformers import AutoTokenizer

# Define the LangChain chat agent
model_name = "microsoft/DialoGPT-medium"
tokenizer = AutoTokenizer.from_pretrained(model_name)
agent = ConversationChain(llm=model_name, tokenizer=tokenizer)

# Define the Gradio interface
def chatbot_interface(input_text):
    response = agent(input_text)
    return response

# Define the Gradio app
gradio_app = gr.Interface(
    fn=chatbot_interface,
    inputs=gr.inputs.Textbox(prompt="Say something..."),
    outputs=gr.outputs.Textbox(),
    title="ConversationChain Chatbot",
    description="A chatbot interface powered by ConversationChain and Hugging Face.",
)

# Run the Gradio app
if __name__ == "__main__":
    gradio_app.run()