Merged COSMIC and GPT UI

README.md

@@ -5,7 +5,7 @@ colorFrom: green
 colorTo: pink
 sdk: gradio
 sdk_version: 4.7.1
-app_file: app/app.py
+app_file: main.py
 pinned: false
 ---
 

app/__init__.py

@@ -29,6 +29,7 @@ roberta = RobertaModel.from_pretrained(
 )
 roberta.eval()
 
+# decide = 0 for running with gpu, device="cpu" for running with gpu
 comet = CSKFeatureExtractor(dir=EXTRACTORS_PATH, device="cpu")
 
 cosmic_args = parse_cosmic_args()

app/cosmic_view.py

@@ -3,10 +3,14 @@ import pickle
 import tempfile
 import gradio as gr
 from tqdm import tqdm
-from views.utils import (
+from app.utils import (
     create_input_instruction,
     format_prediction_ouptut,
     remove_temp_dir,
+    decode_numeric_label,
+    decode_speaker_role,
+    display_sentiment_score_table,
+    sentiment_flow_plot,
     EXAMPLE_CONVERSATIONS,
 )
 from fairseq.data.data_utils import collate_tokens
@@ -15,7 +19,7 @@ import sys
 
 sys.path.insert(0, "../")  # neccesary to load modules outside of app
 
-from views import roberta, comet, COSMIC_MODEL, cosmic_args
+from app import roberta, comet, COSMIC_MODEL, cosmic_args
 from preprocessing import preprocess
 from Model.COSMIC.erc_training.predict_epik import predict, get_valid_dataloader
 
@@ -148,9 +152,11 @@ def cosmic_classifier(input):
 
     # Assuming that there's only one conversation
     conv_id = ids[0]
-    output = format_prediction_ouptut(
-        speakers[conv_id], sentences[conv_id], predictions[0]
-    )
+    speaker_roles = [
+        decode_speaker_role(numeric_role) for numeric_role in speakers[conv_id]
+    ]
+    labels = [decode_numeric_label(pred) for pred in predictions[0]]
+    output = format_prediction_ouptut(speaker_roles, sentences[conv_id], labels)
 
     print()
     print("======= Removing Temporary Directory =======")
@@ -190,9 +196,9 @@ def cosmic_ui():
                 conversation_input = gr.TextArea(
                     value="",
                     label="Input you conversation",
-                    placeholder="Plese input your conversation here.\n\n\n\nMaximum number of lines: 200",
-                    lines=5,
-                    max_lines=200,
+                    placeholder="Plese input your conversation here",
+                    lines=15,
+                    max_lines=15,
                 )
 
                 def on_example_change(input):
@@ -207,15 +213,34 @@ def cosmic_ui():
                     outputs=conversation_input,
                 )
 
-                submit_btn = gr.Button(value="Submit")
-
             with gr.Column():
-                gr.Markdown(
-                    '</br></br></br></br><h3 style="text-align: center;">Predicted Sentiment Labels for the Conversation</h3></br>'
+                output = gr.Textbox(
+                    value="",
+                    label="Predicted Sentiment Labels",
+                    lines=22,
+                    max_lines=22,
+                    interactive=False,
                 )
-                output = gr.Markdown(value="", label="Output")
+        submit_btn = gr.Button(value="Submit")
+        submit_btn.click(cosmic_classifier, conversation_input, output)
 
-                submit_btn.click(cosmic_classifier, conversation_input, output)
+        # reset the output whenever a change in the input is detected
+        conversation_input.change(lambda x: "", conversation_input, output)
 
-            conversation_input.change(lambda x: "", conversation_input, output)
+        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], outputs=[plot_box])
+
+        # reset all outputs whenever a change in the input is detected
+        conversation_input.change(
+            lambda x: ("", None),
+            conversation_input,
+            outputs=[output, plot_box],
+        )
     return cosmic_model

app.py → app/gpt_view.py

@@ -1,14 +1,17 @@
-import os
-import openai
-import seaborn as sns
+import re
+import gradio as gr
 from openai import OpenAI
