ai-detector / app.py
mia37938's picture
Update app.py
7700af6 verified
import gradio as gr
import requests
API_URL = "https://api-inference.huggingface.co/models/Hello-SimpleAI/chatgpt-detector-roberta"
headers = {"Content-Type": "application/json"}
def detect_ai(text):
try:
response = requests.post(API_URL, headers=headers, json={"inputs": text}, timeout=15)
result = response.json()
return str(result)
except Exception as e:
return f"Error: {str(e)}"
def rewrite_text(text):
# Simple humanising rewrite
replacements = {
"important": "crucial", "use": "utilize", "make": "create",
"show": "illustrate", "get": "obtain", "say": "express"
}
for word, replacement in replacements.items():
text = text.replace(word, replacement)
return text
def teacher_feedback_prompt(text):
return f"You are a teacher. Give feedback on this writing:\n\n{text}"
with gr.Blocks() as demo:
gr.Markdown("## AI Detector + Humaniser + Feedback Tool")
input_text = gr.Textbox(lines=10, label="Enter Text")
with gr.Row():
detect_btn = gr.Button("Detect AI")
rewrite_btn = gr.Button("Humanise Text")
feedback_btn = gr.Button("Generate Feedback Prompt")
detect_output = gr.Textbox(label="Detection Result")
rewritten_output = gr.Textbox(label="Humanised Text")
feedback_output = gr.Textbox(label="Teacher Feedback Prompt")
detect_btn.click(detect_ai, input_text, detect_output)
rewrite_btn.click(rewrite_text, input_text, rewritten_output)
feedback_btn.click(teacher_feedback_prompt, rewritten_output, feedback_output)
demo.launch()