Update app.py

app.py

@@ -1,53 +1,46 @@
 import gradio as gr
 import requests
 
-API_URL = "https://api-inference.huggingface.co/models/robinhad/roberta-base-openai-detector"
+API_URL = "https://api-inference.huggingface.co/models/Hello-SimpleAI/chatgpt-detector-roberta"
+headers = {"Content-Type": "application/json"}
 
 def detect_ai(text):
-    headers = {"Content-Type": "application/json"}
-    payload = {"inputs": text}
-    response = requests.post(API_URL, headers=headers, json=payload)
     try:
-        output = response.json()
-        ai_score = 0
-        human_score = 0
-        for item in output:
-            if "ai" in item["label"].lower():
-                ai_score = item["score"]
-            if "human" in item["label"].lower():
-                human_score = item["score"]
-        result = f"AI confidence: {ai_score*100:.2f}%\nHuman confidence: {human_score*100:.2f}%"
-        result += "\n\nLikely AI" if ai_score > human_score else "\n\nLikely Human"
-        return result
-    except:
-        return "Error: Could not interpret model output."
+        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 rewriting logic
+    # Simple humanising rewrite
     replacements = {
-        "important": "crucial", "use": "utilize", "show": "demonstrate",
-        "get": "obtain", "make": "create", "say": "express"
+        "important": "crucial", "use": "utilize", "make": "create",
+        "show": "illustrate", "get": "obtain", "say": "express"
     }
-    for key, value in replacements.items():
-        text = text.replace(key, value)
+    for word, replacement in replacements.items():
+        text = text.replace(word, replacement)
     return text
 
-def feedback_prompt(text):
-    return f"You are a teacher. Please give feedback on the following writing:\n\n{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 + Rewriter + Feedback Prompt Generator")
-    text_input = gr.Textbox(lines=10, label="Enter Text")
-    detect_btn = gr.Button("Detect AI")
-    rewrite_btn = gr.Button("Rewrite Text")
-    feedback_btn = gr.Button("Generate Feedback Prompt")
-
-    ai_output = gr.Textbox(label="Detection Result")
-    rewritten = gr.Textbox(label="Rewritten Text")
-    feedback = gr.Textbox(label="Teacher Prompt")
-
-    detect_btn.click(detect_ai, text_input, ai_output)
-    rewrite_btn.click(rewrite_text, text_input, rewritten)
-    feedback_btn.click(feedback_prompt, rewritten, feedback)
+    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()