Update app.py

app.py

@@ -1,145 +1,30 @@
+import awesome_streamlit as ast
 import streamlit as st
-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()
+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)"
+)