Update app.py

app.py

@@ -233,8 +233,9 @@ def process_and_compare(file1, sheet1, file2, sheet2):
     file_path = 'output/plot.png'
     plt.savefig(file_path, format='png', bbox_inches='tight')
     plt.close()
-
-    return file_path, gr.update(choices=stored_df1.Country.values.tolist()), gr.update(choices=stored_df2.Country.values.tolist())
+    filtered_countries1 = [country for country in countries_list if len(stored_df1.Country.values.tolist()) < 20]
+    filtered_countries2 = [country for country in countries_list if len(stored_df2.Country.values.tolist()) < 20]
+    return file_path, gr.update(choices=filtered_countries1), gr.update(choices=filtered_countries2)
 
 def find_sentences_with_keywords(text, keywords):
     # Split text into sentences using regular expression to match sentence-ending punctuation