update

app.py

@@ -26,6 +26,24 @@ def generate_one_completion(message, temperature):
 
 import json
 
+def get_profile_str(profile):
+    bio_info = profile['bio_information']
+    main_profile = profile['main_profile']
+    red_flag = profile['red_flag']
+    motivation = profile['motivation']
+    profile_str = f"Bio Information:\n"
+    for key, value in bio_info.items():
+        profile_str += f"- {key.replace('_', ' ').title()}: {value}\n"
+    profile_str += f"\nMain Profile:\n"
+    for key, value in main_profile.items():
+        profile_str += f"- {key.title()}: {value['score']} - {value['summary']}\n"
+    profile_str += f"\nRed Flags:\n"
+    for key, value in red_flag.items():
+        profile_str += f"- {key.title()}: {value['score']} - {value['summary']}\n"
+    profile_str += f"\nMotivation:\n"
+    for key, value in motivation.items():
+        profile_str += f"- {key.title()}: {value['score']} - {value['summary']}\n"
+    return profile_str
 
 def generate_prompt_from_profile(profile, version="TestTakersSummary"):
     with open('prompts.json') as f:
@@ -162,26 +180,6 @@ def main_app():
     else:
         st.write("Please upload a profile JSON file or use the example profile.")
 
-    # if st.session_state['analysis']:
-    #     st.markdown(st.session_state['analysis'])
-    #     #st.markdown("### Analysis:")
-    #     #analysis_container = st.container()
-    #     # with analysis_container:
-    #     #st.markdown(st.session_state['analysis'].choices[0].message.content)
-    #
-    #     st.markdown("### Token Usage:")
-    #     token_usage_container = st.expander("Show Token Usage Details")
-    #     with token_usage_container:
-    #         total_tokens = st.session_state['analysis'].usage.total_tokens
-    #         prompt_tokens = st.session_state['analysis'].usage.prompt_tokens
-    #         completion_tokens = st.session_state['analysis'].usage.completion_tokens
-    #         costs = (0.01 / 1000) * prompt_tokens + (0.03 / 1000) * completion_tokens
-    #         st.write(f'**Total tokens:** {total_tokens}')
-    #         st.progress(completion_tokens / 1000)
-    #         st.write(f'**Prompt tokens:** {prompt_tokens}')
-    #         st.write(f'**Completion tokens:** {completion_tokens}')
-    #         st.write(f'**Generation costs ($):** {costs}')
-
 
 # Function to verify credentials and set the session state
 def verify_credentials():
@@ -231,6 +229,7 @@ def sidebar_components():
 
             st.markdown("### Choose the Definition:")
             st.session_state['definition'] = st.selectbox("Select Definition", [1, 2])
+            st.session_state['chat_context'] = st.selectbox("Select Chat Context", ["analysis", "profile"])
 
             # File uploader
             st.markdown("### Upload a profile JSON file")
@@ -265,30 +264,8 @@ def sidebar_components():
             st.sidebar.markdown(
                 "Hello, we hope you learned something about yourself in this report. This chat is here so you can ask any questions you have about your report!  It’s also a great tool to get ideas about how you can use the information in your report for your personal development and achieving your current goals.")
 
-            # question_message = generate_prompt_from_profile(st.session_state['profile'])
-            # question_message.append(
-            #     {"role": "system", "content": st.session_state['analysis'].choices[0].message.content})
-            # question_prompt = (
-            #     f"Based on the earlier profile summary and analysis results about the individual, "
-            #     f"generate two insightful questions that could be asked by that individual for further discussion about themself:\n\n"
-            #     f"Provide me the questions in different new line."
-            #     f"Suggested Questions:\n"
-            # )
-            # question_message.append({"role": "user", "content": question_prompt})
-
-            # questions = generate_one_completion(question_message, 0)
-            #
-            # questions_list = [question.strip() for question in questions.choices[0].message.content.split('\n')
-            #                   if question.strip()]
-
-            # questions_list = []
-            # print(questions_list)
-            #     # Prepare the questions for Markdown display
-            # questions_markdown = "\n\n".join(
-            #     [f"Q{question}" for index, question in enumerate(questions_list[:2])])
-
             # Name to be included in the questions
-            name = st.session_state['profile']['bio_information'].get('Name', 'the individual')
+            # name = st.session_state['profile']['bio_information'].get('Name', 'the individual')
 
             # List of question templates where {} will be replaced with the name
             question_templates = [
@@ -323,8 +300,10 @@ def sidebar_components():
 
             if st.sidebar.button('Submit'):
                 if user_input:
-
-                    chat_prompt_template = create_chat_prompt_template(st.session_state['analysis'],st.session_state['definition'])
+                    if st.session_state['chat_context'] == "profile":
+                        chat_prompt_template = create_chat_prompt_template(get_profile_str(st.session_state['profile']),st.session_state['definition'])
+                    else:
+                        chat_prompt_template = create_chat_prompt_template(st.session_state['analysis'],st.session_state['definition'])
                     response = execute_query(index, chat_prompt_template, user_input)
 
                     st.sidebar.markdown(response)
@@ -337,6 +316,9 @@ if 'show_chat' not in st.session_state:
 if 'definition' not in st.session_state:
     st.session_state['definition'] = 1
 
+if 'chat_context' not in st.session_state:
+    st.session_state['chat_context'] = "analysis"
+
 if 'profile' not in st.session_state:
     st.session_state['profile'] = None