Update app.py

app.py

@@ -16,7 +16,7 @@ def log(msg):
         print(msg, flush=True)
 
 # ======================
-# MODEL 1 — RATING & KATEGORI (ASLI, TIDAK DIUBAH)
+# MODEL 1 — RATING & KATEGORI (ASLI)
 # ======================
 classifier = pipeline(
     "zero-shot-classification",
@@ -40,7 +40,7 @@ abusive_model.to(device)
 id2label = {0: "Tidak Abusif", 1: "Abusif"}
 
 # ======================
-# CONFIG (ASLI — TIDAK DIUBAH)
+# CONFIG (ASLI)
 # ======================
 SEMANTIC_EVENT = {
     "sadisme": [
@@ -99,6 +99,22 @@ def split_kalimat(text):
         if len(k.strip()) >= 20
     ]
 
+def sliding_window(kalimat, window=4):
+    if len(kalimat) < window:
+        return kalimat
+    return [
+        " ".join(kalimat[i:i+window])
+        for i in range(len(kalimat) - window + 1)
+    ]
+
+def event_completeness(text):
+    text = text.lower()
+    return sum(any(w in text for w in words) for words in EVENT_UNITS.values())
+
+def is_metaphor(text):
+    text = text.lower()
+    return any(a in text and b in text for a, b in METAPHOR_PAIRS)
+
 def rating_usia(kategori):
     if "sadisme" in kategori:
         return 21
@@ -109,13 +125,21 @@ def rating_usia(kategori):
     return 0
 
 # ======================
-# MODE 1 — RATING & KATEGORI (ASLI, TIDAK DIUBAH)
+# CORE ANALYSIS + DEBUG (JANGAN DIUBAH)
 # ======================
 def analyze_text(judul, isi):
     kalimat = split_kalimat(isi)
+    windows = sliding_window(kalimat)
+
     detected = set()
 
-    for w in kalimat:
+    log("\n" + "=" * 80)
+    log(f"JUDUL: {judul}")
+
+    for w in windows:
+        log("\n[WINDOW]")
+        log(w)
+
         for kategori, desc in SEMANTIC_EVENT.items():
             res = classifier(
                 w,
@@ -123,14 +147,26 @@ def analyze_text(judul, isi):
                 hypothesis_template="Teks ini menggambarkan {}."
             )
             score = max(res["scores"])
-            if score >= THRESHOLD[kategori]:
-                detected.add(kategori)
+            threshold = THRESHOLD[kategori]
+
+            log(f"  - {kategori} | score={score:.3f} | threshold={threshold}")
+
+            if score < threshold:
+                log("    ❌ FAIL: threshold")
+                continue
+
+            log("    ✅ ACCEPTED")
+            detected.add(kategori)
 
     usia = rating_usia(detected)
+    log(f"\nFINAL KATEGORI: {list(detected)}")
+    log(f"RATING USIA: {usia}")
+    log("=" * 80)
+
     return usia, ", ".join(sorted(detected))
 
 # ======================
-# MODE 2 — FILTER ABUSIVE (BARU, TERPISAH)
+# FILTER ABUSIVE (TERPISAH, AMAN)
 # ======================
 def filter_abusive(isi):
     paragraphs = re.split(r'\n+', isi)
@@ -170,9 +206,9 @@ def filter_abusive(isi):
     return "\n".join(output)
 
 # ======================
-# ROUTER (KHUSUS RAILWAY)
+# ROUTER /ANALYZE (MODE-BASED)
 # ======================
-def api_router(judul, isi, mode):
+def analyze_router(judul, isi, mode):
     if mode == "rating":
         usia, kategori = analyze_text(judul, isi)
         return {
@@ -190,46 +226,26 @@ def api_router(judul, isi, mode):
     }
 
 # ======================
-# UI ASLI (TIDAK DIUBAH)
+# GRADIO API (ROUTE TETAP /analyze)
 # ======================
 demo = gr.Interface(
-    fn=analyze_text,
-    inputs=[
-        gr.Textbox(label="Judul"),
-        gr.Textbox(label="Isi Cerita", lines=10)
-    ],
-    outputs=[
-        gr.Number(label="Rating Usia"),
-        gr.Textbox(label="Kategori")
-    ],
-    title="READEAR",
-    api_name="analyze"
-)
-
-# ======================
-# API UNTUK RAILWAY
-# ======================
-api = gr.Interface(
-    fn=api_router,
+    fn=analyze_router,
     inputs=[
         gr.Textbox(),
         gr.Textbox(),
         gr.Radio(["rating", "abusive"])
     ],
     outputs=gr.JSON(),
-    api_name="predict"
+    api_name="analyze"
 )
 
 # ======================
 # ENTRY POINT
 # ======================
 if __name__ == "__main__":
-    gr.TabbedInterface(
-        [demo, api],
-        ["UI", "API"]
-    ).launch(
+    demo.launch(
         server_name="0.0.0.0",
         server_port=7860,
         ssr_mode=False,
         show_error=True
-    ) 
+    )