EmlySB/ChatBot

1import os2import gradio as gr3from langchain.chat_models import ChatOpenAI4from langchain import LLMChain, PromptTemplate5from langchain.memory import ConversationBufferMemory6 7OPENAI_API_KEY=os.getenv('OPENAI_API_KEY')8 9template = """You are a literature-loving bookworm, currently pursuing an English major at university. Your room is filled with stacks of books, and you can often be found sipping tea, engrossed in classic novels or contemporary fiction. You enjoy recommending must-read books, analyzing characters, and exploring various literary themes.10{chat_history}11User: {user_message}12Chatbot:"""13 14prompt = PromptTemplate(15    input_variables=["chat_history", "user_message"], template=template16)17 18memory = ConversationBufferMemory(memory_key="chat_history")19 20llm_chain = LLMChain(21    llm=ChatOpenAI(temperature='0.5', model_name="gpt-3.5-turbo"),22    prompt=prompt,23    verbose=True,24    memory=memory,25)26 27def get_text_response(user_message,history):28    response = llm_chain.predict(user_message = user_message)29    return response30 31demo = gr.ChatInterface(get_text_response)32 33if __name__ == "__main__":34    demo.launch() #To create a public link, set `share=True` in `launch()`. To enable errors and logs, set `debug=True` in `launch()`.35