Update app.py

app.py

@@ -224,24 +224,42 @@ if button:
                 client_name = metadata.get('client', 'Unknown Client')
                 start_year = metadata.get('start_year', None)
                 end_year_ = metadata.get('end_year', None)
-    
+                
                 try:
                     c_list = json.loads(countries.replace("'", '"'))
-                    country_names = [
-                        get_country_name(code.upper(), region_df) 
-                        for code in c_list 
-                        if len(code) == 2
-                    ]
                 except json.JSONDecodeError:
-                    country_names = []
-    
+                    c_list = []
+                
+                # Only keep country names if the region lookup (get_country_name) 
+                # returns something different than the raw code.
+                matched_countries = []
+                for code in c_list:
+                    if len(code) == 2:
+                        resolved_name = get_country_name(code.upper(), region_df)
+                        # If get_country_name didn't find a match, 
+                        # it typically just returns the same code (like "XX"). 
+                        # We'll consider "successfully looked up" if 
+                        # resolved_name != code.upper().
+                        if resolved_name.upper() != code.upper():
+                            matched_countries.append(resolved_name)
+                
+                # Format the year range
                 start_year_str = f"{int(round(float(start_year)))}" if start_year else "Unknown"
                 end_year_str = f"{int(round(float(end_year_)))}" if end_year_ else "Unknown"
-    
-                additional_text = (
-                    f"**{', '.join(country_names)}**, commissioned by **{client_name}**, "
-                    f"**{start_year_str}-{end_year_str}**"
-                )
+                
+                # Build the final string
+                if matched_countries:
+                    # We have at least 1 valid country name
+                    additional_text = (
+                        f"**{', '.join(matched_countries)}**, commissioned by **{client_name}**, "
+                        f"**{start_year_str}-{end_year_str}**"
+                    )
+                else:
+                    # No valid countries found
+                    additional_text = (
+                        f"Commissioned by **{client_name}**, **{start_year_str}-{end_year_str}**"
+                    )
+                
                 st.markdown(additional_text)
                 st.divider()
     
@@ -276,19 +294,42 @@ if button:
                 client_name = metadata.get('client', 'Unknown Client')
                 start_year = metadata.get('start_year', None)
                 end_year_ = metadata.get('end_year', None)
-    
+                
                 try:
                     c_list = json.loads(countries.replace("'", '"'))
-                    country_names = [get_country_name(code.upper(), region_df) for code in c_list if len(code) == 2]
                 except json.JSONDecodeError:
-                    country_names = []
-    
+                    c_list = []
+                
+                # Only keep country names if the region lookup (get_country_name) 
+                # returns something different than the raw code.
+                matched_countries = []
+                for code in c_list:
+                    if len(code) == 2:
+                        resolved_name = get_country_name(code.upper(), region_df)
+                        # If get_country_name didn't find a match, 
+                        # it typically just returns the same code (like "XX"). 
+                        # We'll consider "successfully looked up" if 
+                        # resolved_name != code.upper().
+                        if resolved_name.upper() != code.upper():
+                            matched_countries.append(resolved_name)
+                
+                # Format the year range
                 start_year_str = f"{int(round(float(start_year)))}" if start_year else "Unknown"
                 end_year_str = f"{int(round(float(end_year_)))}" if end_year_ else "Unknown"
-    
-                additional_text = (
-                    f"**{', '.join(country_names)}**, commissioned by **{client_name}**, **{start_year_str}-{end_year_str}**"
-                )
+                
+                # Build the final string
+                if matched_countries:
+                    # We have at least 1 valid country name
+                    additional_text = (
+                        f"**{', '.join(matched_countries)}**, commissioned by **{client_name}**, "
+                        f"**{start_year_str}-{end_year_str}**"
+                    )
+                else:
+                    # No valid countries found
+                    additional_text = (
+                        f"Commissioned by **{client_name}**, **{start_year_str}-{end_year_str}**"
+                    )
+                
                 st.markdown(additional_text)
                 st.divider()