Update app.py

app.py

@@ -1,30 +1,145 @@
-import awesome_streamlit as ast
 import streamlit as st
-from home import write as write_home
-from pos_tagger import write as write_pos_tagger
-
-st.set_page_config(
-    page_title="Hamshetsnag POS Tagger",
-    page_icon="📚",
-    layout='wide'
-)
-
-PAGES = {
-    "Home": write_home,
-    "POS Tagging": write_pos_tagger,
-}
-
-st.sidebar.title("Navigation")
-selection = st.sidebar.radio("Pages", list(PAGES.keys()))
-
-page = PAGES[selection]
-# with st.spinner(f"Loading {selection} ..."):
-ast.shared.components.write_page(page)
-
-st.sidebar.header("Info")
-st.sidebar.write(
-    "Models available on [HF Hub](https://huggingface.co/onurkeles/hamshetsnag-pos-tagger)"
-)
-st.sidebar.write(
-    "Model source code will be soon available on [GitHub](http://github.com/kelesonur)"
-)
+from flair.data import Sentence
+from flair.models import SequenceTagger
+import os
+
+# Load the Flair model
+model_path = "onurkeles/hamshetsnag-pos-tagger"
+pos_tagger = SequenceTagger.load(model_path)
+
+def tag_pos(text, detailed_output):
+    """Tag parts of speech in a given text, with optional detailed output."""
+    sentence = Sentence(text)
+    pos_tagger.predict(sentence)
+
+    if detailed_output:
+        # Generate detailed information with tag values and probabilities
+        return "\n".join(
+            [f"{token.text}: {token.get_tag('pos').value} ({token.get_tag('pos').score:.2f})" for token in sentence]
+        )
+    else:
+        # Return a simple tagged string
+        return sentence.to_tagged_string()
+
+def home_page():
+    """Displays the home page content."""
+    st.markdown(
+        """
+        <h1 style="text-align:left;">Hamshetsnag POS Tagger</h1>
+        """,
+        unsafe_allow_html=True,
+    )
+
+    st.write("#")
+
+    col = st.columns(2)
+
+    # Include an image/logo for branding
+    col[0].image("images/cauldron.jpg", width=100)
+
+    st.markdown(
+        """
+        <h3 style="text-align:left;">A POS Tagging Tool for the Hamshetsnag Language</h3>
+        """, unsafe_allow_html=True,
+    )
+
+    st.markdown(
+        """
+        **Features:**
+
+        - Tags parts of speech in Hamshetsnag text.
+        - Utilizes a fine-tuned model specifically for this language.
+        - Configurable parameters for customizing tagging behavior.
+        
+        Refer to the [paper](http://kelesonur.github.io/folder/hms_abstract.pdf) for more details.
+        
+        ### Citation:
+        ```bibtex
+        @inproceedings{keles2024hamshetsnag,
+            title={Transfer Learning to the Rescue! Cross-Lingual Transfer for POS Tagging in an Endangered Language: Hamshetsnag},
+            author={Onur Keleş and Baran Günay},
+            booktitle={Turkey Computational Social Science Conference},
+            year={2024},
+            address={Koç University: Istanbul, Turkey}
+        }
+        ```
+        """
+    )
+
+    st.markdown(
+        """
+        <p style="text-align:right;"><em>Please use this tool responsibly and double-check all outputs.</em></p>
+        """,
+        unsafe_allow_html=True,
+    )
+
+def pos_tagging_page():
+    """Displays the POS tagging page content."""
+    st.markdown("# Part-of-Speech Tagging for Hamshetsnag")
+    st.sidebar.header("POS Tagging")
+
+    st.write(
+        '''Detect parts of speech in Hamshetsnag text using the fine-tuned model.'''
+    )
+
+    # Sidebar for configurations
+    st.sidebar.subheader("Configurable Parameters")
+
+    # Detailed Output Checkbox
+    detailed_output = st.sidebar.checkbox(
+        "Detailed Output",
+        value=False,
+        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)
+
+def main():
+    st.set_page_config(
+        page_title="Hamshetsnag POS Tagger",
+        page_icon="📚",
+        layout='wide'
+    )
+
+    # Dictionary mapping pages
+    PAGES = {
+        "Home": home_page,
+        "POS Tagging": pos_tagging_page,
+    }
+
+    st.sidebar.title("Navigation")
+    selection = st.sidebar.radio("Pages", list(PAGES.keys()))
+
+    # Load and display the selected page
+    PAGES[selection]()
+
+    st.sidebar.header("Info")
+    st.sidebar.write(
+        "Models available on [HF Hub](https://huggingface.co/onurkeles/hamshetsnag-pos-tagger)"
+    )
+    st.sidebar.write(
+        "Model source code available on [GitHub](http://github.com/kelesonur)"
+    )
+
+if __name__ == "__main__":
+    main()