Upload streamlit_app.py

Update generate_response parameters

streamlit_app.py

@@ -50,13 +50,17 @@ def update_questionnaire_response(user_response, username):
     st.success("Your responses have been recorded. Thank you!")
 
 
-def generate_responses(text, chat_model="gpt-3.5-turbo", paper_title="", sci_familiarity="", tech_usage="",
-                       read_news="", language_spoken="", additional_requirements="None"):
+def generate_responses(text, chat_model="gpt-3.5-turbo", paper_title="", level_education="",
+                       paper_familiarity="", english_proficiency="", language_spoken="", tech_usage="",
+                       news_read="", books_read="", additional_requirements="None"):
     # Incorporating the parameters into the context
     text = text[:2048]
     user_context = f"""
-    The user has {sci_familiarity} with scientific concepts. He/She {tech_usage} uses technology products. 
-    He/She {read_news} read or watch the news. The primary language spoken at his/her home is(are) {language_spoken}.
+    The user has achieved an education level up tp {level_education}. For the given paper content and concept, the user
+    describes his/her understanding as {paper_familiarity}. In daily routine, the user describes the frequency of using
+    technology such as computers, cell phones, and tablets as {tech_usage}. The user's primary language spoken at home is
+    {language_spoken}, and has {english_proficiency} level of English proficiency. The user {news_read} reads or watches
+    the news and reads approximately {books_read} books in a month.
     """
 
     # print(user_context)
@@ -96,6 +100,7 @@ def survey(user_name):
     title = user_name + '_survey'
     return ss.StreamlitSurvey(title)
 
+
 # connect to/create Deta user database
 db_key = st.secrets["deta_key"]
 deta = Deta(db_key)
@@ -118,25 +123,23 @@ init_sidebar = st.sidebar.empty()
 
 with init_sidebar:
     init_page = option_menu(None,
-                       ["Login", 'Sign Up'],
-                       icons=['lightbulb-fill', 'lightbulb'],
-                       menu_icon="cast",
-                       default_index=0,
-                       styles={})
+                            ["Login", 'Sign Up'],
+                            icons=['lightbulb-fill', 'lightbulb'],
+                            menu_icon="cast",
+                            default_index=0,
+                            styles={})
 
 if init_page == 'Login':
-    st.cache_data.clear()
     name, authentication_status, username = authenticator.login('Login', 'main')
     if authentication_status:
         init_sidebar.empty()
+        st.sidebar.write(f'**Welcome** {name}')
         app_sidebar = st.sidebar.empty()
-        # st.sidebar.write(f'Welcome {name}')
+
         if 'current_page_name' not in st.session_state:
             st.session_state.current_page_name = "Generate Plain Language Summary"  # 设置默认页面
 
         with app_sidebar:
