import openai
def classify_website(website):
prompt = "Website classification:\n\nWebsite: {} \nCategory:".format(website)
model_response = openai.Completion.create(
model="gpt-3.5-turbo",
prompt=prompt,
max_tokens=1,
temperature=0.2,
n=1,
stop=None,
temperature=0.2
)
category = model_response.choices[0].text.strip()
return category
def determine_piracy_type(website):
prompt = "Piracy type detection:\n\nWebsite: {} \nType:".format(website)
model_response = openai.Completion.create(
model="gpt-3.5-turbo",
prompt=prompt,
max_tokens=1,
temperature=0.2,
n=1,
stop=None,
temperature=0.2
)
piracy_type = model_response.choices[0].text.strip()
return piracy_type
Set up OpenAI API credentials
openai.api_key = "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"
Get website list input from the user
websites = input("Enter a comma-separated list of websites to classify: ").split(",")
Classify each website
for website in websites:
website = website.strip()
classification = classify_website(website)
if classification == "Piracy":
piracy_type = determine_piracy_type(website)
print("The website {} is classified as: Piracy - {}".format(website, piracy_type))
else:
print("The website {} is classified as: Legal".format(website))
