Update app.py

app.py

@@ -330,8 +330,8 @@ if selected_app == "4) Create Chatbot":
                     completion = client.chat.completions.create(
                     model="gpt-3.5-turbo-0125",
                     messages=[
-                        {"role": "system", "content": directions},
-                        {"role": "user", "content": prompt}
+                        {{"role": "system", "content": directions}},
+                        {{"role": "user", "content": prompt}}
                     ]
                     )
                 
@@ -387,8 +387,8 @@ if selected_app == "4) Create Chatbot":
                     # Loop through each row of the DataFrame
                     for index, row in df.iterrows():
                         # Create a dictionary with the current question and answer
-                        qa_dict_quest = {"role": "user", "content": row["questions"]}
-                        qa_dict_ans = {"role": "assistant", "content": row["answers"]}
+                        qa_dict_quest = {{"role": "user", "content": row["questions"]}}
+                        qa_dict_ans = {{"role": "assistant", "content": row["answers"]}}
             
                         # Add the dictionary to the result list
                         result.append(qa_dict_quest)
@@ -433,7 +433,7 @@ if selected_app == "4) Create Chatbot":
                 )
                 
                 # Combine the random number and random string into one identifier.
-                combined_string: str = f"{random_number}{random_string}"
+                combined_string: str = f"{{random_number}}{{random_string}}"
             
                 # Create a new collection in ChromeDB with the combined string as its name.
                 collection = client.create_collection(combined_string)
@@ -466,14 +466,14 @@ if selected_app == "4) Create Chatbot":
                     collection.add(
                         ids=[str(i) for i in range(0, L)],  # IDs are just strings
                         documents=dataset["train"]["questions"],  # Enter questions here
-                        metadatas=[{"type": "support"} for _ in range(0, L)],
+                        metadatas=[{{"type": "support"}} for _ in range(0, L)],
                     )
                     
                     if st.session_state.curr_domain != domain:
                         st.session_state.messages = []
                         
                         init_message = init_messages[domain]
-                        st.session_state.messages.append({"role": "assistant", "content": init_message})
+                        st.session_state.messages.append({{"role": "assistant", "content": init_message}})
                         
                         st.session_state.curr_domain = domain
                 
@@ -487,7 +487,7 @@ if selected_app == "4) Create Chatbot":
                     # Display user message in chat message container
                     st.chat_message("user").markdown(prompt)
                     # Add user message to chat history
-                    st.session_state.messages.append({"role": "user", "content": prompt})
+                    st.session_state.messages.append({{"role": "user", "content": prompt}})
                 
                     question = prompt
                 
@@ -496,12 +496,12 @@ if selected_app == "4) Create Chatbot":
                     idx = results["ids"][0]
                     idx = [int(i) for i in idx]
                     ref = pd.DataFrame(
-                        {
+                        {{
                             "idx": idx,
                             "questions": [dataset["train"]["questions"][i] for i in idx],
                             "answers": [dataset["train"]["answers"][i] for i in idx],
                             "distances": results["distances"][0],
-                        }
+                        }}
                     )
                     # special_threshold = st.sidebar.slider('How old are you?', 0, 0.6, 0.1) # 0.3
                     # special_threshold = 0.3
@@ -518,8 +518,8 @@ if selected_app == "4) Create Chatbot":
                         final_ref = ref
                 
                     engineered_prompt = f'''
-                        Based on the context: {ref_from_db_search},
-                        answer the user question: {question}.
+                        Based on the context: {{ref_from_db_search}},
+                        answer the user question: {{question}}.
                     '''
                 
                     directions = chatbot_instructions[domain]
@@ -533,7 +533,7 @@ if selected_app == "4) Create Chatbot":
                         with st.expander("See reference:"):
                             st.table(final_ref) 
                     # Add assistant response to chat history
-                    st.session_state.messages.append({"role": "assistant", "content": response})
+                    st.session_state.messages.append({{"role": "assistant", "content": response}})
             """
             
             st.session_state.clear()