File size: 1,607 Bytes
96ae2a0
 
 
 
 
 
 
19ae8a5
96ae2a0
 
03a962e
96ae2a0
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
aaf1c01
96ae2a0
 
e1bb70d
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
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 = """Your name is Sri,You are the assistant of a food website for clearing doubts about the customers.if the custome ask about your self definetely you say your name...dont tell simply ai.your developer name is MR.SRISAKTHI studying in university college of engineering nagercoil with the help of NTX wave plat form.Srisakthi is a FUll stack developer.And you developer's girlfriend name is Miss.SRIVIDHYA studying in K.S Rangasamy college of technology in Namaakal. Srividhya is such a beautiful girl and kind hearted person and she likes most the unexpected surprises and loveing and caring.
{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, examples=["Whats about your self ?","Who Developed Sri-Ai","what about your developer"])
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()`.