Update app.py

app.py

@@ -1,6 +1,7 @@
 import gradio as gr
 import torch
 import spaces
+import re
 from transformers import AutoModelForCausalLM, AutoTokenizer, TextIteratorStreamer
 from threading import Thread
 
@@ -16,30 +17,66 @@ model = AutoModelForCausalLM.from_pretrained(
     device_map="auto"
 )
 
+# Fungsi interseptor untuk mendeteksi sapaan awal dalam Bahasa Indonesia
+def cek_sapaan_otomatis(message):
+    # Membersihkan teks dari tanda baca dan mengubah ke huruf kecil
+    teks_bersih = re.sub(r'[^\w\s]', '', message.strip().lower())
+    
+    # Daftar kata sapaan dasar
+    kata_sapaan = ["halo", "selamat malam", "selamat pagi", "selamat siang", "selamat sore", "hai", "helo", "hi", "assalamualaikum"]
+    
+    if teks_bersih in kata_sapaan:
+        if "malam" in teks_bersih:
+            return "Halo, selamat malam juga! Ada yang bisa saya bantu untuk diterjemahkan ke Bahasa Biak?"
+        elif "pagi" in teks_bersih:
+            return "Halo, selamat pagi juga! Ada yang bisa saya bantu hari ini?"
+        elif "siang" in teks_bersih:
+            return "Halo, selamat siang juga! Ada yang bisa saya bantu?"
+        elif "sore" in teks_bersih:
+            return "Halo, selamat sore juga! Ada yang bisa saya bantu?"
+        else:
+            return "Halo! Ada yang bisa saya bantu untuk diterjemahkan ke Bahasa Biak?"
+            
+    return None
+
 # Fungsi utama penerjemah dan chat yang memanfaatkan kekuatan penuh model Qwen di GPU
 @spaces.GPU
 def prediksi_terjemahan(message, history):
-    # Mengubah instruksi dasar agar model Qwen bisa fleksibel mengobrol sekaligus menerjemahkan
+    # 1. Cek interseptor sapaan terlebih dahulu agar respon instan dan sesuai harapan
+    respon_sapaan = cek_sapaan_otomatis(message)
+    if respon_sapaan:
+        yield respon_sapaan
+        return
+
+    # 2. Jika bukan sapaan biasa, proses menggunakan Model Qwen2.5
     percakapan = [
         {
             "role": "system", 
             "content": (
                 "Kamu adalah asisten AI yang ramah dan ahli dalam Bahasa Indonesia dan Bahasa Biak. "
-                "Jawablah sapaan dan pertanyaan umum pengguna dengan luwes. Jika pengguna meminta "
-                "terjemahan ke Bahasa Biak, berikan hasil terjemahan yang akurat dan alami."
+                "Jawablah sapaan dan pertanyaan umum pengguna dengan luwes menggunakan Bahasa Indonesia. "
+                "Jika pengguna meminta terjemahan atau memasukkan kalimat Bahasa Indonesia untuk diartikan, "
+                "berikan hasil terjemahan ke Bahasa Biak yang akurat dan alami."
             )
         }
     ]
     
+    # Perbaikan pembacaan riwayat chat (mendukung format Gradio baru v4+)
     if history:
         for interaksi in history:
-            if isinstance(interaksi, (list, tuple)) and len(interaksi) == 2:
+            if isinstance(interaksi, dict): # Format Baru Gradio
+                role = interaksi.get("role")
+                content = interaksi.get("content")
+                if role and content:
+                    percakapan.append({"role": role, "content": str(content)})
+            elif isinstance(interaksi, (list, tuple)) and len(interaksi) == 2: # Format Lama Gradio
                 human, assistant = interaksi
                 if human:
                     percakapan.append({"role": "user", "content": str(human)})
                 if assistant:
                     percakapan.append({"role": "assistant", "content": str(assistant)})
     
+    # Masukkan pesan terbaru dari user
     percakapan.append({"role": "user", "content": message})
     
     input_text = tokenizer.apply_chat_template(
@@ -54,8 +91,8 @@ def prediksi_terjemahan(message, history):
     generation_kwargs = dict(
         **inputs,
         streamer=streamer,
-        max_new_tokens=256,   # Ditingkatkan ke 256 agar jawaban obrolan umum tidak terpotong tengah jalan
-        temperature=0.5,      # Menggunakan 0.5 agar AI bisa menjawab sapaan secara kreatif dan tidak kaku
+        max_new_tokens=256,
+        temperature=0.4,      # Sedikit diturunkan agar model lebih patuh pada instruksi System Prompt
         top_p=0.85,
         repetition_penalty=1.2,
         do_sample=True,
@@ -69,7 +106,8 @@ def prediksi_terjemahan(message, history):
     text_terjemahan = ""
     for new_text in streamer:
         text_terjemahan += new_text
-        yield text_terjemahan.strip()
+        # Menghapus .strip() di loop agar spasi ketikan tidak terpotong di UI Gradio
+        yield text_terjemahan
 
 # ========================================================
 #  PERBAIKAN UI/UX: CUSTOM CSS PUSAT & RAPI (TIDAK MELEBAR)
@@ -87,16 +125,15 @@ desain_kustom_css = """
 }
 """
 
-with gr.Blocks() as demo:
+with gr.Blocks(css=desain_kustom_css) as demo:
     with gr.Column(elem_id="wadah-lebar-chat"):
         gr.Markdown("# 🤖 AI Bahasa Indonesia-Biak", elem_classes="judul-pusat")
         gr.ChatInterface(
             fn=prediksi_terjemahan,
-            
+            type="messages" # Memastikan format riwayat chat Gradio stabil menggunakan struktur dictionary
         )
 
 if __name__ == "__main__":
     demo.launch(
-        theme=gr.themes.Soft(),
-        css=desain_kustom_css
+        theme=gr.themes.Soft()
     )