added flags

modules/result_table.py

@@ -15,6 +15,7 @@ def show_table(p1_df, p2_df):
             row_from_p1 = p1_df.iloc[[i]]
             row_from_p2 = p2_df.iloc[[i]]
 
+            # INTEGRATE IN PREPROCESSING !!!
             # transform strings to list
             row_from_p1["crs_3_code_list"] = [row_from_p1['crs_3_code'].item().split(";")[:-1]]
             row_from_p2["crs_3_code_list"] = [row_from_p2['crs_3_code'].item().split(";")[:-1]]
@@ -25,6 +26,15 @@ def show_table(p1_df, p2_df):
             row_from_p1["sdg_list"] = [row_from_p1['sgd_pred_code'].item()]
             row_from_p2["sdg_list"] = [row_from_p2['sgd_pred_code'].item()]
 
+            try:
+                row_from_p1["flag"] = f"https://flagicons.lipis.dev/flags/4x3/{row_from_p1['country'].item()[:2].lower()}.svg"
+                row_from_p2["flag"] = f"https://flagicons.lipis.dev/flags/4x3/{row_from_p2['country'].item()[:2].lower()}.svg"
+            except:
+                row_from_p1["flag"] = "https://flagicons.lipis.dev/flags/4x3/xx.svg"
+                row_from_p2["flag"] = "https://flagicons.lipis.dev/flags/4x3/xx.svg"
+            
+            print(row_from_p1["flag"].item())
+
             # Correctly append rows to match_df
             st.subheader(f"#{actual_ind}")
             st.caption(f"Similarity: {round(row_from_p1['similarity'].item(), 4) * 100}%")
@@ -35,58 +45,65 @@ def show_table(p1_df, p2_df):
             #AgGrid(match_df)
 
             st.dataframe(
-                match_df[["iati_id", "title_main", "orga_abbreviation", "client", "description_main", "country", "sdg_list", "crs_3_code_list", "crs_5_code_list"]],
+                match_df[["iati_id", "title_main", "orga_abbreviation", "client", "description_main", "country", "flag", "sdg_list", "crs_3_code_list", "crs_5_code_list"]],
                 use_container_width = True,
                 height = 35 + 35 * len(match_df),
                 column_config={
-                    """
-                    "similarity": st.column_config.TextColumn(
-                        "Similarity",
-                        help="simialrity",
-                        disabled=True
-                    ),
-                    """
                     "iati_id": st.column_config.TextColumn(
                         "IATI ID",
                         help="IATI Project ID",
-                        disabled=True
+                        disabled=True,
+                        width="small"
                     ),
                     "orga_abbreviation": st.column_config.TextColumn(
                         "Organization",
                         help="If description not in English, description in other language provided",
-                        disabled=True
+                        disabled=True,
+                        width="small"
                     ),
                     "client": st.column_config.TextColumn(
                         "Client",
                         help="Client organization of customer",
-                        disabled=True
+                        disabled=True,
+                        width="small"
                     ),
                     "title_main": st.column_config.TextColumn(
                         "Title",
                         help="If title not in English, title in other language provided",
-                        disabled=True
+                        disabled=True,
+                        width="large"
                     ),
                     "description_main": st.column_config.TextColumn(
                         "Description",
                         help="If description not in English, description in other language provided",
-                        disabled=True
+                        disabled=True,
+                        width="large"
                     ),
                     "country": st.column_config.TextColumn(
                         "Country",
                         help="Country of project",
-                        disabled=True
+                        disabled=True,
+                        width="small"
+                    ),
+                    "flag": st.column_config.ImageColumn(
+                        "Flag",
+                        help="country flag",
+                        width="small"
                     ),
                     "sdg_list": st.column_config.ListColumn(
                         "SDG Prediction",
                         help="Prediction of SDG's",
+                        width="small"
                     ),
                     "crs_3_code_list": st.column_config.ListColumn(
                         "CRS 3",
                         help="CRS 3 code given by organization",
+                        width="small"
                     ),
                     "crs_5_code_list": st.column_config.ListColumn(
                         "CRS 5",
-                        help="CRS 5 code given by organization"
+                        help="CRS 5 code given by organization",
+                        width="small"
                     ),
                 },
                 hide_index=True,