-from dotenv import load_dotenv
+from app.utils import (
+    format_prediction_ouptut,
+    create_input_instruction,
+    sentiment_flow_plot,
+    display_sentiment_score_table,
+    EXAMPLE_CONVERSATIONS,
+)
 
-import re
-import matplotlib.pyplot as plt
+openai_args = {"api_key": ""}
 
-sample_input = \
-"""
+SAMPLE_INPUT = """
 Visitor: Heyyy
 Visitor: How are you this evening
 Agent: better now ;) call me
@@ -51,8 +54,7 @@ Agent: well, i gotta run. watch that video i sent and i will definitely be prayi
     He is good and He hears you. You are loved by Him, brother. Good night
 """
 
-sample_output = \
-"""
+SAMPLE_OUTPUT = """
 Visitor: Heyyy
 [Greeting]
 Visitor: How are you this evening
@@ -128,7 +130,7 @@ Agent: well, i gotta run. watch that video i sent and i will definitely be prayi
     He is good and He hears you. You are loved by Him, brother. Good night
 [Openness]
 
-Sentiment Flow Analysis on the Visitor's side:
+# 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 \
@@ -143,8 +145,42 @@ The Visitor openly takes the Agent's information and the conversation flows smoo
 show openness toward each other.
 """
 
-def get_completion(conversation, model="gpt-4-1106-preview"):
 
+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, \
@@ -182,169 +218,110 @@ def get_completion(conversation, model="gpt-4-1106-preview"):
 
     Here is a sample input delimited by triple backticks
 
-    ```{sample_input}```
+    ```{SAMPLE_INPUT}```
 
     Here is a same output that you should try to aim for delimited by sqaure brackets
 
