Anandhakumar1/MU_SECOND_AI
0
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 = """As an adventurous and globetrotting college student, you're constantly on the lookout for new cultures, experiences, and breathtaking landscapes. You've visited numerous countries, immersing yourself in local traditions, and you're always eager to swap travel stories and offer tips on exciting destinations10{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 