15

app.py

@@ -1,5 +1,5 @@
 # app.py
-# PapuaTranslate — Flask + SQLAlchemy + Supabase + mT5-LoRA (lazy load)
+# PapuaTranslate — Flask + SQLAlchemy (Supabase/SQLite) + mT5-LoRA (lazy)
 import os, re, logging, threading
 from datetime import datetime, timezone
 from functools import wraps
@@ -9,9 +9,10 @@ from flask import Flask, render_template, request, redirect, url_for, session, j
 logging.basicConfig(level=logging.INFO, format="%(asctime)s | %(levelname)s | %(message)s")
 log = logging.getLogger("papua-app")
 
-# ===== Flask (templates di folder 'templates') =====
-app = Flask(__name__, template_folder="templates", static_folder="static")
-SESSION_SECURE = os.getenv("SESSION_COOKIE_SECURE", "true").lower() in ("1", "true", "yes")
+# ===== Flask =====
+# >> PAKAI folder 'frontend' (sesuai struktur kamu)
+app = Flask(__name__, template_folder="frontend", static_folder="static")
+SESSION_SECURE = os.getenv("SESSION_COOKIE_SECURE", "true").lower() in ("1","true","yes")
 app.config.update(
     SECRET_KEY=os.getenv("SECRET_KEY", "dev-secret-change-me"),
     SESSION_COOKIE_SAMESITE="Lax",
@@ -27,7 +28,7 @@ if not DATABASE_URL:
     DATABASE_URL = "sqlite:////tmp/app.db"
     log.warning("[DB] DATABASE_URL tidak diset; pakai SQLite /tmp/app.db")
 else:
-    # normalisasi skema lama → psycopg2 driver
+    # normalisasi skema → psycopg2
     if DATABASE_URL.startswith("postgres://"):
         DATABASE_URL = DATABASE_URL.replace("postgres://", "postgresql+psycopg2://", 1)
     elif DATABASE_URL.startswith("postgresql://"):
@@ -85,7 +86,7 @@ PAPUA_MAP = {
 }
 def prenorm(text: str) -> str:
     t = re.sub(r"\s+", " ", text.strip())
-    t = t.replace("…", "...").replace("–", "-").replace("—", "-")
+    t = t.replace("…","...").replace("–","-").replace("—","-")
     for pat, repl in PAPUA_MAP.items(): t = re.sub(pat, repl, t, flags=re.IGNORECASE)
     return t
 
@@ -93,8 +94,9 @@ def prenorm(text: str) -> str:
 from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
 from peft import PeftModel
 
-BASE_MODEL_ID = os.getenv("BASE_MODEL_ID", "amosnbn/cendol-mt5-base-inst")
-ADAPTER_ID    = os.getenv("ADAPTER_ID", "amosnbn/papua-lora-ckpt-168")
+# Untuk uji UI cepat gunakan mt5-small & kosongkan adapter, lalu ganti ke model kamu
+BASE_MODEL_ID = os.getenv("BASE_MODEL_ID", "google/mt5-small")
+ADAPTER_ID    = os.getenv("ADAPTER_ID", "")
 DEVICE        = "cuda" if os.getenv("DEVICE", "cpu") == "cuda" else "cpu"
 
 TOK = None
@@ -103,10 +105,10 @@ _MODEL_LOCK = threading.Lock()
 
 def _load_model():
     global TOK, MODEL
-    log.info("[MODEL] loading base=%s adapter=%s", BASE_MODEL_ID, ADAPTER_ID)
+    log.info("[MODEL] loading base=%s adapter=%s", BASE_MODEL_ID, ADAPTER_ID or "-")
     TOK = AutoTokenizer.from_pretrained(BASE_MODEL_ID)
     base = AutoModelForSeq2SeqLM.from_pretrained(BASE_MODEL_ID)
-    MODEL = PeftModel.from_pretrained(base, ADAPTER_ID)
+    MODEL = PeftModel.from_pretrained(base, ADAPTER_ID) if ADAPTER_ID else base
     MODEL.eval().to(DEVICE)
     log.info("[MODEL] ready on %s", DEVICE)
 
@@ -131,7 +133,7 @@ def translate_with_model(text: str, max_new_tokens: int = 48) -> str:
     )
     return tok.decode(outputs[0], skip_special_tokens=True)
 
-# ===== Little utils / logging =====
+# ===== Utils / logging =====
 @app.before_request
 def _log_req():
     if request.path not in ("/health", "/ping", "/favicon.ico"):
@@ -157,13 +159,11 @@ def login_post():
     email = (request.form.get("email") or "").strip().lower()
     pwd   = request.form.get("password") or ""
     if not email or not pwd:
-        flash("Isi email dan password", "error")
-        return redirect(url_for("login_get"))
+        flash("Isi email dan password", "error"); return redirect(url_for("login_get"))
     with SessionLocal() as s:
         u = s.query(User).filter_by(email=email).first()
         if not u or not verify_password(u, pwd):
-            flash("Email atau password salah", "error")
-            return redirect(url_for("login_get"))
+            flash("Email atau password salah", "error"); return redirect(url_for("login_get"))
         session["uid"], session["email"] = u.id, u.email
     return redirect(url_for("index"))
 
@@ -176,12 +176,10 @@ def register_post():
     email = (request.form.get("email") or "").strip().lower()
     pwd   = request.form.get("password") or ""
     if not email or not pwd:
-        flash("Isi email dan password", "error")
-        return redirect(url_for("register_get"))
+        flash("Isi email dan password", "error"); return redirect(url_for("register_get"))
     with SessionLocal() as s:
         if s.query(User).filter_by(email=email).first():
-            flash("Email sudah terdaftar", "error")
-            return redirect(url_for("register_get"))
+            flash("Email sudah terdaftar", "error"); return redirect(url_for("register_get"))
         u = User(email=email); set_password(u, pwd)
         s.add(u); s.commit()
         session["uid"], session["email"] = u.id, u.email
@@ -202,7 +200,6 @@ def index():
                  .order_by(Translation.id.desc())
                  .limit(10).all())
         data = [{"src": it.src, "mt": it.mt, "created_at": it.created_at} for it in items]
-    # index.html kamu memakai 'user' & 'data'
     return render_template("index.html", user=session.get("email"), data=data, device=DEVICE)
 
 @app.get("/about")