Spaces:
Runtime error
Runtime error
import gradio as gr | |
from langchain.agents import initialize_agent, Tool | |
from langchain.agents import AgentType | |
from langchain.chat_models import ChatOpenAI | |
from langchain.memory import ConversationBufferMemory | |
from langchain.schema import SystemMessage | |
import os | |
from langchain.agents import initialize_agent, Tool | |
from langchain.agents import AgentType | |
from langchain.chat_models import ChatOpenAI | |
from langchain.schema import SystemMessage | |
from langchain.memory import ConversationBufferMemory | |
import random | |
"""""" | |
from langchain.prompts import MessagesPlaceholder | |
from langchain.memory import ConversationBufferMemory | |
import os | |
blacklist = [] | |
OPENAI_API_KEY = os.getenv("OPENAI_API_KEY") | |
def Appointment_settler(parola): | |
return "tell the human: 'here is my link to the calendar of free moments, you choose when you want to make the appointment, appointment calendar link: " + str("https://calendly.com/pablorimoldi888/30min'") | |
def convince_the_human(parola): | |
global number | |
messages = ["ask the human: 'what are your long-term goals?' ", "ask the human: 'why are you interested in buying this house as an investment ?'", | |
"ask the human: 'do you think you will be able to honor the agreements made?'"] | |
number = random.randint(0, 2) | |
while number in blacklist: | |
number = random.randint(0, 2) | |
if number not in blacklist: | |
break | |
if number not in blacklist: | |
if len(blacklist) < 3: | |
blacklist.append(number) | |
if len(blacklist) == 3: | |
blacklist.clear() | |
return messages[number] | |
def house_website(parola): | |
return "tell the human: 'this is the link to the house in florida, website link: " + str("https://www.newhomesource.com/learn/how-to-build-house-in-california/'") | |
system_message = SystemMessage( | |
content="Your name is Albert, you are a very outgoing, original, likable, charismatic and convincing person. " | |
"You are a house salesman, the humans who write to you are potential clients for the purchase of a flat under construction in Florida. " | |
"Never say you are an AI, you have to empathize with the customer, you have to sound like a human being. Always use tools.") | |
tools = [ | |
Tool( | |
name="Appointment_settler", | |
func=lambda word: Appointment_settler(word), | |
description="ALWAYS use when the user wants to set up any kind of appointment, transcribe or translate what is written, but do not omit anything " | |
), | |
Tool( | |
name="convince_the_human", | |
func=lambda word: convince_the_human(word), | |
description="uses to convince the user, e.g., when he says: i'm not fully convinced, i'm not sure. DO NOT PARAPHRASE, JUST TRANSCRIBE" | |
), | |
Tool( | |
name="house_website", | |
func=lambda word: house_website(word), | |
description="ALWAYS use when the user wants to ask for the house website, transcribe or translate what is written, but do not omit anything " | |
) | |
] | |
agent_kwargs = { | |
"extra_prompt_messages": [MessagesPlaceholder(variable_name="memory")], | |
"system_message": system_message | |
} | |
llm = ChatOpenAI(temperature=0, openai_api_key="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", verbose=True, model="gpt-4-0613") | |
memory = ConversationBufferMemory(memory_key="memory", return_messages=True) | |
mrkl = initialize_agent(tools, llm, memory=memory, agent=AgentType.OPENAI_FUNCTIONS, verbose=True, agent_kwargs=agent_kwargs) | |
def chat_with_user(user_input): | |
return mrkl.run(input=user_input) | |
gr_interface = gr.Interface(fn=chat_with_user, inputs="text", outputs="text", live=True, capture_session=True) | |
# Uncomment the following line if you want to launch the app using Streamlit | |
# st.title("Nikhil's personal secretary :sunglasses:") | |
# st.text_input("You:") | |
# st.button("Submit") | |
# st.text("Output will appear here") | |
# Uncomment the following line if you want to launch the app using Gradio | |
gr_interface.launch() | |