Update app.py

Added restriction

app.py

@@ -33,6 +33,59 @@ Your responses should only be within the financial subject, any other prompt or
 # MAX_MAX_NEW_TOKENS = 2048
 # DEFAULT_MAX_NEW_TOKENS = 512
 
+restricted_words = [
+    # Olahraga & aktivitas fisik
+    "olahraga", "sport", "sports", "sepak bola", "football", "basket", "basketball", "voli", "futsal", "renang",
+    "bersepeda", "yoga", "fitness", "workout", "gym", "lari", "jogging", "marathon", "bodybuilding",
+
+    # Hiburan & budaya populer
+    "anime", "manga", "komik", "comic", "kartun", "cartoon", "drama", "romance", "horror", "thriller", "action",
+    "comedy", "film", "movie", "tv series", "series", "Netflix", "Disney", "HBO", "Amazon Prime", "YouTube", 
+    "streaming", "idol", "band", "musik", "lagu", "konser", "album", "artis", "aktor", "aktris", "selebritis", 
+    "selebriti", "gosip", "viral", "trending", "meme", "influencer", "content creator", "tiktok", "vlogger", 
+
+    # Kuliner & makanan
+    "makanan", "minuman", "resep", "memasak", "masakan", "masak", "kuliner", "menu", "ayam", "sapi", "ikan", 
+    "rendang", "sate", "bakso", "mie", "lontong", "nasi goreng", "kue", "jajanan", "kopi", "teh", "es krim", 
+
+    # Hubungan, percintaan, kehidupan pribadi
+    "pacaran", "cinta", "romantis", "romansa", "dating", "love", "jodoh", "nikah", "pernikahan", "mantan", 
+    "crush", "friendzone", "friendship", "persahabatan", "PDA", "perasaan", "emosi", 
+
+    # Fashion, kecantikan & gaya hidup
+    "fashion", "baju", "pakaian", "kaos", "celana", "jaket", "sepatu", "tas", "gaya", "OOTD", "skincare", 
+    "makeup", "lipstik", "kecantikan", "beauty", "perawatan", "spa", "salon", 
+
+    # Teknologi umum & AI
+    "AI", "artificial intelligence", "machine learning", "deep learning", "robot", "android", "teknologi", 
+    "smartphone", "gadget", "gawai", "laptop", "hardware", "software", "game", "gaming", "console", "VR", "AR", 
+
+    # Ilmu pengetahuan umum
+    "sains", "science", "astronomi", "planet", "galaksi", "bintang", "meteor", "luar angkasa", "kimia", "biologi", 
+    "fisika", "matematika", "matematika diskrit", "aljabar", "statistika", "geografi", "sejarah", 
+
+    # Filosofi & agama
+    "filsafat", "filosofi", "agama", "teologi", "psikologi", "mitologi", "mythology", "spiritual", "meditasi", 
+    "karma", "reinkarnasi", 
+
+    # Hukum & kriminalitas
+    "hukum", "UU", "undang-undang", "legal", "ilegal", "pengadilan", "hakim", "jaksa", "kriminal", "pencurian", 
+    "korupsi", "penipuan", "pembunuhan", "kejahatan", "detektif", "penjara", "narkoba", "obat-obatan", 
+
+    # Politik, militer, konflik
+    "politik", "partai", "kampanye", "pemilu", "kandidat", "presiden", "dpr", "militer", "tentara", "perang", 
+    "konflik", "senjata", "roket", "bom", "rudal", "pemberontak", "terorisme", 
+
+    # Alam, hewan, dan lingkungan
+    "binatang", "hewan", "anjing", "kucing", "pets", "flora", "fauna", "tumbuhan", "tanaman", "hutan", "alam", 
+    "kebun", "cuaca", "iklim", "climate", "bencana", "gempa", "banjir", "angin", "sustainability", "green energy",
+
+    # Lain-lain di luar literasi keuangan
+    "astrologi", "zodiak", "ramalan", "santet", "sihir", "mistis", "dongeng", "cerita rakyat", "fantasi", "fiksi", 
+    "hantu", "alien", "konspirasi", "misteri"
+]
+
+
 model_id = "FOLZi/FinID_v2_8B_Chat"
 quantization_config = BitsAndBytesConfig(load_in_8bit=True)
 
@@ -70,6 +123,15 @@ ct_end_token = "<|endoftext|>"
 
 @spaces.GPU()
 def generate(message, history):
+
+    # Pendeteksian via primitif TOO BAD!
+    matched_words = [word for word in restricted_words if re.search(rf"\b{word}\b", message, re.IGNORECASE)]
+    
+    # Jika ditemukan, hentikan proses dan kembalikan pesan peringatan
+    if matched_words:
+        yield "Maaf, saya hanya dapat menjawab pertanyaan yang berkaitan dengan literasi keuangan."
+        return
+    
     history = [] # Disabled history.
     history_transformer_format = history + [[message, ""]]
     stop = StopOnTokens()