Update app.py

app.py

@@ -31,9 +31,11 @@ def predict_class(text):
         st.error(f"Error during prediction: {e}")
         return None
 
+
 uploaded_files_dir = "uploaded_files"
 os.makedirs(uploaded_files_dir, exist_ok=True)
 
+
 class_colors = {
     0: "#d62728",  # Level 1
     1: "#ff7f0e",  # Level 2
@@ -47,6 +49,8 @@ st.image("logo2.png", width=70)
 
 st.markdown('<div style="position: absolute; top: 5px; left: 5px;"></div>', unsafe_allow_html=True)
 
+# col1, col2 = st.columns([1, 3])
+
 st.title("Paper Citation Classifier")
 
 option = st.radio("Select input type:", ("Text", "PDF"))
@@ -60,10 +64,12 @@ if option == "Text":
     
     categories = pills("Select WoS category", options)
     
+    # categories = st.multiselect("Select WoS categories:", options)
+
     combined_text = f"{title_input} [SEP] {abstract_input} [SEP] {full_text_input} [SEP] {affiliations_input} [SEP] {' [SEP] '.join(categories)}"
 
     if st.button("Predict"):
-        if not combined_text.strip():  # Check if combined_text is empty or contains only whitespace
+        if not any([title_input, abstract_input, full_text_input, affiliations_input]):
             st.warning("Please enter paper text.")
         else:
             with st.spinner("Predicting..."):
@@ -97,8 +103,8 @@ elif option == "PDF":
             st.text(file_text)
 
             if st.button("Predict from PDF Text"):
-                if not file_text.strip():  # Check if file_text is empty or contains only whitespace
-                    st.warning("Please upload a PDF file with text content.")
+                if not file_text.strip():
+                    st.warning("Please upload a PDF with text content.")
                 else:
                     with st.spinner("Predicting..."):
                         predicted_class = predict_class(file_text)