Spaces:
Running
Running
File size: 3,160 Bytes
22f7204 aa3742a 22f7204 aa3742a 22f7204 |
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 |
# Copyright 2023 by Jan Philip Wahle, https://jpwahle.com/
# All rights reserved.
import os
import random
import time
import gradio as gr
import openai
openai.api_key = os.environ.get("OPENAI_API_KEY")
def create_prompt(sentence, paraphrase_type):
prompt = {
"messages": [
{
"role": "user",
"content": f"Given the following sentence, generate a paraphrase with the following types. Sentence: {sentence}. Paraphrase Types: {paraphrase_type}"
}
]
}
return prompt
paraphrase_types = [
"Derivational Changes",
"Inflectional Changes",
"Modal Verb Changes",
"Spelling changes",
"Change of format",
"Same Polarity Substitution (contextual)",
"Same Polarity Substitution (habitual)",
"Same Polarity Substitution (named ent.)",
"Converse substitution",
"Opposite polarity substitution (contextual)",
"Opposite polarity substitution (habitual)",
"Synthetic/analytic substitution",
"Coordination changes",
"Diathesis alternation",
"Ellipsis",
"Negation switching",
"Subordination and nesting changes",
"Direct/indirect style alternations",
"Punctuation changes",
"Syntax/discourse structure changes",
"Entailment",
"Identity",
"Non-paraphrase",
"Addition/Deletion",
"Change of order",
"Semantic-based"
]
with gr.Blocks() as demo:
description = gr.Markdown("""
## Paraphrase Type Generator
This demo uses a fine-tuned ChatGPT-3.5 model to generate paraphrases given specific paraphrase types.
**How to use:**
1. Select one or many type of paraphrase from the dropdown menu.
2. Enter a sentence in the text box.
3. Click the "Submit" button or hit enter.
4. The application will generate a paraphrase of the input sentence based on the selected type.
""")
chatbot = gr.Chatbot()
types = gr.Dropdown(
paraphrase_types,
value="Syntax-based changes",
multiselect=True,
allow_custom_value=True,
)
msg = gr.Textbox()
submit = gr.Button("Submit")
clear = gr.Button("Clear")
def user(user_message, history):
return "", history + [[user_message, None]]
def generate_paraphrase(user_message, paraphrase_type, history):
history[-1][1] = ""
prompt = create_prompt(history[-1][0], paraphrase_type)
bot_message = openai.ChatCompletion.create(
model="ft:gpt-3.5-turbo-0613:personal::7xbU0xQ2",
messages=prompt["messages"],
)
history[-1][1] = ""
for character in bot_message.choices[0].message.content:
history[-1][1] += character
time.sleep(0.01)
yield history
msg.submit(user, [msg, chatbot], [msg, chatbot], queue=False).then(
generate_paraphrase, [msg, types, chatbot], chatbot
)
submit.click(user, [msg, chatbot], [msg, chatbot], queue=False).then(
generate_paraphrase, [msg, types, chatbot], chatbot
)
clear.click(lambda: None, None, chatbot, queue=False)
demo.queue()
demo.launch() |