Update api.py

api.py

@@ -28,7 +28,7 @@ except ImportError as e:
 import google.generativeai as genai
 
 try:
-    from scoring.specificity import load_model as load_specificity_model, predict_batch
+    from scoring.specificity import load_model as load_specificity_model, predict_batch, ProblematicList
 except ImportError as e:
     print(f"Error importing scoring component: {e}")
     print("Please ensure scoring is in your Python path or installed.")
@@ -283,8 +283,11 @@ def generate_problematics_and_scores(descriptions: list[str]):
              logger.error("Gemini API returned an empty result.")
              raise HTTPException(status_code=502, detail="LLM returned an empty problematic.")
     
-    # --- 2. Evaluate Specificity using External API ---
-    results = predict_batch(interrogative_probs)
+    # --- 2. Evaluate Specificity ---
+    if not interrogative_probs:
+        return []
+    problematic_list_input = ProblematicList(problematics=interrogative_probs)
+    results = predict_batch(problematic_list_input)
     return results