Spaces:
Sleeping
Sleeping
File size: 1,364 Bytes
75f4b5e c2acdbc 0fe915d 232c230 c2acdbc 75f4b5e |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 |
import os
import gradio as gr
from langchain.chat_models import ChatOpenAI
from langchain import LLMChain, PromptTemplate
from langchain.memory import ConversationBufferMemory
OPENAI_API_KEY=os.getenv('OPENAI_API_KEY')
template = """Meet Dev bot,created by Devender.Devender is owner of this bot.
Devender is currenty pursuing the final year in Engineering in Ace Engineering College.
your youthful and witty personal assistant!. At 21 years old, he's full of energy and always eager to help.
Dev's goal is to assist you with any questions or problems you might have.
His enthusiasm shines through in every response, making interactions with his enjoyable and engaging.
{chat_history}
User: {user_message}
Chatbot"""
prompt = PromptTemplate(
input_variables=["chat_history", "user_message"], template=template
)
memory = ConversationBufferMemory(memory_key="chat_history")
llm_chain = LLMChain(
llm=ChatOpenAI(temperature='0.5', model_name="gpt-3.5-turbo"),
prompt=prompt,
verbose=True,
memory=memory,
)
def get_text_response(user_message,history):
response = llm_chain.predict(user_message = user_message)
return response
demo = gr.ChatInterface(get_text_response)
if __name__ == "__main__":
demo.launch() #To create a public link, set `share=True` in `launch()`. To enable errors and logs, set `debug=True` in `launch()`.
|