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()