..

app.py

@@ -32,9 +32,36 @@ bert_model_name = "bert-base-uncased"
 bert_tokenizer = AutoTokenizer.from_pretrained(bert_model_name)
 bert_model = BertForSequenceClassification.from_pretrained(bert_model_name, num_labels=3).to(device)  
 
-# Define Topic Labels
-darija_topic_labels = ["Customer Service", "Retention Service", "Billing Issue"]
-english_topic_labels = ["Support Request", "Subscription Issue", "Payment Dispute"]
+# Libellés en Darija (Arabe et Latin)
+darija_topic_labels = [
+    "مشكيل ف الشبكة (Mochkil f réseau)",        # Problème de réseau
+    "مشكيل ف الانترنت (Mochkil f internet)",      # Problème d'Internet
+    "مشكيل ف الفاتورة (Mochkil f l'factura)",     # Problème de facturation et paiement
+    "مشكيل ف التعبئة (Mochkil f l'recharge)",     # Problème de recharge et forfaits
+    "مشكيل ف التطبيق (Mochkil f l'application)",  # Problème avec l’application (Orange et Moi...)
+    "مشكيل ف بطاقة SIM (Mochkil f carte SIM)",    # Problème avec la carte SIM
+    "مساعدة تقنية (Mosa3ada technique)",         # Assistance technique
+    "العروض والتخفيضات (Offres w promotions)",   # Offres et promotions
+    "طلب معلومات (Talab l'ma3loumat)",           # Demande d'information
+    "شكاية (Chikaya)",                            # Réclamation
+    "حاجة أخرى (Chi haja okhra)"                 # Autre
+]
+
+# Libellés en Anglais
+english_topic_labels = [
+    "Network Issue",
+    "Internet Issue",
+    "Billing & Payment Issue",
+    "Recharge & Plans",
+    "App Issue",
+    "SIM Card Issue",
+    "Technical Support",
+    "Offers & Promotions",
+    "General Inquiry",
+    "Complaint",
+    "Other"
+]
+
 
 def transcribe_audio(audio):
     """Convert audio to text, translate it, and classify topics in both Darija and English."""