Update app.py

app.py

@@ -38,7 +38,7 @@ ARABIC_STOP_WORDS = {
     'منذ', 'نفس', 'حيث', 'هناك', 'جدا', 'ذات', 'ضمن', 'انه', 'لدى',
     'عليه', 'مثل', 'أما', 'لدي', 'فيه', 'كلم', 'لكن', 'ايضا', 'لازم',
      'يجب', 'صار', 'صارت', 'ضد', 'يا', 'لا', 'اما',
-    'بها', 'ان', 'به', 'الي', 'لما', 'انا', 'اليك', 'لي', 'لك','اذا','بلا','او','لديك','لديه', 'قلت',
+    'بها', 'ان', 'به', 'الي', 'لما', 'انا', 'اليك', 'لي', 'لك','اذا','بلا','او','لديك','لديه','اني','كنت','ليس','ايها', 'قلت',
 
     # Middle group prefixed with "و"
     'وثم', 'وأو', 'ول', 'وب', 'وك', 'ولل', 'وال', 
@@ -163,6 +163,9 @@ def get_country_coordinates():
         'Mauritania': [21.0079, -10.9408]
     }
 def create_topic_map(summaries):
+    # Debug print to check incoming data
+    print("DEBUG - First summary emotions:", summaries[0]['top_emotions'])
+    
     coordinates = get_country_coordinates()
     m = folium.Map(location=[27.0, 42.0], zoom_start=5)
     
@@ -176,18 +179,21 @@ def create_topic_map(summaries):
     for summary in summaries:
         country_en = COUNTRY_MAPPING.get(summary['country'])
         if country_en and country_en in coordinates:
-            # Mapping emotions back to their original labels
+            # Direct label mapping (no conversion needed)
             REVERSE_EMOTION_LABELS = {
-                'Negative': 'LABEL_0',
-                'Positive': 'LABEL_1',
-                'Neutral': 'LABEL_2'
+                'LABEL_0': 'LABEL_0',
+                'LABEL_1': 'LABEL_1',
+                'LABEL_2': '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"
+            # Get dominant sentiment
+            dominant_emotion = summary['top_emotions'][0]['emotion'] if summary['top_emotions'] else "LABEL_2"
             dominant_label = REVERSE_EMOTION_LABELS.get(dominant_emotion, 'LABEL_2')
             circle_color = sentiment_colors.get(dominant_label, 'gray')
             
+            # Debug print
+            print(f"DEBUG - Country: {country_en}, Emotion: {dominant_emotion}, Label: {dominant_label}, Color: {circle_color}")
+            
             # Create popup content
             popup_content = f"""
                 <b>{country_en}</b><br>
@@ -206,7 +212,7 @@ def create_topic_map(summaries):
                 color=circle_color,
                 fill=True
             ).add_to(m)
-            
+    
     # Add legend
     legend_html = """
     <div style="position: fixed; bottom: 50px; left: 50px; z-index: 1000; background-color: white; padding: 10px; border: 2px solid grey; border-radius: 5px">
@@ -220,6 +226,7 @@ def create_topic_map(summaries):
     
     return m
 
+
 def create_arabic_wordcloud(text, title):
     wordcloud = WordCloud(
         width=1200,