harshguptarmp/generativeAI

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 helpful assistant to answer all user queries.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