change some details

.idea/.gitignore

@@ -0,0 +1,8 @@
+# Default ignored files
+/shelf/
+/workspace.xml
+# Editor-based HTTP Client requests
+/httpRequests/
+# Datasource local storage ignored files
+/dataSources/
+/dataSources.local.xml

.idea/.name

@@ -0,0 +1 @@
+scattertext_funtion.py

.idea/huggingface.iml

@@ -0,0 +1,8 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<module type="PYTHON_MODULE" version="4">
+  <component name="NewModuleRootManager">
+    <content url="file://$MODULE_DIR$" />
+    <orderEntry type="inheritedJdk" />
+    <orderEntry type="sourceFolder" forTests="false" />
+  </component>
+</module>

.idea/inspectionProfiles/profiles_settings.xml

@@ -0,0 +1,6 @@
+<component name="InspectionProjectProfileManager">
+  <settings>
+    <option name="USE_PROJECT_PROFILE" value="false" />
+    <version value="1.0" />
+  </settings>
+</component>

.idea/misc.xml

@@ -0,0 +1,7 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<project version="4">
+  <component name="Black">
+    <option name="sdkName" value="Python 3.9" />
+  </component>
+  <component name="ProjectRootManager" version="2" project-jdk-name="Python 3.9" project-jdk-type="Python SDK" />
+</project>

.idea/modules.xml

@@ -0,0 +1,8 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<project version="4">
+  <component name="ProjectModuleManager">
+    <modules>
+      <module fileurl="file://$PROJECT_DIR$/.idea/huggingface.iml" filepath="$PROJECT_DIR$/.idea/huggingface.iml" />
+    </modules>
+  </component>
+</project>

.idea/vcs.xml

@@ -0,0 +1,7 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<project version="4">
+  <component name="VcsDirectoryMappings">
+    <mapping directory="" vcs="Git" />
+    <mapping directory="$PROJECT_DIR$/scattertext" vcs="Git" />
+  </component>
+</project>

scattertext_funtion.py

@@ -20,7 +20,8 @@ uploaded_file = st.file_uploader("Upload your text document", type=["csv", "txt"
 # read data
 if uploaded_file is not None:
     # choose function
-    function_choice = st.selectbox('Choose file source', ('Choose...', 'Customized', 'Download from Online Databases'))
+    function_choice = st.selectbox('Choose file source', ['Choose...', 'Customized', 'Download from Online Databases'])
+
     # function1: generate plot from customized file
     if st.button('Customized'):
         # proceed data
@@ -29,7 +30,7 @@ if uploaded_file is not None:
         elif uploaded_file.name.endswith(".txt"):
             df = pd.read_table(uploaded_file, sep='\t')  # TODO : doc: assume contents are seperated by Tabs.
 
-            chosen_column = st.selectbox("Choose text column for analysis", df.head())
+            chosen_column = st.selectbox("Choose text column for analysis", df.columns)
             # convert to scattertext corpus
             corpus = stx.CorpusFromPandas(
                 df,
@@ -39,10 +40,8 @@ if uploaded_file is not None:
                 ). build()
             # create visualization
             # customize parameters
-            input_category_name = input('Enter the category name')
-            customize_category_name = st.text_input('Customize parameter', input_category_name)
-            input_non_category_name = input('Enter the non-category name')
-            customize_non_category_name = st.text_input('Customize parameter', input_non_category_name)
+            customize_category_name = st.text_input('Enter the category name')
+            customize_non_category_name = st.text_input('Enter the non-category name')
             html = stx.produce_scattertext_explorer(corpus,
                                        category=chosen_column,
                                        category_name=customize_category_name,
@@ -59,7 +58,7 @@ if uploaded_file is not None:
         # scopus & lens
         if uploaded_file.name.endswith(".csv"):
             df = pd.read_csv(uploaded_file)
-            chosen_column = st.selectbox("Choose text column for analysis BESIDES ABSTRACT", df.head())
+            chosen_column = st.selectbox("Choose text column for analysis BESIDES ABSTRACT", df.columns)
             if chosen_column == 'Abstract':
                 st.write("This column cannot be selected, please select again")
             else: