Update app.py

app.py

@@ -1,4 +1,5 @@
 import streamlit as st
+
 def main():
     st.title("Amazon Title Suggestion")
     if "title" not in st.session_state:
@@ -11,46 +12,57 @@ def main():
         st.session_state.submitted_title = False
     if "submitted_ner_keywords" not in st.session_state:
         st.session_state.submitted_ner_keywords = False
+    
     if not st.session_state.submitted_title:
         submit_title()
-    if st.session_state.submitted_title and not st.session_state.submitted_ner_keywords:
+    elif st.session_state.submitted_title and not st.session_state.submitted_ner_keywords:
         submit_ner_keywords()
+
 def ner_title(title):
-  word_list = title.split()
-  indexed_dict = {index: word for index, word in enumerate(word_list)}
-  return indexed_dict
+    word_list = title.split()
+    indexed_dict = {index: word for index, word in enumerate(word_list)}
+    return indexed_dict
+
 def submit_title():
     title = st.text_input("Enter Product Title:")
     if st.button("Submit Title"):
         st.session_state.title = title
-        ner=ner_title(title)
+        ner = ner_title(title)
         st.session_state.submitted_title = True
-        st.session_state.ner_dict = ner#{"Feature": "Design", "Price": "Affordable"}
-        st.session_state.selected_keywords = []
+        st.session_state.ner_dict = ner
+
 def submit_ner_keywords():
     st.subheader("Product Features:")
     selected_features = []
     for key, value in st.session_state.ner_dict.items():
         if st.checkbox(f"{key}: {value}"):
             selected_features.append(value)
-    # st.subheader("Product Features:")
-    # for key, value in st.session_state.ner_dict.items():
-    #     #st.session_state.selected_ner[key] = st.checkbox(f"{key}: {value}")
-    #     st.checkbox(f"{key}: {value}")
+    
     st.subheader("Select Search Terms:")
-    keyword_list = ['a','b','c','f','g',"Feature", "Price", "Quality", "Availability"]
+    keyword_list = ['a', 'b', 'c', 'f', 'g', "Feature", "Price", "Quality", "Availability"]
     for keyword in keyword_list:
         st.checkbox(keyword, key=keyword)
+    
     if st.button("Suggest Titles"):
         model2_keywords = [keyword for keyword in keyword_list if st.session_state[keyword]]
         st.session_state.selected_keywords = model2_keywords
         st.session_state.submitted_ner_keywords = True
         st.write("Selected Keywords for Model2:", model2_keywords)
         st.write("Selected features for Model2:", selected_features)
+    
     if st.button("Reset"):
         st.session_state.title = ""
         st.session_state.submitted_title = False
         st.session_state.submitted_ner_keywords = False
         st.experimental_rerun()
+
+    if st.button("Reset All"):
+        st.session_state.title = ""
+        st.session_state.ner_dict = {}
+        st.session_state.selected_keywords = []
+        st.session_state.submitted_title = False
+        st.session_state.submitted_ner_keywords = False
+        st.experimental_rerun()
+
 if __name__ == "__main__":
-    main()
+    main()