Update ner_logic.py

ner_logic.py

@@ -51,13 +51,22 @@ def refine_labels_batch(misc_items, full_sentence):
         generated_ids = llm_model.generate(model_inputs.input_ids, max_new_tokens=1500, do_sample=False)
         output = tokenizer.batch_decode(generated_ids[:, model_inputs.input_ids.shape[1]:], skip_special_tokens=True)[0]
         
-        json_match = re.search(r'\[\s*\{.*\}\s*\]', output, re.DOTALL)
-        if json_match:
-            results = json.loads(json_match.group(0))
-            for r in results:
-                if not isinstance(r.get("karar"), str) or r["karar"] in ["True", "False"]:
-                    r["karar"] = "MISC"
-            return results
+        # JSON bloğunu ayıklamak için regex kullan
+    json_match = re.search(r'\[.*\]', output, re.DOTALL)
+    if json_match:
+        raw_json = json_match.group(0)
+        # Bazen model son süslü parantezi unutur, basit bir tamamlama denenebilir
+        if raw_json.count('{') > raw_json.count('}'):
+            raw_json += "}]"
+            
+        results = json.loads(raw_json)
+        return results
+    else:
+        # Hata durumunda log bas ki ne geldiğini görelim
+        print(f"LLM Yanlış Format Döndürdü: {output}")
         return []
+except Exception as e:
+    print(f"Parsing Hatası: {str(e)}")
+    return []
     except:
         return []