-            # 'History'
-            # 'folder2',
             page = option_menu(None, ["Generate Plain Language Summary", 'Questionnaire', 'Setup'],
                                icons=['house', 'question-circle', 'gear'],
                                menu_icon="None",
@@ -207,6 +210,26 @@ if init_page == 'Login':
             else:
                 submit_text = abstract_text
             # Check if the "Submit" button is clicked
+
+            st.write("#### What is your familiarity with the concepts of the paper?")
+            st.markdown('''
+                    * No Familiarity: entirely unfamiliar, no prior knowledge
+                    * Limited Familiarity: basic awareness of the concepts in the paper, but not in-depth knowledge
+                    * Moderate Familiarity: reasonable understanding of the concepts in the paper, encountered before, or some background knowledge
+                    * Good Familiarity: a solid understanding due to prior exposure or study
+                    * Expert: highly knowledgeable and experienced in the field and has worked extensively with these concepts
+                ''')
+            paper_familiarity = st.select_slider(
+                label="paper_familiarity",
+                options=['No Familiarity', 'Limited Familiarity', 'Moderate Familiarity',
+                         'Good Familiarity', 'Expert Familiarity'],
+                label_visibility="collapsed",
+            )
+
+            if 'paper_familiarity' not in st.session_state:
+                st.session_state['paper_familiarity'] = ''
+            st.session_state['paper_familiarity'] = paper_familiarity
+
             if st.button("Submit"):
                 if abstract_text == '' and uploaded_file == None:
                     st.warning('Please paste Abstract or upload a file.', icon="⚠️")
@@ -221,14 +244,20 @@ if init_page == 'Login':
                     st.session_state.draft_response_content = generate_responses(
                         text=submit_text,
                         paper_title=title_text,
-                        sci_familiarity=user_response[
-                            'paper_familiarity'],
+                        level_education=user_response[
+                            'level_education'],
+                        paper_familiarity=st.session_state['paper_familiarity'],
+                        english_proficiency=user_response[
+                            'english_proficiency'],
+                        language_spoken=user_response[
+                            'language_spoken'],
                         tech_usage=user_response[
                             'tech_usage'],
-                        read_news=user_response[
+                        news_read=user_response[
                             'news_read'],
-                        language_spoken=user_response[
-                            'language_spoken'],)
+                        books_read=user_response[
+                            'books_read'],
+                    )
 
             container = st.empty()
             # Output from function
@@ -246,15 +275,21 @@ if init_page == 'Login':
                     st.session_state.draft_response_content = generate_responses(
                         text=submit_text,
                         paper_title=title_text,
-                        sci_familiarity=user_response[
-                            'paper_familiarity'],
+                        level_education=user_response[
+                            'level_education'],
+                        paper_familiarity=st.session_state['paper_familiarity'],
+                        english_proficiency=user_response[
+                            'english_proficiency'],
+                        language_spoken=user_response[
+                            'language_spoken'],
                         tech_usage=user_response[
                             'tech_usage'],
-                        read_news=user_response[
+                        news_read=user_response[
                             'news_read'],
-                        language_spoken=user_response[
-                            'language_spoken'],
-                        additional_requirements=additional_prompt)
+                        books_read=user_response[
+                            'books_read'],
+                        additional_requirements=additional_prompt,
+                    )
                     container.empty()
                     container.text_area(label="Plain Language Summary", value=st.session_state.draft_response_content,
                                         height=350)
@@ -284,7 +319,7 @@ if init_page == 'Login':
             if 'questionnaire_response' not in st.session_state:
                 st.session_state['questionnaire_response'] = {}
             # print(st.session_state['questionnaire_response'])
-            page_number = 12
+            page_number = 11
             survey_pages = survey.pages(page_number,
                                         on_submit=lambda: update_questionnaire_response(
                                             st.session_state['questionnaire_response'], username))
@@ -340,29 +375,11 @@ if init_page == 'Login':
                         desired_information.append(other_info)
                     st.session_state['questionnaire_response']['desired_information'] = desired_information
                 elif survey_pages.current == 5:
-                    st.write("#### What is your familiarity with the concepts of the paper?")
-                    st.markdown('''
-                        * No Familiarity: entirely unfamiliar, no prior knowledge
-                        * Limited Familiarity: basic awareness of the concepts in the paper, but not in-depth knowledge
-                        * Moderate Familiarity: reasonable understanding of the concepts in the paper, encountered before, or some background knowledge
-                        * Good Familiarity: a solid understanding due to prior exposure or study
-                        * Expert: highly knowledgeable and experienced in the field and has worked extensively with these concepts
-                    ''')
-                    paper_familiarity = survey.select_slider(
-                        label="paper_familiarity",
-                        options=['No Familiarity', 'Limited Familiarity', 'Moderate Familiarity',
-                                 'Good Familiarity', 'Expert Familiarity'],
-                        # min_value=1,
-                        # max_value=5,
-                        label_visibility="collapsed",
-                    )
-                    st.session_state['questionnaire_response']['paper_familiarity'] = paper_familiarity
-                elif survey_pages.current == 6:
                     st.write("#### what is your level of english proficiency?")
                     english_proficiency = st.slider("English Proficiency (1-5):", min_value=1, max_value=5, value=1)
                     st.session_state['questionnaire_response']['english_proficiency'] = english_proficiency
 
-                elif survey_pages.current == 7:
+                elif survey_pages.current == 6:
                     st.write("#### What is the primary language spoken in your home? (click from the list and others)")
                     languages = ['English', 'Spanish', ]
                     language_options = {}
@@ -378,7 +395,7 @@ if init_page == 'Login':
                         language_spoken.append(other_language)
                     st.session_state['questionnaire_response']['language_spoken'] = language_spoken
 
-                elif survey_pages.current == 8:
+                elif survey_pages.current == 7:
                     st.write("#### Do you speak other languages? How fluent are you in each language?")
                     language_fluency = {}
                     language_index = 1
@@ -405,7 +422,7 @@ if init_page == 'Login':
                             language_fluency.update({other_language: fluency})
                     st.session_state['questionnaire_response']['other_language'] = language_fluency
 
-                elif survey_pages.current == 9:
+                elif survey_pages.current == 8:
                     st.write(
                         "#### How much do you use technology (computers, cell phones, tablets, GPS, internet, etc.)?")
                     st.markdown('''
@@ -424,7 +441,7 @@ if init_page == 'Login':
                         label_visibility="collapsed",
                     )
                     st.session_state['questionnaire_response']['tech_usage'] = tech_usage
-                elif survey_pages.current == 10:
+                elif survey_pages.current == 9:
                     st.write("#### How often do you read or watch/listen to the news?")
                     news_read = survey.radio(
                         label="news_read",
@@ -435,11 +452,11 @@ if init_page == 'Login':
                         horizontal=False,
                     )
                     st.session_state['questionnaire_response']['news_read'] = news_read
-                elif survey_pages.current == 11:
+                elif survey_pages.current == 10:
                     st.write("#### How many books do you read or listen to a month?")
                     books_read = survey.radio(
                         label="books_read",
-                        options=["None", "1-3", "4-6", "7+"],
+                        options=["0", "1-3", "4-6", "7+"],
                         index=0,
                         label_visibility="collapsed",
                         horizontal=True,