Update app.py

app.py

@@ -50,15 +50,16 @@ option = st.radio("Select input type:", ("Text", "PDF"))
 
 if option == "Text":
     # Input text boxes for abstract, full text, and affiliations
+    title_input = st.text_area("Enter Title:")
     abstract_input = st.text_area("Enter Abstract:")
     full_text_input = st.text_area("Enter Full Text:")
     affiliations_input = st.text_area("Enter Affiliations:")
 
     # Select categories using pills
-    categories = st.multiselect("Select categories:", ["Category 1", "Category 2", "Category 3", "Category 4"])
+    categories = st.multiselect("Select WoS categories:", ["Nursing", "Physics", "Maths", "Chemical", "Nuclear" ,"Other"])
 
     # Combine selected categories with [SEP]
-    combined_text = f"{abstract_input} [SEP] {full_text_input} [SEP] {affiliations_input} [SEP] {' [SEP] '.join(categories)}"
+    combined_text = f"{title_input} [SEP] {abstract_input} [SEP] {full_text_input} [SEP] {affiliations_input} [SEP] {' [SEP] '.join(categories)}"
 
     if st.button("Predict"):
         with st.spinner("Predicting..."):