Update app.py

app.py

@@ -95,38 +95,45 @@ def get_cached_embeddings(text, tokenizer, model):
 
 def create_theme_map(summaries, topic_model):
     """Create an interactive map showing theme distributions across countries"""
-    # Create a base map centered on the Arab world
-    m = folium.Map(location=[25, 45], zoom_start=4)
-    
-    # Convert country names to coordinates
-    cc = coco.CountryConverter()
-    
-    for summary in summaries:
-        try:
-            # Get country coordinates
-            country_iso = cc.convert(names=[summary['country']], to='ISO2')
-            country_data = cc.convert(names=[summary['country']], to='name_short')
-            
-            # Create popup content with theme information
-            popup_content = f"""
-                <h4>{summary['country']}</h4>
-                <b>Top Themes:</b><br>
-                {'<br>'.join([f"• {topic['topic']}: {topic['count']}" 
-                             for topic in summary['top_topics'][:5]])}
-            """
-            
-            # Add marker for each country
-            folium.CircleMarker(
-                location=[cc.convert(country_iso, to='latitude')[0],
-                         cc.convert(country_iso, to='longitude')[0]],
-                radius=20,
-                popup=folium.Popup(popup_content, max_width=300),
-                color='red',
-                fill=True,
-                fill_opacity=0.7
-            ).add_to(m)
-            
-    return m
+    try:
+        # Create a base map centered on the Arab world
+        m = folium.Map(location=[25, 45], zoom_start=4)
+        
+        # Convert country names to coordinates
+        cc = coco.CountryConverter()
+        
+        for summary in summaries:
+            try:
+                # Get country coordinates
+                country_iso = cc.convert(names=[summary['country']], to='ISO2')
+                country_data = cc.convert(names=[summary['country']], to='name_short')
+                
+                # Create popup content with theme information
+                popup_content = f"""
+                    <h4>{summary['country']}</h4>
+                    <b>Top Themes:</b><br>
+                    {'<br>'.join([f"• {topic['topic']}: {topic['count']}" 
+                                 for topic in summary['top_topics'][:5]])}
+                """
+                
+                # Add marker for each country
+                folium.CircleMarker(
+                    location=[cc.convert(country_iso, to='latitude')[0],
+                             cc.convert(country_iso, to='longitude')[0]],
+                    radius=20,
+                    popup=folium.Popup(popup_content, max_width=300),
+                    color='red',
+                    fill=True,
+                    fill_opacity=0.7
+                ).add_to(m)
+            except Exception as e:
+                st.warning(f"Could not process {summary['country']}: {str(e)}")
+                continue
+                
+        return m
+    except Exception as e:
+        st.error(f"Error creating map: {str(e)}")
+        return None
 
 def split_text(text, max_length=512):
     """Split text into chunks of maximum token length while preserving word boundaries."""