Update app.py

app.py

@@ -12,6 +12,8 @@ from langchain import HuggingFaceHub, LLMChain, PromptTemplate
 from langchain.memory import ConversationBufferWindowMemory
 from langchain.chains import ConversationalRetrievalChain
 
+os.environ["OPENAI_API_KEY"] ='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'
+
 trait_content_df=pd.read_csv('AI Personality Chart trait_content.csv')
 trait_content_df=trait_content_df.drop(0,axis=0)
 trait_content_df.rename(columns={'Column 1':'Question','Column 2':'Options','Column 3':'Traits','Column 4':'Content'},inplace=True)
@@ -63,6 +65,12 @@ trait_content_df_org=pd.read_csv('AI Personality Chart trait_content.csv')
 trait_content_df_org=trait_content_df_org.drop(0,axis=0)
 trait_content_df_org.rename(columns={'Column 1':'Question','Column 2':'Options','Column 3':'Traits','Column 4':'Content'},inplace=True)
 
+# Import the second CSV file
+traits_description_df = pd.read_csv('Traits Description - Sheet1.csv')
+
+# Extract unique values from the "Traits" column of trait_content_df_org
+unique_traits = trait_content_df_org['Traits'].unique()
+
 
 def ui():
     # Initialize a dictionary to store responses
@@ -97,11 +105,11 @@ def ui():
         st.write(question)
         st.write(selected_option)
 
-    # Generate a prompt based on selected options
+    # Extract traits from the selected options
     selected_traits = [responses[question] for question in responses]
     options_list = []
     traits_list = []
-    content_list = []
+    # content_list = []
 
     for trait_str in selected_traits:
         matching_rows = trait_content_df_org[trait_content_df_org["Options"] == trait_str]
@@ -109,24 +117,24 @@ def ui():
         if not matching_rows.empty:
             options_list.append(matching_rows["Options"].values[0])
             traits_list.append(matching_rows["Traits"].values[0])
-            content_list.append(matching_rows["Content"].values[0])
 
-    prompt = f"The following are Traits {', '.join(traits_list)}, and the content for the options is {', '.join(content_list)}"
+    # Get unique traits from the selected options
+    unique_traits = list(set(traits_list))
+
+    # Create a content list as a formatted string
+    formatted_content_list = ",".join([f"'{trait.lower()}'" for trait in unique_traits])
 
     # Display user input field
     name_input = st.text_input("Enter your name:")
 
     # Add a submit button
     if st.button("Submit"):
-        # Generate a chatbot response
+        # Generate a chatbot response using the formatted content list as part of the prompt
+        prompt = f"The following are Traits {formatted_content_list}"
         bio = conversational_chat(prompt, name_input)
         st.write(bio)
 
 
 
-
-    
-    
-
 if __name__=='__main__':
     ui()