-    [{sample_output}]
+    [{SAMPLE_OUTPUT}]
     """
-    
-    client = OpenAI()
-    
-    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
-    )
-
-    analysis = response.choices[0].message.content
-
-    def extract_conv_with_labels(analysis):
-        analysis = analysis.replace("\n", " ")
-        BETWEEN_BACKTICKS = "\\`\\`\\`(.*?)\\`\\`\\`"
-        match = re.search(BETWEEN_BACKTICKS, analysis)
-        if match:
-            conv_with_labels = match.group()[4:-4]
-        else:
-            return "OUTPUT IS IN WRONG FORMAT"
-        
-        # just reformatting it for better format
-        conv_with_labels = conv_with_labels.split('] ')
-        temp = [utterance + ']' for utterance in conv_with_labels[:-1]]
-        conv_with_labels = temp + [conv_with_labels[-1]]
-        return conv_with_labels
-
-    grouped_sentiments = {
-        'Acceptance': 3,
-        'Openness': 3,
-        'Interest': 2,
-        'Curiosity': 2,
-        'Informative': 1,
-        'Greeting': 0,
-        'None': 0,
-        'Uninterested': -1,
-        'Anxious': -2,
-        'Confused': -2,
-        'Annoyed': -2,
-        'Remorse': -2,
-        'Disapproval': -3,
-        'Accusatory': -3,
-        'Denial': -3,
-        'Obscene': -3
-    }
-
-
-    def sentiment_flow_plot(conv):
-        conv_with_labels = extract_conv_with_labels(analysis)
-        num_utterances = len(conv_with_labels)
-
-        visitor_Y = [''] * num_utterances
-        agent_Y = [''] * num_utterances
-
-        for i in range(num_utterances):
-            utterance = conv_with_labels[i]
-            match = re.search(r'\[(.*?)\]$', utterance)
-            if match:
-                label = match.group(1)
-            else:
-                print("OUTPUT IS IN WRONG FORMAT")
-                break
-            
-            if utterance.startswith('Visitor'):
-                visitor_Y[i] = label
-                if i == 0:
-                    agent_Y[i] = 'None'
-                else:
-                    agent_Y[i] = agent_Y[i-1]
-            elif utterance.startswith('Agent'):
-                agent_Y[i] = label
-                if i == 0:
-                    visitor_Y[i] = 'None'
-                else:
-                    visitor_Y[i] = visitor_Y[i-1]
-
-        X = range(1,num_utterances+1)
-        visitor_Y_converted = [grouped_sentiments[visitor_Y[i]] for i in range(num_utterances)]
-        agent_Y_converted = [grouped_sentiments[agent_Y[i]] for i in range(num_utterances)]
-
-
-        fig, ax = plt.subplots()
-        sns.set(style="whitegrid")
-
-        ax.plot(X, visitor_Y_converted, label='Visitor', color='blue', marker='o')
-        ax.plot(X, agent_Y_converted, label='Agent', color='green', marker='o')
-
-        plt.legend(loc='upper left', bbox_to_anchor=(1,1))
-        plt.subplots_adjust(right=0.8)
-
-        plt.yticks(ticks=[-3,-2,-1,0,1,2,3])
-        
-        # y_labels = {-3: 'Disapproval/Accusatory/Denial/Obscene', -2: 'Anxious/Confused\nAnnoyed/Remorse', -1: 'Uninterested', 0: 'Greeting/None',
-        #          1: 'Informative', 2: 'Interest/Curiosity', 3: 'Acceptance/Openness'}
-        
-        # cell_text = [[label] for label in y_labels.values()]
-        # plt.table(cellText=cell_text, rowLabels=list(y_labels.keys()), loc='left')
-
-        # plt.tick_params(axis='y', labelsize=10)
-
-        plt.xlabel('Timestamp')
-        plt.ylabel('Sentiment Score')
-        plt.title('Sentiment Flow Plot')
-
-        plt.close(fig)
-
-        return fig
-    
-    fig = sentiment_flow_plot(analysis)
+    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.")
 
-    return response.choices[0].message.content, fig
 
 def set_key(key):
-    os.environ['OPENAI_API_KEY'] = key
-    load_dotenv()
+    openai_args["api_key"] = key
     return
 
-aligned_markdown_table = """
-<div style='text-align: right; font-size: small;'>
-
-| Sentiment Score | Sentiment Label |
-|:---------------:|:---------------:|
-| 3               | Acceptance, Openness |
-| 2               | Interest, Curiosity |
-| 1               | Informative  |
-| 0               | Greeting  |
-| -1              | Uninterested |
-| -2              | Anxious, Confused, Annoyed, Remorse |
-| -3              | Disapproval, Accusatory, Denial, Obscene |
-
-</div>
-"""
-
-import gradio as gr
 
-with gr.Blocks() as gpt_analysis:
-    gr.Markdown("## Conversation Sentiment Analysis Report")
-    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)
-    btn_key = gr.Button(value="Submit Key")
-    btn_key.click(set_key, inputs=api_key)
-    with gr.Row():
-        with gr.Column():
-            conversation = gr.Textbox(label="Input", lines=4)
-        with gr.Column():
-            output_box = gr.Textbox(value="", label="Output",lines=4)
-    btn = gr.Button(value="Submit")
-    with gr.Row():
-        with gr.Column(scale=2):
-            gr.Markdown(aligned_markdown_table)
-        with gr.Column(scale=2):
-            plot_box = gr.Plot(label="Analysis Plot")
-
-        
-    btn.click(get_completion, inputs=conversation, outputs=[output_box, plot_box])
-
-gr.TabbedInterface([gpt_analysis], ["GPT Anlysis"]).launch(inline=False)
+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

app/utils.py

@@ -1,38 +1,34 @@
 import gradio as gr
 import os
-import sys
-
+import re
+import seaborn as sns
+import matplotlib.pyplot as plt
 from preprocessing.preprocess import decode_numeric_label, decode_speaker_role
 
 
 def create_input_instruction():
     gr.Markdown(
         """
-        # Instructions
+        # Predict Sentiment Labels
         Get started by inputting the conversation into the text box below. The
         conversation should be in the format of `<speaker>:<message>` where speaker is
         either Visitor or Agent. There should only be one message per line.
         
         <b>For example:</b><br/>
