Update app.py

app.py

@@ -505,6 +505,8 @@ if uploaded_file is not None:
         
         df['country'] = df['country'].str.strip()
         df = df.dropna(subset=['country', 'poem'])
+        # Extract 10 rows per country
+        sampled_df = df.groupby('country').apply(lambda x: x.head(10)).reset_index(drop=True)
         
         st.subheader("Topic Modeling Settings")
         col1, col2 = st.columns(2)
@@ -550,7 +552,7 @@ if uploaded_file is not None:
 
 
                 summaries, topic_model = process_and_summarize(
-                    df,
+                    sampled_df,
                     bert_tokenizer,
                     bert_model,
                     emotion_classifier,