Update ai_text_detector_valid_final.py

ai_text_detector_valid_final.py

@@ -5,16 +5,18 @@ import requests
 import numpy as np
 
 # Hugging Face Token
-HF_TOKEN = os.getenv("HF_TOKEN")  # export HF_TOKEN="your_token" before running
+HF_TOKEN = os.getenv("HF_TOKEN")  # Hugging Face token (optional if space is public)
+SZEGEDAI_URL = "https://hf.space/embed/SzegedAI/AI_Detector/api/predict/"
+HEADERS = {"Authorization": f"Bearer {HF_TOKEN}"} if HF_TOKEN else {}
 
 # Headers for API
 headers = {"Authorization": f"Bearer {HF_TOKEN}"} if HF_TOKEN else {}
 
 # Multiple AI text detection models
 MODELS = {
-    "DeBERTa Detector": "microsoft/deberta-v3-base",
-    "MonkeyDAnh": "MonkeyDAnh/deberta-v3-base-finetuned-ai-human-detector",
-    "Andreas122001": "Andreas122001/roberta-base-openai-detector"
+    "DeBERTa Detector": "distilbert-base-uncased-finetuned-sst-2-english",
+    "MonkeyDAnh":"MonkeyDAnh/my-awesome-ai-detector-roberta-base-v4-human-vs-machine-finetune",
+    "Andreas122001":"andreas122001/roberta-academic-detector"
     # SzegedAI handled separately since it's a Space
 }
 
@@ -30,22 +32,31 @@ def run_hf_model(model_id, text):
         return {"Human Probability": float(probs[0]*100), "AI Probability": float(probs[1]*100)}
     except Exception as e:
         return {"error": str(e)}
-
-def run_szegedai(text):
-    """Call the SzegedAI Space API"""
+def szegedai_predict(text):
     try:
-        response = requests.post(
-            "https://huggingface.co/spaces/SzegedAI/AI_Detector/run/predict",
-            headers=headers,
-            json={"data": [text]},
-            timeout=30
-        )
+        payload = {"data": [text]}
+        response = requests.post(SZEGEDAI_URL, json=payload, headers=HEADERS, timeout=30)
         response.raise_for_status()
         result = response.json()
-        return result  # Raw result, we can format it later
+
+        raw = result["data"][0]  # e.g. "Human Probability: 99.83% | AI Probability: 0.17%"
+
+        human_match = re.search(r"Human[^0-9]*([\d.]+)%", raw)
+        ai_match = re.search(r"AI[^0-9]*([\d.]+)%", raw)
+
+        if human_match and ai_match:
+            human_prob = float(human_match.group(1))
+            ai_prob = float(ai_match.group(1))
+            return {
+                "Human Probability": round(human_prob, 2),
+                "AI Probability": round(ai_prob, 2),
+            }
+        else:
+            return {"error": f"Unexpected response: {raw}"}
+
     except Exception as e:
         return {"error": str(e)}
-
+        
 def detect_text(text):
     results = {}
     # Transformers models
@@ -53,7 +64,7 @@ def detect_text(text):
         results[name] = run_hf_model(model_id, text)
 
     # SzegedAI (Space)
-    results["SzegedAI Detector"] = run_szegedai(text)
+    results["SzegedAI Detector"] = szegedai_predict(text)
 
     # Final verdict (simple rule-based)
     ai_probs = []