src/visualization/visualize.py

import streamlit as st
import os
retval = os.getcwd()
os.chdir("src/")
from src.models.predict_model 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).*
        """
    )
    st.write(retval)

    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))


if __name__ == "__main__":
    visualize()