Update app.py

app.py

@@ -18,25 +18,56 @@ current_dir = os.path.dirname(os.path.abspath(__file__))
 font_path = os.path.join(current_dir, "ArabicR2013-J25x.ttf")
 
 ARABIC_STOP_WORDS = {
-    'في', 'من', 'إلى', 'على', 'عن', 'مع', 'خلال', 'حتى', 'إذا', 'ثم',
-    'أو', 'و', 'ف', 'ل', 'ب', 'ك', 'لل', 'ال', 'هذا', 'هذه', 'ذلك',
-    'تلك', 'هؤلاء', 'هم', 'هن', 'هو', 'هي', 'نحن', 'انت', 'انتم',
-    'كان', 'كانت', 'يكون', 'تكون', 'اي', 'كل', 'بعض', 'غير', 'حول',
-    'عند', 'قد', 'لقد', 'لم', 'لن', 'لو', 'ما', 'ماذا', 'متى', 'كيف',
-    'اين', 'لماذا', 'الذي', 'التي', 'الذين', 'اللاتي', 'اللواتي',
-    'الان', 'بين', 'فوق', 'تحت', 'امام', 'خلف', 'حين', 'قبل', 'بعد',
-    'و', 'أن', 'في', 'كل', 'لم', 'لن', 'له', 'من', 'هو', 'هي', 
-    'كما', 'لها', 'منذ', 'وقد', 'ولا', 'نفس', 'ولم', 'حيث', 'هناك',
-    'جدا', 'ذات', 'ضمن', 'انه', 'لدى', 'عليه', 'مثل', 'وله', 'عند',
-    'أما', 'هذه', 'وأن', 'وكل', 'وقال', 'لدي', 'وكان', 'فيه', 'وهي',
-    'وهو', 'تلك', 'كلم', 'لكن', 'وفي', 'وقف', 'ولقد', 'ومن', 'وهذا',
-    'اول', 'ضمن', 'انها', 'جميع', 'الذي', 'قبل', 'بعد', 'حول', 'ايضا',
-    'لازم', 'علي', 'يجب', 'صار', 'صارت', 'تحت', 'ضد',
-    'يا', 'لا', 'اما', 'بها', 'ان', 'به', 'الي', 'لما',
-    'انا', 'اليك', 'لي', 'لك', 'او', 'قلت'
+ARABIC_STOP_WORDS = {
+    # First group
+    'في', 'من', 'إلى', 'على', 'علي', 'عن', 'مع', 'خلال', 'حتي', 'حتى', 'إذا',
+    
+    # Middle group
+    'ثم', 'أو', 'و', 'ل', 'ب', 'ك', 'لل', 'ال', 'هذا', 
+    'هذه', 'ذلك', 'تلك', 'هؤلاء', 'هم', 'هن', 'هو', 'هي', 'نحن',
+    'انت', 'انتم', 'كان', 'كانت', 'يكون', 'تكون', 'اي', 'كل',
+    'بعض', 'غير', 'حول', 'عند', 'قد', 'لقد', 'لم', 'لن', 'لو',
+    'ما', 'ماذا', 'متى', 'كيف', 'اين', 'لماذا', 'الذي', 'التي',
+    'الذين', 'اللاتي', 'اللواتي', 'الان', 'بين', 'فوق', 'تحت',
+    'امام', 'خلف', 'حين', 'قبل', 'بعد', 'أن', 'له', 'كما', 'لها',
+    'منذ', 'نفس', 'حيث', 'هناك', 'جدا', 'ذات', 'ضمن', 'انه', 'لدى',
+    'عليه', 'مثل', 'أما', 'لدي', 'فيه', 'كلم', 'لكن', 'ايضا', 'لازم',
+     'يجب', 'صار', 'صارت', 'ضد', 'يا', 'لا', 'اما',
+    'بها', 'ان', 'به', 'الي', 'لما', 'انا', 'اليك', 'لي', 'لك', 'قلت',
+
+    # Middle group prefixed with "و"
+    'وثم', 'وأو', 'ول', 'وب', 'وك', 'ولل', 'وال', 
+    'وهذا', 'وهذه', 'وذلك', 'وتلك', 'وهؤلاء', 'وهم', 'وهن', 'وهو', 'وهي', 'ونحن',
+    'وانت', 'وانتم', 'وكان', 'وكانت', 'ويكون', 'وتكون', 'واي', 'وكل',
+    'وبعض', 'وغير', 'وحول', 'وعند', 'وقد', 'ولقد', 'ولم', 'ولن', 'ولو',
+    'وما', 'وماذا', 'ومتى', 'وكيف', 'واين', 'ولماذا', 'والذي', 'والتي',
+    'والذين', 'واللاتي', 'واللواتي', 'والان', 'وبين', 'وفوق', 'وتحت',
+    'وامام', 'وخلف', 'وحين', 'وقبل', 'وبعد', 'وأن', 'وله', 'وكما', 'ولها',
+    'ومنذ', 'ونفس', 'وحيث', 'وهناك', 'وجدا', 'وذات', 'وضمن', 'وانه', 'ولدى',
+    'وعليه', 'ومثل', 'وأما', 'وفيه', 'وكلم', 'ولكن', 'وايضا', 'ولازم',
+     'ويجب', 'وصار', 'وصارت', 'وضد', 'ويا', 'ولا', 'واما',
+    'وبها', 'وان', 'وبه', 'والي', 'ولما', 'وانا', 'واليك', 'ولي', 'ولك', 'وقلت',
+    
+    # Last group
+    'وفي', 'ومن', 'وعلى', 'وعلي', 'وعن', 'ومع', 'وحتى', 'وإذا',
+    'وهذا', 'وهذه', 'وذلك', 'وتلك', 'وهو', 'وهي', 'ونحن',
+    'وكان', 'وكانت', 'وكل', 'وبعض', 'وحول', 'وعند', 'وقد',
+    'ولقد', 'ولم', 'ولن', 'وما', 'وكيف', 'واين', 'والذي',
+    'وبين', 'وقبل', 'وبعد', 'وله', 'ولها', 'وهناك', 'وانه',
+    'ولدى', 'وعليه', 'ومثل',
+
+    # Arabic numbers
+    'واحد', 'اثنان', 'ثلاثة', 'أربعة', 'خمسة', 'ستة', 'سبعة', 
+    'ثمانية', 'تسعة', 'عشرة',
+
+    # Arabic ordinals
+    'الأول', 'الثاني', 'الثالث', 'الرابع', 'الخامس', 'السادس', 
+    'السابع', 'الثامن', 'التاسع', 'العاشر'
 }
 
 
+
+
 COUNTRY_MAPPING = {
     'مصر': 'Egypt',
     'السعودية': 'Saudi Arabia',
@@ -141,9 +172,17 @@ def create_topic_map(summaries):
     for summary in summaries:
         country_en = COUNTRY_MAPPING.get(summary['country'])
         if country_en and country_en in coordinates:
-            # Get dominant sentiment
+            # Mapping emotions back to their original labels
+            REVERSE_EMOTION_LABELS = {
+                'Negative': 'LABEL_0',
+                'Positive': 'LABEL_1',
+                'Neutral': 'LABEL_2'
+            }
+            
+            # Get dominant sentiment and map it back to original label
             dominant_emotion = summary['top_emotions'][0]['emotion'] if summary['top_emotions'] else "Neutral"
-            circle_color = sentiment_colors.get(dominant_emotion, 'gray')
+            dominant_label = REVERSE_EMOTION_LABELS.get(dominant_emotion, 'LABEL_2')
+            circle_color = sentiment_colors.get(dominant_label, 'gray')
             
             # Create popup content
             popup_content = f"""