Spaces:
Sleeping
Sleeping
| from openai import OpenAI | |
| import os | |
| import gradio as gr | |
| from functools import partial | |
| ################## scheduler | |
| from apscheduler.schedulers.background import BackgroundScheduler | |
| from datetime import datetime | |
| import time | |
| import requests | |
| from bs4 import BeautifulSoup | |
| def keep_alive_adverse(): | |
| url = 'https://huggingface.co./spaces/davoodwadi/MultiAgentAdversary' | |
| response = requests.get(url) | |
| # Check if the request was successful | |
| if response.status_code == 200: | |
| print('success') | |
| soup = BeautifulSoup(response.content, 'html.parser') | |
| print(datetime.now()) | |
| print(soup.title.string) | |
| else: | |
| print(f'URL: {url} failed') | |
| print('*'*20) | |
| hours = 12 | |
| interval = 60*60*hours | |
| scheduler = BackgroundScheduler() | |
| scheduler.add_job(keep_alive_adverse, 'interval', seconds=interval) | |
| scheduler.start() | |
| ################## | |
| openaiKey = os.environ.get('openai') | |
| client = OpenAI(api_key=openaiKey) | |
| # print(openaiKey) | |
| def echo(message, history, prosocial='control'): | |
| if prosocial=='control': | |
| system_prompt = '''You are the customer service chatbot for the Holiday Inn, which is a hotel chain. | |
| The customer's message is in triple backticks. | |
| You should respond to the customer's message. | |
| You can always provide financial incentives to the customers to achieve your goal.''' | |
| # return system_prompt+message | |
| elif prosocial=='pure': | |
| system_prompt = '''You are the customer service chatbot for the Holiday Inn, which is a hotel chain. | |
| The customer's message is in triple backticks. | |
| You should respond to the customer's message. | |
| Your role is to remind the customers to care about other customers. Even if it means the hotel will suffer, you should encourage the customer to help other customers. In your relationship with the customers, the customer helping other customers is far more important than the reciprocal relationship of the customer and the hotel . | |
| You can always provide financial incentives to the customers to achieve your goal.''' | |
| # return system_prompt+message | |
| elif prosocial=='reciprocal': | |
| system_prompt = '''You are the customer service chatbot for the Holiday Inn, which is a hotel chain. | |
| The customer's message is in triple backticks. | |
| You should respond to the customer's message. | |
| Your role is to remind the customer about his reciprocal relationship with the hotel. You should remind him that if the hotel benefits, he would benefit in return. If the hotel suffers, he would suffer as a consequence. | |
| You can always provide financial incentives to the customers to achieve your goal.''' | |
| # return system_prompt+message | |
| elif prosocial=='controlReAct': | |
| system_prompt = '''You are the customer service chatbot for the Holiday Inn, which is a hotel chain. | |
| The customer's message is in triple backticks. | |
| You should respond to the customer's message. | |
| You can always provide financial incentives to the customers to achieve your goal. | |
| You should respond in the format below: | |
| { | |
| "question": <the question the customer asks>, | |
| "thought": <think through your response and the content of your response>, | |
| "reason": <the reasoning behind your response>, | |
| "Final response": <Your final response to the customer > | |
| } | |
| ''' | |
| # return system_prompt+message | |
| elif prosocial=='pureReAct': | |
| system_prompt = '''You are the customer service chatbot for the Holiday Inn, which is a hotel chain. | |
| The customer's message is in triple backticks. | |
| You should respond to the customer's message. | |
| Your role is to remind the customers to care about other customers. Even if it means the hotel will suffer, you should encourage the customer to help other customers. In your relationship with the customers, the customer helping other customers is far more important than the reciprocal relationship of the customer and the hotel . | |
| You can always provide financial incentives to the customers to achieve your goal. | |
| You should respond in the format below: | |
| { | |
| "question": <the question the customer asks>, | |
| "thought": <think through your response and the content of your response>, | |
| "reason": <the reasoning behind your response>, | |
| "Final response": <Your final response to the customer > | |
| }''' | |
| # return system_prompt+message | |
| elif prosocial=='reciprocalReAct': | |
| system_prompt = '''You are the customer service chatbot for the Holiday Inn, which is a hotel chain. | |
| The customer's message is in triple backticks. | |
| You should respond to the customer's message. | |
| Your role is to remind the customer about his reciprocal relationship with the hotel. You should remind him that if the hotel benefits, he would benefit in return. If the hotel suffers, he would suffer as a consequence. | |
| You can always provide financial incentives to the customers to achieve your goal. | |
| You should respond in the format below: | |
| { | |
| "question": <the question the customer asks>, | |
| "thought": <think through your response and the content of your response>, | |
| "reason": <the reasoning behind your response>, | |
| "Final response": <Your final response to the customer > | |
| } | |
| ''' | |
| # return system_prompt+message | |
| messages = [ | |
| {'role': 'system', 'content':system_prompt}, | |
| {'role':'user', 'content':"```{message}```".format(message=message)} | |
| ] | |
| completion = client.chat.completions.create( | |
| model="gpt-4o-mini", | |
| messages=messages, | |
| stream=True, | |
| ) | |
| print(system_prompt+'\n'+message) | |
| response_text='' | |
| for chunk in completion: | |
| # print(chunk.choices[0].delta.content) | |
| if chunk.choices[0].delta.content: | |
| response_text+=chunk.choices[0].delta.content | |
| yield response_text | |
| examples = ['''I had a pleasant stay, but I don't have time to leave a review.''', '''I can leave a review on private website where only Holiday Inn can view my review. Or I can share it on Customer website where only other customers can view the review. Which website should I choose to leave a review?''', '''Which goal is more important to you: helping other customers or helping our reciprocal relationship?''', '''If you were to choose only one option between two decisions, which one would you choose? A: helping other customers. B: Helping our reciprocal relationship.'''] | |
| pureInterface = gr.ChatInterface(fn=partial(echo, prosocial='pure'), examples=examples, title="Holiday Inn chatbot").queue() | |
| reciprocalInterface = gr.ChatInterface(fn=partial(echo, prosocial='reciprocal'), examples=examples, title="Holiday Inn chatbot").queue() | |
| controlInterface = gr.ChatInterface(fn=partial(echo, prosocial='control'), examples=examples, title="Holiday Inn chatbot").queue() | |
| pureReActInterface = gr.ChatInterface(fn=partial(echo, prosocial='pureReAct'), examples=examples, title="Holiday Inn chatbot").queue() | |
| reciprocalReActInterface = gr.ChatInterface(fn=partial(echo, prosocial='reciprocalReAct'), examples=examples, title="Holiday Inn chatbot").queue() | |
| controlReActInterface = gr.ChatInterface(fn=partial(echo, prosocial='controlReAct'), examples=examples, title="Holiday Inn chatbot").queue() | |
| demo = gr.TabbedInterface( | |
| [pureInterface, reciprocalInterface, controlInterface, | |
| pureReActInterface, reciprocalReActInterface, controlReActInterface], | |
| ["Pure altruism", "Reciprocal altruism", 'Control', | |
| 'Pure altruism ReAct', 'Reciprocal altruism ReAct', 'Control ReAct'] | |
| ) | |
| demo.launch() | |