Add displaying selected domain and text type to polemo2 dataset

app.py

@@ -115,6 +115,33 @@ with dataframe_head:
     st.dataframe(df_to_show)
     st.text_area(label="Latex code", value=df_to_show.style.to_latex())
 
+    if selected_dataset == "clarin-pl/polemo2-official":
+        st.subheader("First 10 observations of a chosen domain and text type")
+        domain = st.selectbox(
+            label="Select domain",
+            options=["all", "hotels", "medicine", "products", "reviews"],
+        )
+        text_type = st.selectbox(
+            label="Select text type", options=["Full text", "Tokenized to sentences"]
+        )
+        text_type_mapping_dict = {
+            "Full text": "text",
+            "Tokenized to sentences": "sentence",
+        }
+
+        polemo_subset = load_dataset(
+            selected_dataset, f"{domain}_{text_type_mapping_dict[text_type]}"
+        )
+        df = pd.concat(
+            [
+                polemo_subset["train"].to_pandas(),
+                polemo_subset["validation"].to_pandas(),
+                polemo_subset["test"].to_pandas(),
+            ]
+        ).head(10)
+        st.dataframe(df)
+        st.text_area(label="Latex code", value=df.style.to_latex())
+
 with word_searching:
     st.header("Observations containing a chosen word")
     searched_word = st.text_input(label="Enter the word you are looking for below")