Add debug error handling

app.py

@@ -2,9 +2,7 @@ import gradio as gr
 import requests
 import os
 
-# Load API token from environment variable
 HF_TOKEN = os.environ.get("HF_TOKEN")
-
 API_URL = "https://api-inference.huggingface.co/models/Hate-speech-CNERG/dehatebert-mono-english"
 
 def query(text):
@@ -16,32 +14,35 @@ def predict(comment):
     if not comment.strip():
         return "⚠️ Please enter some text.", ""
     
-    result = query(comment)
-    
-    if isinstance(result, list):
-        result = result[0][0]
-        label = result["label"]
-        score = result["score"]
+    try:
+        result = query(comment)
+        
+        # Show raw result for debugging
+        if isinstance(result, dict):
+            return f"API Error: {result}", ""
         
-        if label == "HATE":
-            prediction = "🚨 Hate Speech Detected"
-            probability = round(score * 100, 2)
-        else:
-            prediction = "✅ No Hate Speech"
-            probability = round((1 - score) * 100, 2)
+        if isinstance(result, list):
+            result = result[0][0]
+            label = result["label"]
+            score = result["score"]
+            
+            if label == "HATE":
+                prediction = "🚨 Hate Speech Detected"
+                probability = round(score * 100, 2)
+            else:
+                prediction = "✅ No Hate Speech"
+                probability = round((1 - score) * 100, 2)
+            
+            return prediction, f"{probability}%"
         
-        return prediction, f"{probability}%"
+        return f"Unexpected response: {result}", ""
     
-    return "❌ Error - Model loading, try again.", ""
+    except Exception as e:
+        return f"Exception: {str(e)}", ""
 
-# Gradio UI
 demo = gr.Interface(
     fn=predict,
-    inputs=gr.Textbox(
-        lines=4,
-        placeholder="Enter a comment to analyze...",
-        label="Comment"
-    ),
+    inputs=gr.Textbox(lines=4, placeholder="Enter a comment...", label="Comment"),
     outputs=[
         gr.Textbox(label="Prediction"),
         gr.Textbox(label="Confidence")