Spaces:
Running
Running
import gradio as gr | |
import numpy as np | |
from datetime import date | |
from predictors import predict_bc_scores, predict_mc_scores, predict_1on1_scores | |
from analysis import depth_analysis | |
from predictors import predict_quillbot | |
from plagiarism import plagiarism_check, build_date | |
from highlighter import analyze_and_highlight | |
from utils import extract_text_from_pdf, len_validator | |
import yaml | |
from functools import partial | |
np.set_printoptions(suppress=True) | |
with open("config.yaml", "r") as file: | |
params = yaml.safe_load(file) | |
model_list = params["MC_OUTPUT_LABELS"] | |
analyze_and_highlight_bc = partial(analyze_and_highlight, model_type="bc") | |
analyze_and_highlight_quillbot = partial(analyze_and_highlight, model_type="quillbot") | |
def ai_generated_test(option, input, models): | |
if option == "Human vs AI": | |
return predict_bc_scores(input), None | |
elif option == "Human vs AI Source Models": | |
return predict_bc_scores(input), predict_1on1_scores(input, models) | |
return None, None | |
# COMBINED | |
def main( | |
ai_option, | |
plag_option, | |
input, | |
models, | |
year_from, | |
month_from, | |
day_from, | |
year_to, | |
month_to, | |
day_to, | |
domains_to_skip, | |
): | |
formatted_tokens = plagiarism_check( | |
plag_option, | |
input, | |
year_from, | |
month_from, | |
day_from, | |
year_to, | |
month_to, | |
day_to, | |
domains_to_skip, | |
) | |
depth_analysis_plot = depth_analysis(input) | |
bc_score = predict_bc_scores(input) | |
mc_score = predict_1on1_scores(input, models) | |
quilscore = predict_quillbot(input) | |
return ( | |
bc_score, | |
mc_score, | |
formatted_tokens, | |
depth_analysis_plot, | |
quilscore, | |
) | |
# START OF GRADIO | |
title = "Copyright Checker" | |
months = { | |
"January": "01", | |
"February": "02", | |
"March": "03", | |
"April": "04", | |
"May": "05", | |
"June": "06", | |
"July": "07", | |
"August": "08", | |
"September": "09", | |
"October": "10", | |
"November": "11", | |
"December": "12", | |
} | |
with gr.Blocks() as demo: | |
today = date.today() | |
# dd/mm/YY | |
d1 = today.strftime("%d/%B/%Y") | |
d1 = d1.split("/") | |
domain_list = ["com", "org", "net", "int", "edu", "gov", "mil"] | |
gr.Markdown( | |
""" | |
# Copyright Checker | |
""" | |
) | |
with gr.Row(): | |
input_text = gr.Textbox(label="Input text", lines=6, placeholder="") | |
file_input = gr.File(label="Upload PDF") | |
file_input.change( | |
fn=extract_text_from_pdf, inputs=file_input, outputs=input_text | |
) | |
char_count = gr.Textbox(label="Minumum Character Limit Check") | |
input_text.change(fn=len_validator, inputs=input_text, outputs=char_count) | |
with gr.Row(): | |
models = gr.Dropdown( | |
model_list, | |
value=model_list, | |
multiselect=True, | |
label="Models to test against", | |
) | |
with gr.Row(): | |
with gr.Column(): | |
ai_option = gr.Radio( | |
[ | |
"Human vs AI", | |
"Human vs AI Source Models", | |
# "Human vs AI Source Models (1 on 1)", | |
], | |
label="Choose an option please.", | |
) | |
with gr.Column(): | |
plag_option = gr.Radio( | |
["Standard", "Advanced"], label="Choose an option please." | |
) | |
with gr.Row(): | |
with gr.Column(): | |
only_ai_btn = gr.Button("AI Check") | |
with gr.Column(): | |
only_plagiarism_btn = gr.Button("Source Check") | |
with gr.Column(): | |
quillbot_check = gr.Button("Humanized Text Check") | |
with gr.Row(): | |
with gr.Column(): | |
bc_highlighter_button = gr.Button("Human vs. AI Highlighter") | |
with gr.Column(): | |
quillbot_highlighter_button = gr.Button("Humanized Highlighter") | |
with gr.Row(): | |
depth_analysis_btn = gr.Button("Detailed Writing Analysis") | |
with gr.Row(): | |
full_check_btn = gr.Button("Full Check") | |
gr.Markdown( | |
""" | |
## Output | |
""" | |
) | |
with gr.Row(): | |
with gr.Column(): | |
bcLabel = gr.Label(label="Source") | |
with gr.Column(): | |
mcLabel = gr.Label(label="Creator") | |
with gr.Row(): | |
with gr.Column(): | |
bc_highlighter_output = gr.HTML(label="Human vs. AI Highlighter") | |
# with gr.Column(): | |
# mc1on1Label = gr.Label(label="Creator(1 on 1 Approach)") | |
with gr.Row(): | |
with gr.Column(): | |
QLabel = gr.Label(label="Humanized") | |
with gr.Row(): | |
quillbot_highlighter_output = gr.HTML(label="Humanized Highlighter") | |
with gr.Group(): | |
with gr.Row(): | |
month_from = gr.Dropdown( | |
choices=months, | |
label="From Month", | |
value="January", | |
interactive=True, | |
) | |
day_from = gr.Textbox(label="From Day", value="01") | |
year_from = gr.Textbox(label="From Year", value="2000") | |
# from_date_button = gr.Button("Submit") | |
with gr.Row(): | |
month_to = gr.Dropdown( | |
choices=months, | |
label="To Month", | |
value=d1[1], | |
interactive=True, | |
) | |
day_to = gr.Textbox(label="To Day", value=d1[0]) | |
year_to = gr.Textbox(label="To Year", value=d1[2]) | |
# to_date_button = gr.Button("Submit") | |
with gr.Row(): | |
domains_to_skip = gr.Dropdown( | |
domain_list, | |
multiselect=True, | |
label="Domain To Skip", | |
) | |
with gr.Row(): | |
with gr.Column(): | |
sentenceBreakdown = gr.HighlightedText( | |
label="Source Detection Sentence Breakdown", | |
combine_adjacent=True, | |
color_map={ | |
"[1]": "red", | |
"[2]": "orange", | |
"[3]": "yellow", | |
"[4]": "green", | |
}, | |
) | |
with gr.Row(): | |
with gr.Column(): | |
writing_analysis_plot = gr.Plot(label="Writing Analysis Plot") | |
full_check_btn.click( | |
fn=main, | |
inputs=[ | |
ai_option, | |
plag_option, | |
input_text, | |
models, | |
year_from, | |
month_from, | |
day_from, | |
year_to, | |
month_to, | |
day_to, | |
domains_to_skip, | |
], | |
outputs=[ | |
bcLabel, | |
mcLabel, | |
# mc1on1Label, | |
sentenceBreakdown, | |
writing_analysis_plot, | |
QLabel, | |
], | |
api_name="main", | |
) | |
only_ai_btn.click( | |
fn=ai_generated_test, | |
inputs=[ai_option, input_text, models], | |
# outputs=[bcLabel, mcLabel, mc1on1Label], | |
outputs=[bcLabel, mcLabel], | |
api_name="ai_check", | |
) | |
quillbot_check.click( | |
fn=predict_quillbot, | |
inputs=[input_text], | |
outputs=[QLabel], | |
api_name="quillbot_check", | |
) | |
only_plagiarism_btn.click( | |
fn=plagiarism_check, | |
inputs=[ | |
plag_option, | |
input_text, | |
year_from, | |
month_from, | |
day_from, | |
year_to, | |
month_to, | |
day_to, | |
domains_to_skip, | |
], | |
outputs=[ | |
sentenceBreakdown, | |
], | |
api_name="plagiarism_check", | |
) | |
depth_analysis_btn.click( | |
fn=depth_analysis, | |
inputs=[input_text], | |
outputs=[writing_analysis_plot], | |
api_name="depth_analysis", | |
) | |
quillbot_highlighter_button.click( | |
fn=analyze_and_highlight_quillbot, | |
inputs=[input_text], | |
outputs=[quillbot_highlighter_output], | |
api_name="humanized_highlighter", | |
) | |
bc_highlighter_button.click( | |
fn=analyze_and_highlight_bc, | |
inputs=[input_text], | |
outputs=[bc_highlighter_output], | |
api_name="bc_highlighter", | |
) | |
date_from = "" | |
date_to = "" | |
if __name__ == "__main__": | |
demo.launch( | |
share=True, | |
auth=("polygraf-admin", "test@aisd"), | |
allowed_paths=[".", "images"], | |
) | |