Update app.py

app.py

@@ -125,35 +125,52 @@ def get_country_coordinates():
         'Mauritania': [21.0079, -10.9408]
     }
 def create_topic_map(summaries):
-    """Create an interactive map showing topic distribution"""
     coordinates = get_country_coordinates()
-    
-    # Create base map centered on Arab world
     m = folium.Map(location=[25.0, 30.0], zoom_start=4)
     
+    # Color mapping for sentiments
+    sentiment_colors = {
+        'Positive': 'green',
+        'Negative': 'red',
+        'Neutral': 'blue'
+    }
+    
     for summary in summaries:
-        country = summary['country']
-        if country in coordinates:
-            # Get top topic
-            top_topic = summary['top_topics'][0]['topic'] if summary['top_topics'] else "No topics"
-            top_emotion = summary['top_emotions'][0]['emotion'] if summary['top_emotions'] else "No emotion"
+        country_en = COUNTRY_MAPPING.get(summary['country'])
+        if country_en and country_en in coordinates:
+            # Get dominant sentiment
+            dominant_emotion = summary['top_emotions'][0]['emotion'] if summary['top_emotions'] else "Neutral"
+            circle_color = sentiment_colors.get(dominant_emotion, 'gray')
             
             # Create popup content
             popup_content = f"""
-                <b>{country}</b><br>
-                Top Topic: {top_topic}<br>
-                Main Emotion: {top_emotion}<br>
+                <b>{country_en}</b><br>
+                <b>Sentiment Distribution:</b><br>
+                {'<br>'.join(f"• {e['emotion']}: {e['count']}" for e in summary['top_emotions'][:3])}<br>
+                <b>Top Topic:</b><br>
+                {summary['top_topics'][0]['topic'] if summary['top_topics'] else 'No topics'}<br>
                 Total Poems: {summary['total_poems']}
             """
             
             # Add marker
             folium.CircleMarker(
-                location=coordinates[country],
+                location=coordinates[country_en],
                 radius=10,
                 popup=folium.Popup(popup_content, max_width=300),
-                color='red',
+                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">
+    <p><b>Sentiment:</b></p>
+    <p><span style="color: green;">●</span> Positive</p>
+    <p><span style="color: red;">●</span> Negative</p>
+    <p><span style="color: blue;">●</span> Neutral</p>
+    </div>
+    """
+    m.get_root().html.add_child(folium.Element(legend_html))
     
     return m
 
@@ -514,9 +531,8 @@ if uploaded_file is not None:
                             st.write(f"• Topic {row['Topic']}: {topic_name} ({row['Count']} poems)")
 
                     with tab3:
-                        st.subheader("Topic Distribution Map")
+                        st.subheader("Topic and Sentiment Distribution Map")
                         topic_map = create_topic_map(summaries)
-                        # Display the map
                         st.components.v1.html(topic_map._repr_html_(), height=600)
     
     except Exception as e: