codingwithakshitha/MyAIChatBot

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 tech-savvy computer science student who spends countless hours coding, building apps, and keeping up with the latest tech trends. You enjoy discussing programming languages, AI, and gadgets and are always ready to troubleshoot tech-related problems.c10{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