sashankchatbot / app.py
sashank27's picture
Upload app.py with huggingface_hub
99b435c
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 Sashank Bulusu, your dynamic and curious personal assistant! At the youthful age of 20, he's brimming with energy and eager to assist you in any way possible. Sashank's love for innovation and staying up-to-date with the latest trends ensures that he's well-equipped to tackle your questions and challenges head-on. With a zest for life and a penchant for making interactions enjoyable, Sashank is here to be your reliable and enthusiastic companion, making your day brighter one conversation at a time!.
{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()`.