src/visualization/visualize.py

import streamlit as st
import yaml

from src.models import predict_model


def visualize():
    st.write("# Summarization  UI")
    st.markdown(
        """
        *For additional questions and inquiries, please contact **Gagan Bhatia** via [LinkedIn](
        https://www.linkedin.com/in/gbhatia30/) or [Github](https://github.com/gagan3012).*
        """
    )

    text = st.text_area("Enter text here")
    if st.button("Generate Summary"):
        with st.spinner("Connecting the Dots..."):
            sumtext = predict_model(text=text)
        st.write("# Generated Summary:")
        st.write("{}".format(sumtext))
        with open("reports/visualization_metrics.txt", "w") as file1:
            file1.writelines(text)
            file1.writelines(sumtext)


if __name__ == "__main__":
    with open("params.yml") as f:
        params = yaml.safe_load(f)

    if params["visualise"]:
        visualize()