Update app.py

app.py

@@ -17,23 +17,34 @@ model = AutoModelForCausalLM.from_pretrained(
     device_map="auto"
 )
 
-# Fungsi interseptor untuk mendeteksi sapaan awal dalam Bahasa Indonesia
+# =====================================================================
+# MODIFIKASI 1: Fungsi interseptor sapaan yang lebih fleksibel (Regex)
+# =====================================================================
 def cek_sapaan_otomatis(message):
+    # Menghapus tanda baca dan mengubah ke huruf kecil
     teks_bersih = re.sub(r'[^\w\s]', '', message.strip().lower())
-    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?"
-            
+    # Deteksi variasi sapaan dasar secara parsial (termasuk salah ketik seperti hala/helo)
+    apakah_menyapa = any(kata in teks_bersih for kata in ["halo", "hala", "hai", "helo", "hi", "p "])
+    
+    # Deteksi waktu
+    apakah_malam = "malam" in teks_bersih
+    apakah_pagi = "pagi" in teks_bersih
+    apakah_siang = "siang" in teks_bersih
+    apakah_sore = "sore" in teks_bersih
+    
+    # Logika penentu respon instan Bahasa Indonesia
+    if apakah_malam or (apakah_menyapa and apakah_malam):
+        return "Halo, selamat malam juga! Ada yang bisa saya bantu untuk diterjemahkan ke Bahasa Biak?"
+    elif apakah_pagi or (apakah_menyapa and apakah_pagi):
+        return "Halo, selamat pagi juga! Ada yang bisa saya bantu hari ini?"
+    elif apakah_siang or (apakah_menyapa and apakah_siang):
+        return "Halo, selamat siang juga! Ada yang bisa saya bantu?"
+    elif apakah_sore or (apakah_menyapa and apakah_sore):
+        return "Halo, selamat sore juga! Ada yang bisa saya bantu?"
+    elif apakah_menyapa or teks_bersih == "p":
+        return "Halo! Ada yang bisa saya bantu untuk diterjemahkan ke Bahasa Biak?"
+        
     return None
 
 # Fungsi utama penerjemah dan chat di GPU
@@ -48,10 +59,10 @@ def prediksi_terjemahan(message, history):
         {
             "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 menggunakan Bahasa Indonesia. "
-                "Jika pengguna meminta terjemahan atau memasukkan kalimat Bahasa Indonesia untuk diartikan, "
-                "berikan hasil terjemahan ke Bahasa Biak yang akurat dan alami."
+                "Kamu adalah asisten AI yang ramah. Kamu menguasai Bahasa Indonesia dan Bahasa Biak. "
+                "Aturan interaksi:\n"
+                "1. Jika pengguna menyapa (seperti 'halo', 'selamat malam') atau bertanya umum, balaslah dengan ramah menggunakan Bahasa Indonesia.\n"
+                "2. Jika pengguna meminta terjemahan atau memasukkan kalimat untuk diartikan ke Bahasa Biak, berikan hasil terjemahan yang akurat."
             )
         }
     ]
@@ -82,13 +93,16 @@ def prediksi_terjemahan(message, history):
     inputs = tokenizer([input_text], return_tensors="pt").to(model.device)
     streamer = TextIteratorStreamer(tokenizer, skip_prompt=True, skip_special_tokens=True)
     
+    # =====================================================================
+    # MODIFIKASI 2: Penyesuaian hyperparameter agar Qwen lebih luwes chitchat
+    # =====================================================================
     generation_kwargs = dict(
         **inputs,
         streamer=streamer,
         max_new_tokens=256,
-        temperature=0.4,
-        top_p=0.85,
-        repetition_penalty=1.2,
+        temperature=0.6,      # Diubah ke 0.6 agar tidak terlalu kaku saat mengobrol biasa
+        top_p=0.9,            # Diubah ke 0.9 untuk memberikan variasi kosakata dasar yang lebih natural
+        repetition_penalty=1.1, # Diturunkan ke 1.1 agar kalimat mengalir lebih baik
         do_sample=True,
         eos_token_id=tokenizer.eos_token_id,
         pad_token_id=tokenizer.pad_token_id if tokenizer.pad_token_id is not None else tokenizer.eos_token_id
@@ -118,13 +132,11 @@ desain_kustom_css = """
 }
 """
 
-# Perbaikan Gradio 6.0: Parameter css dipindahkan dari gr.Blocks ke demo.launch
 with gr.Blocks() 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
-            # Perbaikan Gradio 6.0: Menghapus argumen type="messages"
         )
 
 if __name__ == "__main__":