Create app.py

app.py

@@ -0,0 +1,61 @@
+import gradio as gr
+import torch
+import transformers
+from synthid_text import detector_bayesian, synthid_mixin, logits_processing
+
+# Configurations and model selection
+MODEL_NAME = "google/gemma-7b-it"  # Choose the model (Gemma models used in SynthID)
+DEVICE = torch.device('cuda:0') if torch.cuda.is_available() else torch.device('cpu')
+
+# Initialize tokenizer and model configuration
+tokenizer = transformers.AutoTokenizer.from_pretrained(MODEL_NAME)
+tokenizer.pad_token = tokenizer.eos_token
+tokenizer.padding_side = "left"
+
+# Watermarking configuration
+CONFIG = synthid_mixin.DEFAULT_WATERMARKING_CONFIG
+
+# Logits processor for SynthID
+logits_processor = logits_processing.SynthIDLogitsProcessor(
+    **CONFIG, top_k=40, temperature=0.5
+)
+
+# Function to check for AI-generated content using SynthID
+def check_plagiarism(text):
+    # Tokenize and process the input text
+    tokens = tokenizer.encode_plus(text, return_tensors="pt", truncation=True, padding=True)
+    tokens = tokens.to(DEVICE)
+    
+    # Use SynthID's bayesian detector to check for AI generation likelihood
+    try:
+        is_ai_generated = detector_bayesian.is_generated(tokens['input_ids'], logits_processor, threshold=0.5)
+        if is_ai_generated:
+            return "Flagged as AI-generated content (Academic Integrity Warning)."
+        else:
+            return "Content appears to be human-generated."
+    except Exception as e:
+        return f"Error in detection process: {e}"
+
+# Define the Gradio interface
+def create_plagiarism_checker():
+    with gr.Blocks() as app:
+        gr.Markdown("## Plagiarism and Academic Integrity Checker")
+        gr.Markdown("Paste your text below to check if it is AI-generated using SynthID detection.")
+
+        # Input textbox for users to paste text
+        text_input = gr.Textbox(placeholder="Paste your text here...", label="Input Text", lines=10)
+
+        # Output box to display the result
+        output = gr.Textbox(label="Integrity Check Result", interactive=False)
+        
+        # Button to initiate the check
+        check_button = gr.Button("Check Text")
+        
+        # Define the click event for the button
+        check_button.click(fn=check_plagiarism, inputs=text_input, outputs=output)
+
+    return app
+
+# Launch the app
+plagiarism_checker_app = create_plagiarism_checker()
+plagiarism_checker_app.launch()