Update app.py

app.py

@@ -16,27 +16,18 @@ model = AutoModelForCausalLM.from_pretrained(
     device_map="auto"
 )
 
-# Fungsi pembantu untuk memproses sapaan tanpa memakan kuota/resource GPU (Bebas Error ZeroGPU)
-def cek_sapaan(message):
-    pesan_bersih = message.lower().strip()
-    sapaan_id = ["halo", "hai", "selamat siang", "slamat siang", "selamat pagi", "slamat pagi", "selamat malam", "slamat malam", "apa kabar"]
-    
-    if any(sapaan in pesan_bersih for sapaan in sapaan_id):
-        respon_sapaan = "Hai, selamat siang! Ada yang bisa saya bantu untuk diterjemahkan ke dalam Bahasa Biak hari ini? 😊"
-        if "pagi" in pesan_bersih:
-            respon_sapaan = "Hai, selamat pagi! Ada yang bisa saya bantu untuk diterjemahkan ke dalam Bahasa Biak hari ini? 😊"
-        elif "malam" in pesan_bersih:
-            respon_sapaan = "Hai, selamat malam! Ada yang bisa saya bantu untuk diterjemahkan ke dalam Bahasa Biak hari ini? 😊"
-        return respon_sapaan
-    return None
-
-# Fungsi utama penerjemah yang khusus berjalan di GPU Hugging Face
+# Fungsi utama penerjemah dan chat yang memanfaatkan kekuatan penuh model Qwen di GPU
 @spaces.GPU
-def proses_gpu_translation(message, history):
+def prediksi_terjemahan(message, history):
+    # Mengubah instruksi dasar agar model Qwen bisa fleksibel mengobrol sekaligus menerjemahkan
     percakapan = [
         {
             "role": "system", 
-            "content": "Kamu adalah asisten AI penerjemah yang ahli. Tugasmu adalah menerjemahkan teks dari Bahasa Indonesia ke Bahasa Biak secara akurat, alami, ringkas, dan langsung."
+            "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."
+            )
         }
     ]
     
@@ -63,8 +54,8 @@ def proses_gpu_translation(message, history):
     generation_kwargs = dict(
         **inputs,
         streamer=streamer,
-        max_new_tokens=128,
-        temperature=0.1,
+        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
         top_p=0.85,
         repetition_penalty=1.2,
         do_sample=True,
@@ -80,43 +71,25 @@ def proses_gpu_translation(message, history):
         text_terjemahan += new_text
         yield text_terjemahan.strip()
 
-# Fungsi jembatan utama untuk mengatur alur distribusi respons
-def prediksi_terjemahan(message, history):
-    # 1. Cek apakah pesan berupa sapaan ringan (di luar GPU)
-    sapaan_terdeteksi = cek_sapaan(message)
-    if sapaan_terdeteksi:
-        yield sapaan_terdeteksi
-        return
-        
-    # 2. Jika bukan sapaan, lempar perintah secara aman ke GPU untuk diterjemahkan
-    for respon in proses_gpu_translation(message, history):
-        yield respon
-
 # ========================================================
 #  PERBAIKAN UI/UX: CUSTOM CSS PUSAT & RAPI (TIDAK MELEBAR)
 # ========================================================
 desain_kustom_css = """
 #wadah-lebar-chat {
-    max-width: 800px !important;   /* Membatasi lebar agar pas seperti ChatGPT */
-    margin: 0 auto !important;      /* Menjaga posisi tetap presisi di tengah layar */
+    max-width: 800px !important;
+    margin: 0 auto !important;
     float: none !important;
     padding-top: 20px;
 }
 .judul-pusat {
-    text-align: center !important;  /* Memaksa teks judul berada di tengah layar */
+    text-align: center !important;
     margin-bottom: 15px !important;
 }
 """
 
 with gr.Blocks() as demo:
-    
-    # SOLUSI VISUAL: Memasukkan judul dan chatbot ke dalam kolom pembatas 800px yang sama
     with gr.Column(elem_id="wadah-lebar-chat"):
-        
-        # Menyisipkan teks judul dengan kelas CSS khusus agar sejajar tegak lurus di tengah
         gr.Markdown("# 🤖 AI Bahasa Indonesia-Biak", elem_classes="judul-pusat")
-        
-        # Kotak chatbot otomatis mengikuti lebar container 800px tepat di bawah judul
         gr.ChatInterface(
             fn=prediksi_terjemahan,