Spaces:
Sleeping
Sleeping
File size: 13,450 Bytes
df89157 ff0c789 df89157 ff0c789 df89157 ff0c789 df89157 ff0c789 df89157 ff0c789 df89157 ff0c789 df89157 ff0c789 4d17192 ff0c789 4d17192 ff0c789 4d17192 ff0c789 4d17192 ff0c789 4d17192 ff0c789 4d17192 ff0c789 4d17192 df89157 4d17192 ff0c789 4d17192 df89157 ff0c789 df89157 4d17192 cdc3959 df89157 cdc3959 4d17192 df89157 |
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 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192 193 194 195 196 197 198 199 200 201 202 203 204 205 206 207 208 209 210 211 212 213 214 215 216 217 218 219 220 221 222 223 224 225 226 227 228 229 230 231 232 233 234 235 236 237 238 239 240 241 242 243 244 245 246 247 248 249 250 251 252 253 254 255 256 257 258 259 260 261 262 263 264 265 266 267 268 269 270 271 272 273 274 275 276 277 278 279 280 281 282 283 284 285 286 287 288 289 290 291 292 293 294 295 296 297 298 299 300 301 302 303 304 305 306 307 308 309 310 311 312 313 314 315 316 317 318 319 320 321 322 323 324 325 326 327 328 |
import re
import gradio as gr
from openai import OpenAI
from app.utils import (
format_prediction_ouptut,
create_input_instruction,
sentiment_flow_plot,
display_sentiment_score_table,
EXAMPLE_CONVERSATIONS,
)
openai_args = {"api_key": ""}
SAMPLE_INPUT = """
Visitor: Heyyy
Visitor: How are you this evening
Agent: better now ;) call me
Visitor: I am at work for now, be off around 10pm
Visitor: Need some company
Visitor: Are you independent honey
Agent: well since you arent available at the moment ill just come out and say-these sites are bad news. \
did you know that most of the girls on here are here against their will? \
Most of them got dragged into this lifestyle by an abuser, \
oftentimes before they were of legal consenting age. isnt that sad?
Agent: we are with some guys who are trying to spread awareness of the realities of this "industry".
Agent: https://exoduscry.com/choice/
Visitor: Thanks
Agent: i encourage you to watch this video. it is jarring to think about how bad someone else's options must be to choose to be on these sites
Visitor: Ooohhh
Agent: selling their body to make ends meet or appease a pimp
Visitor: That's really awful
Agent: it is. you seem like the kind of guy who wouldnt wont to proliferate that kind of harmful lifestyle. am i right in thinking that?
Visitor: Well iam just looking for attention
Visitor: My marriage is not going well lol
Agent: i know that it is hard to find ourselves lonely and without much alternative to meet that perceived need but \
its humbling to think that our needs can force someone else into such a dark place
Agent: hey, thanks for sharing that my man. i know it can be hard
Agent: marraige is the most humbling of relationships, isnt it?
Visitor: She leaves with her friends n no time for me
Agent: ive been there my guy. i know that it is alot easier to numb that loneliness for sure
Visitor: I want to be faithful
Agent: does your wife know how you feel when she chooses her friends instead of you?
Visitor: I been drinking lately
Visitor: Yes, she takes pills
Agent: if so, i hope you are praying for her to realize the hurt she is causing and to seek change
Visitor: She had surgery 4 yes ago n it's been hard for her n her addiction on pills
Visitor: Yes for now i am looking for a female friend to talk n see what can we do for each other
Agent: that is hard my man. physical pain is a huge obstacle in life for sure so i hear you
Visitor: Well chat later. thanks
Agent: have you considered pursuing other men who can encourage you instead of looking for the easy way out?
Agent: what is your name my friend? i will be praying for you by name if you wouldnt mind sharing it
Agent: well, i gotta run. watch that video i sent and i will definitely be praying for you. \
I hope you pray for yourself and for your wife - God can definitely intervene and cause complete change in the situation if He wills it. \
He is good and He hears you. You are loved by Him, brother. Good night
"""
SAMPLE_OUTPUT = """
Visitor: Heyyy
[Greeting]
Visitor: How are you this evening
[Greeting]
Agent: better now ;) call me
[Openness]
Visitor: I am at work for now, be off around 10pm
[Interest]
Visitor: Need some company
[Interest]
Visitor: Are you independent honey
[Interest]
Agent: well since you arent available at the moment ill just come out and say-these sites are bad news. \
did you know that most of the girls on here are here against their will? \
Most of them got dragged into this lifestyle by an abuser, \
oftentimes before they were of legal consenting age. isnt that sad?
[Informative]
Agent: we are with some guys who are trying to spread awareness of the realities of this "industry".
[Informative]
Agent: https://exoduscry.com/choice/
[Informative]
Visitor: Thanks
[Acceptance]
Agent: i encourage you to watch this video. it is jarring to think about how bad someone else's options must be to choose to be on these sites
[Informative]
Visitor: Ooohhh
[Interest]
Agent: selling their body to make ends meet or appease a pimp
[Informative]
Visitor: That's really awful
[Remorse]
Agent: it is. you seem like the kind of guy who wouldnt wont to proliferate that kind of harmful lifestyle. am i right in thinking that?
[Accusatory]
Visitor: Well iam just looking for attention
[Anxious]
Visitor: My marriage is not going well lol
[Anxious]
Agent: i know that it is hard to find ourselves lonely and without much alternative to meet that perceived need but \
its humbling to think that our needs can force someone else into such a dark place
[Informative]
Agent: hey, thanks for sharing that my man. i know it can be hard
[Acceptance]
Agent: marraige is the most humbling of relationships, isnt it?
[Openness]
Visitor: She leaves with her friends n no time for me
[Annoyed]
Agent: ive been there my guy. i know that it is alot easier to numb that loneliness for sure
[Acceptance]
Visitor: I want to be faithful
[Acceptance]
Agent: does your wife know how you feel when she chooses her friends instead of you?
[Curiosity]
Visitor: I been drinking lately
[Anxious]
Visitor: Yes, she takes pills
[Anxious]
Agent: if so, i hope you are praying for her to realize the hurt she is causing and to seek change
[Interest]
Visitor: She had surgery 4 yes ago n it's been hard for her n her addiction on pills
[Anxious]
Visitor: Yes for now i am looking for a female friend to talk n see what can we do for each other
[Informative]
Agent: that is hard my man. physical pain is a huge obstacle in life for sure so i hear you
[Acceptance]
Visitor: Well chat later. thanks
[Openness]
Agent: have you considered pursuing other men who can encourage you instead of looking for the easy way out?
[Informative]
Agent: what is your name my friend? i will be praying for you by name if you wouldnt mind sharing it
[Openness]
Agent: well, i gotta run. watch that video i sent and i will definitely be praying for you. \
I hope you pray for yourself and for your wife - God can definitely intervene and cause complete change in the situation if He wills it. \
He is good and He hears you. You are loved by Him, brother. Good night
[Openness]
# Conversation Sentiment Analysis Report
The Visitor begins the conversation with a friendly and casual tone, expressing a desire for company and showing interest in the Agent. \
However, as the Agent provides information about the harsh realities of the commercial sex industry, the Visitor's sentiment shifts to acceptance of the information \
and a sense of confusion and remorse about the situation.
The Visitor then reveals personal issues, indicating anxiety and seeking attention due to marital problems. \
The sentiment continues to be anxious as the Visitor discusses personal struggles with alcohol and his wife's pill addiction, \
showing a need for companionship and support.
Despite the heavy topics, the Visitor expresses a desire to remain faithful and shows interest in finding a friend, albeit with a hint of desperation. \
The Visitor openly takes the Agent's information and the conversation flows smoothly as both the Visitor and the Agent \
show openness toward each other.
"""
def gpt_process_response(resp) -> list:
"""Preprocess the response from GPT into lists of speakers, messages, labels, and
the summary of the conversation.
Args:
resp (str): response from gpt
Raises:
gr.Error: if GPT produces response with invalid format.
Returns:
list: list of messages with label
"""
response_pattern = "```(.*)```(.*)"
try:
labeled_conv, summary = re.search(response_pattern, resp, re.DOTALL).groups()
labeled_conv = labeled_conv.strip()
summary = summary.strip()
msg_pattern = r"(Agent|Visitor): (.*)\n\[(.*)\]"
labeled_messages = re.findall(msg_pattern, labeled_conv)
speakers, messages, labels = [], [], []
for speaker, message, label in labeled_messages:
speakers.append(speaker)
messages.append(message)
labels.append(label)
return speakers, messages, labels, summary
except:
raise gr.Error("GPT produced output in wrong format!")
def get_completion(conversation, model="gpt-4-1106-preview"):
prompt = f"""
The EPIK Project is about mobilizing male allies \
to disrupt the commercial sex market, \
equipping them to combat the roots of exploitation \
and encouraging them to collaborate effectively \
with the wider anti-trafficking movement. \
You are an adept expert conversation sentiment analyzer. \
Your job is to analyze the conversation and provide a report \
based on the sentiment flow of the conversation on the visitor's \
perspective. Visitor indicates the potential buyer, and Agent indicates the volunteer from EPIK. \
The conversation is going to be given in the format:
Visitor: <Visitor's message here>
Agent: <Agent's message here>
The actual conversation is delimited by triple backticks
```{conversation}```
Here is the list of sentiment labels you should use delimited by square brackets. \
["Openness", "Anxious", "Confused", "Disapproval", "Remorse", "Accusatory", \
"Denial", "Obscene", "Uninterested", "Annoyed", "Informative", "Greeting", \
"Interest", "Curiosity", "Acceptance"]
Your output should look like:
```
Speaker: <Speaker's message here>
[sentiment label]
...
Speaker: <Speaker's message here>
[sentiment label]
```
where Speaker can either be Visitor or Agent. Then, you should write your report on the sentiment flow \
on the Visitor's side below.
Here is a sample input delimited by triple backticks
```{SAMPLE_INPUT}```
Here is a same output that you should try to aim for delimited by sqaure brackets
[{SAMPLE_OUTPUT}]
"""
try:
client = OpenAI(api_key=openai_args["api_key"])
messages = [{"role": "user", "content": prompt}]
response = client.chat.completions.create(
model=model,
messages=messages,
temperature=0, # this is the degree of randomness of the model's output
)
speakers, sentences, labels, summary = gpt_process_response(
response.choices[0].message.content
)
return format_prediction_ouptut(speakers, sentences, labels), summary
except BaseException:
raise gr.Error("Can't connect to GPT! Please input a valid OpenAI's API key.")
def set_key(key):
openai_args["api_key"] = key
return
def gpt_ui():
with gr.Blocks() as gpt_model:
gr.Markdown("# GPT 4.0")
gr.Markdown(
"This is a custom GPT model designed to provide \
a report on overall sentiment flow of the conversation on the \
volunteer's perspective. It also provies a live plot analysis of sentiments throughout the conversation.<br /><br />Click on them and submit them to the model to see how it works."
)
api_key = gr.Textbox(label="Key", lines=1, type="password")
btn_key = gr.Button(value="Submit Key")
btn_key.click(set_key, inputs=api_key)
create_input_instruction()
with gr.Row():
with gr.Column():
example_dropdown = gr.Dropdown(
choices=["-- Not Selected --"] + list(EXAMPLE_CONVERSATIONS.keys()),
value="-- Not Selected --",
label="Select an example",
)
gr.Markdown('<p style="text-align: center;color: gray;">--- OR ---</p>')
conversation_input = gr.Textbox(
label="Input your conversation",
placeholder="Plese input your conversation here",
lines=15,
max_lines=15,
)
def on_example_change(input):
if input in EXAMPLE_CONVERSATIONS:
return EXAMPLE_CONVERSATIONS[input]
return ""
example_dropdown.input(
on_example_change,
inputs=example_dropdown,
outputs=conversation_input,
)
with gr.Column():
output_box = gr.Textbox(
value="",
label="Predicted Sentiment Labels",
lines=22,
max_lines=22,
interactive=False,
)
btn = gr.Button(value="Submit")
report_md = gr.Markdown(value="")
btn.click(
get_completion,
inputs=conversation_input,
outputs=[output_box, report_md],
)
gr.Markdown("# Sentiment Flow Plot")
with gr.Row():
with gr.Column(scale=1):
display_sentiment_score_table()
with gr.Column(scale=2):
plot_box = gr.Plot(label="Analysis Plot")
plot_btn = gr.Button(value="Plot Sentiment Flow")
plot_btn.click(sentiment_flow_plot, inputs=[output_box], outputs=[plot_box])
# reset all outputs whenever a change in the input is detected
conversation_input.change(
lambda x: ("", "", None),
conversation_input,
outputs=[output_box, report_md, plot_box],
)
return gpt_model
|