-        &nbsp;&nbsp;&nbsp;&nbsp;Visitor:Hi! How are you?<br/>
-        &nbsp;&nbsp;&nbsp;&nbsp;Visitor:Are you there?</br>
-        &nbsp;&nbsp;&nbsp;&nbsp;Agent:Hi! I'm good. And you?
+       Visitor:  Hi! How are you?<br/>
+       Visitor:  Are you there?</br>
+       Agent:  Hi! I'm good. And you?
             
         You can also choose from one of the examples below to get a quickstart.
         """
     )
 
 
-def format_prediction_ouptut(numeric_roles, messages, numeric_labels):
+def format_prediction_ouptut(roles, messages, labels):
     output = ""
 
-    for numeric_role, message, numeric_label in zip(
-        numeric_roles, messages, numeric_labels
-    ):
-        label = decode_numeric_label(numeric_label)
-        role = decode_speaker_role(numeric_role)
-
-        output += f"<span style='color:blue;'>[{label}]</span> {role}:{message}\n\n"
+    for role, message, label in zip(roles, messages, labels):
+        output += f"{role}: {message}\n[{label}]\n\n"
 
     return output
 
@@ -54,8 +50,105 @@ def remove_temp_dir(temp_dir):
     os.rmdir(temp_dir)
 
 
+SENTIMENT_GROUP_MAPPING = {
+    "Acceptance": 3,
+    "Openness": 3,
+    "Interest": 2,
+    "Curiosity": 2,
+    "Informative": 1,
+    "Greeting": 0,
+    "None": 0,
+    "Uninterested": -1,
+    "Anxious": -2,
+    "Confused": -2,
+    "Annoyed": -2,
+    "Remorse": -2,
+    "Disapproval": -3,
+    "Accusatory": -3,
+    "Denial": -3,
+    "Obscene": -3,
+}
+
+
+def sentiment_flow_plot(labeled_conv):
+    """Generate the sentiment flow plot using the output from the label predecting
+
+    Args:
+        labeled_conv (str): the displayed output of a sentiment-label prediction.
+
+    Returns:
+        _The sentiment flow plot figure.
+    """
+    msg_pattern = r"(Agent|Visitor): (.*)\n\[(.*)\]"
+    # find the components of each message, including the speaker, message, and label
+    component_lst = re.findall(msg_pattern, labeled_conv)
+
+    visitor_Y = []
+    agent_Y = []
+
+    for speaker, _, label in component_lst:
+        if speaker == "Visitor":
+            visitor_Y.append(label)
+            if agent_Y == []:
+                agent_Y.append("None")
+            else:
+                agent_Y.append(agent_Y[-1])
+        else:
+            agent_Y.append(label)
+            if visitor_Y == []:
+                visitor_Y.append("None")
+            else:
+                visitor_Y.append(visitor_Y[-1])
+
+    X = range(1, len(agent_Y) + 1)
+    visitor_Y_converted = [SENTIMENT_GROUP_MAPPING[visitor_Y[i]] for i in range(len(X))]
+    agent_Y_converted = [SENTIMENT_GROUP_MAPPING[agent_Y[i]] for i in range(len(X))]
+
+    fig, ax = plt.subplots()
+    sns.set(style="whitegrid")
+
+    ax.plot(X, visitor_Y_converted, label="Visitor", color="blue", marker="o")
+    ax.plot(X, agent_Y_converted, label="Agent", color="green", marker="o")
+
+    plt.legend(loc="upper left", bbox_to_anchor=(1, 1))
+    plt.subplots_adjust(right=0.8)
+
+    plt.yticks(ticks=[-3, -2, -1, 0, 1, 2, 3])
+
+    plt.xlabel("Timestamp")
+    plt.ylabel("Sentiment Score")
+    plt.title("Sentiment Flow Plot")
+    plt.close(fig)
+
+    return fig
+
+
+SENTIMENT_SCORE_TABLE = """
+<div align="center" style='font-size: small;'>
+
+|  Sentiment Score  |  Sentiment Label  |
+| :---------------: | :---------------: |
+|         3         | Acceptance, Openness |
+|         2         | Interest, Curiosity |
+|         1         | Informative  |
+|         0         | Greeting  |
+|        -1         | Uninterested |
+|        -2         | Anxious, Confused, Annoyed, Remorse |
+|        -3         | Disapproval, Accusatory, Denial, Obscene |
+
+</div>
+"""
+
+
+def display_sentiment_score_table():
+    gr.Markdown(
+        "After submit the conversation and run the model to predict the sentiment labels from the above, we can generate the plot of the sentimen labels found in this conversation to see how the sentiment flow changes over time.\nThe sentiment labels are grouped into corresponding scores below:"
+    )
+    gr.Markdown(SENTIMENT_SCORE_TABLE)
+
+
 EXAMPLE_CONVERSATIONS = {
-    "Example 1": "Visitor:Hey Jess, available for outcall tonight?\n\nAgent:hey hey what you looking for luv?\n\nVisitor:Some company in El Segundo\n\nAgent:out with friends or just the two of us?\n\nVisitor:Just the two\n\nAgent:looking for some connection then...\n\nVisitor:Yes\n\nAgent:I get it. Looking for companionship, it's a very legit need. I'll level with you. We're a group of guys trying to show other men the harms of the sex trade industry, both to buyers and sellers. You up for a little conversation? No pressure... no judgement.\n\nVisitor:No, but interested in the harms in companionship?\n\nAgent:The truth is nearly all of the women on these sites are not in this career by choice. We work with survivors. Their money goes to pimps.. their lives are controlled.\n\nVisitor:Really?\n\nVisitor:Even when they say \"independent\"?\n\nAgent:Yes. It's all part of the human trafficking system... pimps find & recruit nearly all women who are doing this, they are very, very good at manipulation and control, and working in the background. Sadly, this modern day slavery will be with us as long as there are willing buyers. That's why we are trying to work this angle. The more men that know the truth, the fewer will want to exploit someone in this way.\n\nAgent:I need to move on, but just consider what I've told you tonight. Polarisproject.org is a great resource to learn more... it's nothing like what Hollywood shows ya. We work with the women & some guys who have escaped out of this lifestyle. It is modern-day slavery, straight up. Thanks for the chat. Good night!\n\nVisitor:Thanks, fyi it worked. Hung it up, I'm to old to me messin around in this crazy stuff take care\n\nAgent:I am so glad to hear this friend. You calling this number tonight was not just a coincidence. Take care also!!\n\n",
-    "Example 2": "Visitor:What's up sexy\n\nAgent:Thank you! I'm hangin\n\nVisitor:Are you available at the moment\n\nAgent:what's your name\n\nVisitor:David ... Yours\n\nAgent:Dakota is not available right now, but my name is Jeff and I'm working with a bunch of guys trying to curb the demand for victims of human trafficking. Did you ever consider that Dakota could be a victim of trafficking?\n\nVisitor:Fuck no I just wanted to get some pussy .... Bye\n\nAgent:Just so you know your number is in our system now. Understand that next time you click on one of these ads you could be walking into an undercover cop and many of these girls are underage. That is a crime that will follow you the rest of your life!\n\nVisitor:. You think the predators your looking for going to fall into your lap? Lol your not a cop stop trying to be one ... You want to help?\n\nVisitor:Just so you know I don't give no fucks about you or no fucking cops... Threatening me will not get you shit from me ,no info no nothing you stupid cunt..\n\nAgent:Hey man, I did not want to threaten you. I want you to be aware of the risks. Have you heard that most women in prostitution do it against their will, and many are minors?\n\nVisitor:I never fuck with minors that's for sure cause you can tell if they are not of age\n\nAgent:Yeah, many people think it's harmless and these women are in complete control. But, the truth is, most of these women are enslaved, forced to sell themselves by pimps who take 100% of their money. I hope you can see these women in a different light.\n\nVisitor:I know most are blacks living off a woman cause they can't get no job ... I won't see a female if she has a guy around\n\nAgent:I'm curious, is there something missing inside of you, or are you angry about something not right in your life that causes you to seek out connection with these girls?\n\nVisitor:No actually I am on gps monitor and can't go out and meet no one so I got needs and feel sometimes hooking up with one of these girls here and there is w\n\nVisitor:hat works for my situation....but I'm not disrespectful to them or be doing ugly things ... My ex wife cheated on me after 8yrs of marriage I don't reall\n\nVisitor:y trust women like enough for a relationship\n\nVisitor:I'm being honest with you cause even tho I don't know u I believe in doing the right thing and your movement with alot of work is going to save lives so\n\nVisitor:that's good\n\nAgent:Hey David, I'm very sorry for your situation and the pain you must be in. Were you raised with any faith perspective?\n\nVisitor:Yes I actually was baptized for the first time in my life in 2018 and have made some drastic changes to my life.... I'm not perfect but I am working hard\n\nAgent:Okay, David. That's great news. We're all a work in progress. God is faithful to forgive us of our sins if we just ask him. Read 1 John 1:9. Can I direct you to some resources to help you overcome this urge to buy sex?\n\nVisitor:Honestly I'm not ready for that is all I can say at the moment.... Sorry but I don't like to lie\n\nAgent:Hey, no worries. When you're ready, check out www.celebraterecovery.com as they have a lot of resources that might help. I will be praying for you, David. I've had a lot of pain in my life and only in my relationship with Jesus have I found true peace that surpasses my understanding. I pray that for you too.\n\nVisitor:What kinda pain ? Do you mind sharing?\n\nVisitor:And what are y'all doing to put those pieces of shit in prison who are hurting these women\n\nAgent:I've had several people close to me pass away tragically in the past few years. Thanks for asking. I'm just trusting God each day trusting he's in control.\n\nAgent:Hey, honestly, the best way we can solve this problem of evil pieces of shit harming these girls is to dry up the demand for these girls by encouraging men like us to stop calling these ads. If the demand dries up, evil predatory scums' business with die.\n\nAgent:Can I get your commitment to do your part and stop engaging these girls on their escort ads?\n\nVisitor:Yes I will but I can promise you that it will not stop matter of fact it will get worse ... They will beat them more and make them do it more only way is\n\nVisitor:to put the pimps in prison for alot of years....\n\nAgent:Hey, I'll be praying for you. Feel free to call me any time if you need to talk. My number is (615) 628-7545. God bless you, brother.\n\nVisitor:Thank u\n\n",
-    "Example 3": "Visitor:Heyyy\n\nVisitor:How are you this evening\n\nAgent:better now ;) call me\n\nVisitor:Iam at work for now ,be off around 10pm\n\nVisitor:Need some company\n\nVisitor:Are you independent honey\n\nAgent: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?\n\nAgent:we are with some guys who are trying to spread awareness of the realities of this \"industry\".\n\nAgent:https://exoduscry.com/choice/\n\nVisitor:Thanks\n\nAgent: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\n\nVisitor:Ooohhh\n\nAgent:selling their body to make ends meet or appease a pimp\n\nVisitor:That's really awful\n\nAgent: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?\n\nVisitor:Well iam just looking for attention\n\nVisitor:My marriage is not going well lol\n\nAgent: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\n\nAgent:hey, thanks for sharing that my man. i know it can be hard\n\nAgent:marraige is the most humbling of relationships, isnt it?\n\nVisitor:She leaves with her friends n no time for me\n\nAgent:ive been there my guy. i know that it is alot easier to numb that loneliness for sure\n\nVisitor:I want to be faithful\n\nAgent:does your wife know how you feel when she chooses her friends instead of you?\n\nVisitor:I been drinking lately\n\nVisitor:Yes , she takes pills\n\nAgent:if so, i hope you are praying for her to realize the hurt she is causing and to seek change\n\nVisitor:She had surgery 4 yes ago n it's been hard for her n her addiction on pills\n\nVisitor:Yes for now iam looking for a female friend to talk n see what can we do for each other\n\nAgent:that is hard my man. physical pain is a huge obstacle in life for sure so i hear you\n\nVisitor:Well chat later .thanks\n\nAgent:have you considered pursuing other men who can encourage you instead of looking for the easy way out?\n\nAgent:what is your name my friend? i will be praying for you by name if you wouldnt mind sharing it\n\nAgent: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\n\n",
+    "Example 1": "Visitor: Hey Jess, available for outcall tonight?\n\nAgent: hey hey what you looking for luv?\n\nVisitor: Some company in El Segundo\n\nAgent: out with friends or just the two of us?\n\nVisitor: Just the two\n\nAgent: looking for some connection then...\n\nVisitor: Yes\n\nAgent: I get it. Looking for companionship, it's a very legit need. I'll level with you. We're a group of guys trying to show other men the harms of the sex trade industry, both to buyers and sellers. You up for a little conversation? No pressure... no judgement.\n\nVisitor: No, but interested in the harms in companionship?\n\nAgent: The truth is nearly all of the women on these sites are not in this career by choice. We work with survivors. Their money goes to pimps.. their lives are controlled.\n\nVisitor: Really?\n\nVisitor: Even when they say \"independent\"?\n\nAgent: Yes. It's all part of the human trafficking system... pimps find & recruit nearly all women who are doing this, they are very, very good at manipulation and control, and working in the background. Sadly, this modern day slavery will be with us as long as there are willing buyers. That's why we are trying to work this angle. The more men that know the truth, the fewer will want to exploit someone in this way.\n\nAgent: I need to move on, but just consider what I've told you tonight. Polarisproject.org is a great resource to learn more... it's nothing like what Hollywood shows ya. We work with the women & some guys who have escaped out of this lifestyle. It is modern-day slavery, straight up. Thanks for the chat. Good night!\n\nVisitor: Thanks, fyi it worked. Hung it up, I'm to old to me messin around in this crazy stuff take care\n\nAgent: I am so glad to hear this friend. You calling this number tonight was not just a coincidence. Take care also!!\n\n",
+    "Example 2": "Visitor: What's up sexy\n\nAgent: Thank you! I'm hangin\n\nVisitor: Are you available at the moment\n\nAgent: what's your name\n\nVisitor: David ... Yours\n\nAgent: Dakota is not available right now, but my name is Jeff and I'm working with a bunch of guys trying to curb the demand for victims of human trafficking. Did you ever consider that Dakota could be a victim of trafficking?\n\nVisitor: Fuck no I just wanted to get some pussy .... Bye\n\nAgent: Just so you know your number is in our system now. Understand that next time you click on one of these ads you could be walking into an undercover cop and many of these girls are underage. That is a crime that will follow you the rest of your life!\n\nVisitor: . You think the predators your looking for going to fall into your lap? Lol your not a cop stop trying to be one ... You want to help?\n\nVisitor: Just so you know I don't give no fucks about you or no fucking cops... Threatening me will not get you shit from me ,no info no nothing you stupid cunt..\n\nAgent: Hey man, I did not want to threaten you. I want you to be aware of the risks. Have you heard that most women in prostitution do it against their will, and many are minors?\n\nVisitor: I never fuck with minors that's for sure cause you can tell if they are not of age\n\nAgent: Yeah, many people think it's harmless and these women are in complete control. But, the truth is, most of these women are enslaved, forced to sell themselves by pimps who take 100% of their money. I hope you can see these women in a different light.\n\nVisitor: I know most are blacks living off a woman cause they can't get no job ... I won't see a female if she has a guy around\n\nAgent: I'm curious, is there something missing inside of you, or are you angry about something not right in your life that causes you to seek out connection with these girls?\n\nVisitor: No actually I am on gps monitor and can't go out and meet no one so I got needs and feel sometimes hooking up with one of these girls here and there is w\n\nVisitor: hat works for my situation....but I'm not disrespectful to them or be doing ugly things ... My ex wife cheated on me after 8yrs of marriage I don't reall\n\nVisitor: y trust women like enough for a relationship\n\nVisitor: I'm being honest with you cause even tho I don't know u I believe in doing the right thing and your movement with alot of work is going to save lives so\n\nVisitor: that's good\n\nAgent: Hey David, I'm very sorry for your situation and the pain you must be in. Were you raised with any faith perspective?\n\nVisitor: Yes I actually was baptized for the first time in my life in 2018 and have made some drastic changes to my life.... I'm not perfect but I am working hard\n\nAgent: Okay, David. That's great news. We're all a work in progress. God is faithful to forgive us of our sins if we just ask him. Read 1 John 1:9. Can I direct you to some resources to help you overcome this urge to buy sex?\n\nVisitor: Honestly I'm not ready for that is all I can say at the moment.... Sorry but I don't like to lie\n\nAgent: Hey, no worries. When you're ready, check out www.celebraterecovery.com as they have a lot of resources that might help. I will be praying for you, David. I've had a lot of pain in my life and only in my relationship with Jesus have I found true peace that surpasses my understanding. I pray that for you too.\n\nVisitor: What kinda pain ? Do you mind sharing?\n\nVisitor: And what are y'all doing to put those pieces of shit in prison who are hurting these women\n\nAgent: I've had several people close to me pass away tragically in the past few years. Thanks for asking. I'm just trusting God each day trusting he's in control.\n\nAgent: Hey, honestly, the best way we can solve this problem of evil pieces of shit harming these girls is to dry up the demand for these girls by encouraging men like us to stop calling these ads. If the demand dries up, evil predatory scums' business with die.\n\nAgent: Can I get your commitment to do your part and stop engaging these girls on their escort ads?\n\nVisitor: Yes I will but I can promise you that it will not stop matter of fact it will get worse ... They will beat them more and make them do it more only way is\n\nVisitor: to put the pimps in prison for alot of years....\n\nAgent: Hey, I'll be praying for you. Feel free to call me any time if you need to talk. My number is (615) 628-7545. God bless you, brother.\n\nVisitor: Thank u\n\n",
+    "Example 3": "Visitor: Heyyy\n\nVisitor: How are you this evening\n\nAgent: better now ;) call me\n\nVisitor: Iam at work for now ,be off around 10pm\n\nVisitor: Need some company\n\nVisitor: Are you independent honey\n\nAgent: 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?\n\nAgent: we are with some guys who are trying to spread awareness of the realities of this \"industry\".\n\nAgent: https://exoduscry.com/choice/\n\nVisitor: Thanks\n\nAgent: 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\n\nVisitor: Ooohhh\n\nAgent: selling their body to make ends meet or appease a pimp\n\nVisitor: That's really awful\n\nAgent: 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?\n\nVisitor: Well iam just looking for attention\n\nVisitor: My marriage is not going well lol\n\nAgent: 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\n\nAgent: hey, thanks for sharing that my man. i know it can be hard\n\nAgent: marraige is the most humbling of relationships, isnt it?\n\nVisitor: She leaves with her friends n no time for me\n\nAgent: ive been there my guy. i know that it is alot easier to numb that loneliness for sure\n\nVisitor: I want to be faithful\n\nAgent: does your wife know how you feel when she chooses her friends instead of you?\n\nVisitor: I been drinking lately\n\nVisitor: Yes , she takes pills\n\nAgent: if so, i hope you are praying for her to realize the hurt she is causing and to seek change\n\nVisitor: She had surgery 4 yes ago n it's been hard for her n her addiction on pills\n\nVisitor: Yes for now iam looking for a female friend to talk n see what can we do for each other\n\nAgent: that is hard my man. physical pain is a huge obstacle in life for sure so i hear you\n\nVisitor: Well chat later .thanks\n\nAgent: have you considered pursuing other men who can encourage you instead of looking for the easy way out?\n\nAgent: what is your name my friend? i will be praying for you by name if you wouldnt mind sharing it\n\nAgent: 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\n\n",
 }

main.py

@@ -4,10 +4,11 @@
 #
 
 import gradio as gr
-from views import cosmic_view
+from app import cosmic_view, gpt_view
 
 cosmic_model = cosmic_view.cosmic_ui()
-demo = gr.TabbedInterface([cosmic_model], ["COSMIC"])
+gpt_model = gpt_view.gpt_ui()
+demo = gr.TabbedInterface([cosmic_model, gpt_model], ["COSMIC", "GPT Analysis"])
 
 if __name__ == "__main__":
     demo.launch()