bugs fix and UI updates

.streamlit/config.toml

@@ -0,0 +1,4 @@
+[theme]
+backgroundColor="#0a3144"
+secondaryBackgroundColor="#0f4d60"
+textColor="#dedede"

README.md

@@ -4,9 +4,11 @@ emoji: ✉️
 colorFrom: purple
 colorTo: yellow
 sdk: streamlit
-sdk_version: 1.10.0
+sdk_version: 1.25.0
+python_version: 3.9
 app_file: app.py
 pinned: false
+duplicated_from: loxzdigital/Model-CC-Space
 ---
 
 Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference

app.py

@@ -14,6 +14,7 @@ import joblib
 from bokeh.models.widgets import Div
 
 import email
+import os
 #from ipyfilechooser import FileChooser
 
 #from IPython.display import display
@@ -22,7 +23,10 @@ from bs4 import BeautifulSoup
 import matplotlib.pyplot as plt
 import numpy as np
 import timeit
+import shutil
 
+CURRENT_THEME = "blue"
+IS_DARK_THEME = True
 
 def table_data():
     # creating table data
@@ -179,7 +183,7 @@ def add_bg_from_url():
         f"""
          <style>
          .stApp {{
-             background-image: linear-gradient(#45eff5,#1C8D99);
+             background-image: linear-gradient(135deg,#061c2c,#084e69 35%,#3e7e89);
              background-attachment: fixed;
              background-size: cover
             
@@ -262,6 +266,10 @@ target_variables = [
 uploaded_file = st.file_uploader(
     "Please upload your email (In HTML Format)", type=["html"])
 
+def save_file(uploaded_file):
+    with open(os.path.join("./",uploaded_file.name),"wb") as f:
+        f.write(uploaded_file.getbuffer())
+
 if uploaded_file is None:
     # upload_img = PIL.Image.open(uploaded_file)
     upload_img = None
@@ -298,7 +306,6 @@ st.markdown("""---""")
 def get_files_from_aws(bucket, prefix):
     """
         get files from aws s3 bucket
-
     bucket (STRING): bucket name
     prefix (STRING): file location in s3 bucket
     """
@@ -323,6 +330,7 @@ if st.button('Generate Predictions'):
     if uploaded_file is None:
         st.error('Please upload a email (HTML format)')
     else:
+        save_file(uploaded_file)
         placeholder = st.empty()
         placeholder.text('Loading Data')
 
@@ -488,5 +496,4 @@ if st.button('Generate Predictions'):
                 
 
         placeholder.empty()
-        #st.write(time.time() - start_time)
-
+        #st.write(time.time() - start_time)

requirements.txt

@@ -1,11 +1,14 @@
+altair<5
+pickle5
+numpy
+pandas
+scikit-learn==0.24.1
+Cython==0.29.36
 bokeh==2.4.1
 joblib
 boto3
-pandas
 urlextract
-scikit-learn==0.24.1
 bs4
 matplotlib
-numpy
 plotly
-streamlit==1.21.0
+streamlit==1.25.0