Create app.py

app.py

@@ -0,0 +1,70 @@
+import os
+import gradio as gr
+from fastapi import FastAPI
+from pydantic import BaseModel
+from transformers import AutoTokenizer, AutoModelForSeq2SeqLM, pipeline
+
+# ====== KONFIG ======
+MODEL_ID = os.getenv("MODEL_ID", "hasmar03/mt5_id2md")  # ganti kalau perlu
+MAX_LEN  = int(os.getenv("MAX_LEN", "128"))
+
+# ====== LOAD MODEL SEKALI SAJA (startup) ======
+tokenizer = AutoTokenizer.from_pretrained(MODEL_ID)
+model     = AutoModelForSeq2SeqLM.from_pretrained(MODEL_ID)
+pipe = pipeline(
+    "text2text-generation",
+    model=model,
+    tokenizer=tokenizer,
+    max_length=MAX_LEN,
+)
+
+# ====== LOGIKA TERJEMAH ======
+def _build_prompt(text: str, direction: str):
+    # sesuaikan dengan cara Anda melatih prompt-nya
+    if direction == "id2md":
+        return f"translate Indonesian to Mandar: {text}"
+    elif direction == "md2id":
+        return f"translate Mandar to Indonesian: {text}"
+    else:
+        return text
+
+def translate_fn(text: str, arah: str):
+    prompt = _build_prompt(text, "id2md" if "Indonesia" in arah else "md2id")
+    out = pipe(prompt)[0]["generated_text"]
+    return out
+
+# ====== UI GRADIO (untuk testing manual di browser) ======
+with gr.Blocks(title="Mandar ↔ Indonesia Translator") as demo:
+    gr.Markdown("### Mandar ↔ Indonesia Translator")
+    arah = gr.Radio(
+        ["Indonesia → Mandar", "Mandar → Indonesia"],
+        value="Indonesia → Mandar",
+        label="Arah"
+    )
+    src = gr.Textbox(label="Teks sumber", lines=3)
+    btn = gr.Button("Terjemahkan")
+    out = gr.Textbox(label="Hasil", lines=3)
+    btn.click(translate_fn, inputs=[src, arah], outputs=out)
+
+# ====== API FASTAPI (untuk Android / cURL) ======
+class TranslateReq(BaseModel):
+    text: str
+    direction: str = "id2md"   # "id2md" atau "md2id"
+
+class TranslateResp(BaseModel):
+    translation: str
+
+fastapi_app = FastAPI()
+
+@fastapi_app.get("/health")
+def health():
+    return {"ok": True}
+
+@fastapi_app.post("/translate", response_model=TranslateResp)
+def translate_api(body: TranslateReq):
+    prompt = _build_prompt(body.text, body.direction)
+    out = pipe(prompt)[0]["generated_text"]
+    return {"translation": out}
+
+# Mount: root (/) tetap UI Gradio, endpoint /translate adalah FastAPI
+app = gr.mount_gradio_app(fastapi_app, demo, path="/")