Create app.py

app.py

@@ -0,0 +1,56 @@
+import streamlit as st
+def main():
+    st.title("Amazon Title Suggestion")
+    if "title" not in st.session_state:
+        st.session_state.title = ""
+    if "ner_dict" not in st.session_state:
+        st.session_state.ner_dict = {}
+    if "selected_keywords" not in st.session_state:
+        st.session_state.selected_keywords = []
+    if "submitted_title" not in st.session_state:
+        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:
+        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
+def submit_title():
+    title = st.text_input("Enter Product Title:")
+    if st.button("Submit Title"):
+        st.session_state.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 = []
+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"]
+    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 __name__ == "__main__":
+    main()