Create 2_🌍_trainui.py

pages/2_🌍_trainui.py

@@ -0,0 +1,34 @@
+import streamlit as st
+import pandas as pd
+import time
+from streamlit_extras.streaming_write import write
+
+
+def stream_example():
+    for word in data.split():
+        yield word + " "
+        time.sleep(0.5)
+
+
+st.set_page_config(layout="wide", page_title="Amazon Review | Trainer")
+st.title("Amazon Review  Trainer-Just a mock UI ")
+
+
+data_upload = st.file_uploader(label="Train File", type=["xlsx", "csv", "xls"])
+
+col1, col2 = st.columns([1, 1])
+with col1:
+    theme = st.checkbox(label="Theme")
+with col2:
+    subtheme = st.checkbox(label="Subtheme")
+
+
+if data_upload is not None:
+    if data_upload.name.split(".")[-1] in ["xls", "xlsx"]:
+        df = pd.read_excel(io=data_upload, engine="openpyxl")
+    if data_upload.name.split(".")[-1] in ["csv"]:
+        df = pd.read_csv(filepath_or_buffer=data_upload)
+    df.columns = [x.lower() for x in df.columns.tolist()]
+    data = ",".join(df.columns.values.tolist())
+    # st.write(data)
+    write(stream_example)