Update func_ai.py

func_ai.py

@@ -27,8 +27,24 @@ classifier = pipeline(
 
 
 def classify_comment(text):
-    translated_text = GoogleTranslator(source='auto', target="en").translate(text)
-    result = classifier(translated_text, ["interrogative", "non-interrogative"], clean_up_tokenization_spaces=True)
+    if not text:
+        print("Received empty text for classification.")
+        return "non-interrogative"
+    print(f"Classifying comment: {text}")
+    try:
+        translated_text = GoogleTranslator(source='auto', target="en").translate(text)
+    except Exception as e:
+        print(f"Translation failed: {e}")
+        return "non-interrogative"
+    if not translated_text:
+        print("Translation returned empty text.")
+        return "non-interrogative"
+    print(f"Translated text: {translated_text}")
+    try:
+        result = classifier(translated_text, ["interrogative", "non-interrogative"], clean_up_tokenization_spaces=True)
+    except Exception as e:
+        print(f"Classification failed: {e}")
+        return "non-interrogative"
     top_class = result['labels'][0]
     return top_class