Update pos_tagger.py

pos_tagger.py

@@ -13,9 +13,10 @@ def tag_pos(text, detailed_output):
 
     if detailed_output:
         # Generate detailed information with tag values and probabilities
-        return "\n".join(
-            [f"{token.text}: {token.pos} ({token.get_tag('pos').score:.2f})" for token in sentence]
-        )
+        output = ""
+        for label in sentence.get_labels('pos'):
+            output += f"{label.text}: {label.value} ({label.score:.2f}) "
+        st.success(output.strip())
     else:
         # Return a simple tagged string
         return sentence.to_tagged_string()
@@ -38,24 +39,25 @@ def write():
         help="If checked, output shows detailed tag information (probability scores, etc.).",
     )
 
- # Input field for text
-    input_text = st.text_area(label='Enter a text: ', height=100, value="Put example text here.")
-    
-    # Provide example sentences with translations
-    example_sentences = [
-        "tuute acertsetser topoldetser. aaav ta? (TR: Kâğıdı büzüştürdün attın. Oldu mu?)",
-        "Baran u Baden teran. (TR: Baran ve Bade koştu.)",
-        "Onurun ennush nu İremin terchushe intzi shad kızdırmısh aaav. (TR: Onur'un düşüşü ve İrem'in koşuşu beni kızdırdı.)"
-    ]
-
-    st.write("## Example Sentences:")
-    for example in example_sentences:
-        if st.button(f"Use: {example.split('(TR:')[0].strip()}"):
-            input_text = example.split('(TR:')[0].strip()  # Update the input text directly with the Hamshetsnag part
-            break  # Only use the first clicked example
-
-    if st.button("Tag POS"):
-        with st.spinner('Processing...'):
-            # Tag the input text and format output as per settings
-            output = tag_pos(input_text, detailed_output)
-            st.success(output)
+# Input field for text
+input_text = st.text_area(label='Enter a text: ', height=100, value="Put example text here.")
+state = st.session_state
+
+# Provide example sentences with translations
+example_sentences = [
+    "tuute acertsetser topoldetser. aaav ta? (TR: Kâğıdı büzüştürdün attın. Oldu mu?)",
+    "Baran u Baden teran. (TR: Baran ve Bade koştu.)",
+    "Onurun ennush nu İremin terchushe intzi shad kızdırmısh aaav. (TR: Onur'un düşüşü ve İrem'in koşuşu beni kızdırdı.)"
+]
+
+st.write("## Example Sentences:")
+for example in example_sentences:
+    if st.button(f"Use: {example.split('(TR:')[0].strip()}"):
+        state.input_text = example.split('(TR:')[0].strip()
+        break
+
+if st.button("Tag POS"):
+    with st.spinner('Processing...'):
+        input_text = state.input_text
+        output = tag_pos(input_text, detailed_output)
+        st.success(output)