Change to new setup

app.py

@@ -28,22 +28,17 @@ st.markdown(
 ### Setting up the session state 
 
 if 'inserted' not in st.session_state:
-    with open('base.txt', 'r') as file:
-        st.session_state.base_text = file.read()
-    with open('bridging.txt', 'r') as file:
-        st.session_state.bridging_text = file.read()
-    with open('personalization.txt', 'r') as file:
-        st.session_state.personalization_text = file.read()
-    with open('both.txt', 'r') as file:
-        st.session_state.both_text = file.read()
-
     # web app state
-    st.session_state["model"] = "openai/gpt-5.2-chat"
     st.session_state.max_messages = 50
     st.session_state.messages = []
+    st.session_state.API = ""
+    st.session_state.client = ""
+    
     st.session_state.user_data = {}
+    st.session_state.user_data["BASE_URL"] = ""
+    st.session_state.user_data["MODEL_PATH"] = ""
+    st.session_state.user_data["url_id"] = True
     st.session_state.user_data["user_id"] = str(random.randint(100000, 999999))
-    st.session_state.user_data["model"] = st.session_state["model"]
     st.session_state.user_data["start_time"] = datetime.now()
     st.session_state.inserted = 1
 
@@ -61,30 +56,36 @@ if 'id' not in st.query_params:
     st.query_params['id'] = st.session_state.user_data["user_id"] 
 
 def setup_messages():
-    # 1 = base,
-    # 2 = bridging,
-    # 3 = personalization,
-    # 4 = bridging + personalization
-    # 5 = true control 
+    # 1 = true control,
+    # 2 = base,
+    # 3 = bridging,
+    # 4 = GPT
 
     if st.query_params["p"] == "1":
-        st.session_state.system_message = st.session_state.base_text 
+        pass
+        
     elif st.query_params["p"] == "2":
-        st.session_state.system_message = st.session_state.bridging_text 
+        st.session_state.user_data["BASE_URL"] = "https://openrouter.ai/api/v1"
+        st.session_state.user_data["MODEL_PATH"] = "meta-llama/llama-3.3-70b-instruct"
+        st.session_state.API = "OPENROUTER_API_KEY"
+        
     elif st.query_params["p"] == "3":
-        st.session_state.system_message = format_personalization(st.session_state.personalization_text)
+        st.session_state.user_data["BASE_URL"] = "https://tinker.thinkingmachines.dev/services/tinker-prod/oai/api/v1"
+        st.session_state.user_data["MODEL_PATH"] = "tinker://808e4f02-e847-54ae-bc75-f14ee885ce5a:train:0/sampler_weights/final_sampler"
+        st.session_state.API = "TINKER_API_KEY"
+        
     elif st.query_params["p"] == "4":
-        st.session_state.system_message = format_personalization(st.session_state.both_text)
-    elif st.query_params["p"] == "5":
-        st.session_state.system_message = st.session_state.base_text 
+        st.session_state.user_data["BASE_URL"] = "https://openrouter.ai/api/v1"
+        st.session_state.user_data["MODEL_PATH"] = "openai/gpt-5.4"
+        st.session_state.API = "OPENROUTER_API_KEY"
+
 
-    st.session_state.messages = [{ "role": "system", "content": st.session_state.system_message}]
     st.session_state.convo_start_time = datetime.now()
     st.session_state.user_data["random_pid"] = st.query_params['id']
     st.session_state.user_data["condition"] = st.query_params['p']
-    st.session_state.setup = True
 
-client = OpenAI(base_url="https://openrouter.ai/api/v1", api_key=st.secrets["OPENROUTER_API_KEY"])
+    st.session_state.client = OpenAI(base_url=st.session_state.user_data["BASE_URL"], api_key=st.secrets[st.session_state.API])
+    st.session_state.setup = True
 
 if st.session_state.setup == False:
     setup_messages()
@@ -95,23 +96,17 @@ if len(st.session_state.messages) == 1 and st.session_state.inserted < 2:
 ### App interface 
 with st.sidebar:
     st.markdown("# Let's talk!")
-    st.markdown(f"""
-    {"☑" if st.session_state.submitted else "☐"} **Step 1. Complete a form**
-
-    {"☑" if len(st.session_state.messages) > 0 else "☐"} **Step 2. Type in the chat box to start a conversation**""")
+    st.markdown(f"""# **Step 1. Type in the chat box to start a conversation**""")
 
     st.success("Ask, request, or talk to the chatbot about something you consider **politically polarizing** or something that people from different US political parties might disagree about.", icon='🎯')
 
     st.markdown("🚫 Please avoid greetings and start the conversation with a question or a statement about a politically polarizing topic. **Note: the chatbot's knowledge only goes up to late August 2025.**")
 
-    st.markdown(f"""
-    {"☑" if st.session_state.inserted > 1 else "☐"} **Step 3. Use the *Submit Interaction* button to get your chatbot word** 
+    st.markdown(f"""# **Step 2. Use the *Submit Interaction* button to get your chatbot word** 
     
     ⚠️ You must respond **at least 5 times** before you will see a *Submit Interaction* button. You can continue before submitting, but **you must Submit Interaction and enter your chatbot word to proceed with the survey**.
 
     ❗ Do not share any personal information (e.g., name or address). Do not use AI tools to write your responses. If you encounter any technical issues, please let us know. It might sometimes take 30 seconds or more to generate a response, so please be patient.
-
-    {"🎉 **All done! Please enter your WORD in the survey and press *Next*.**" if st.session_state.inserted > 1 else ""}
     """)
                 
 for message in st.session_state.messages:
@@ -127,7 +122,7 @@ if len(st.session_state.messages) >= st.session_state.max_messages:
 elif st.session_state.inserted > 1:
     st.markdown("## Copy your WORD!")
     st.markdown('**Your chatbot WORD is:**')
-    st.markdown(f'## POTATOES')
+    st.markdown(f'## TOMATOES')
     st.markdown('**Please copy the WORD and enter it into the survey field below.**')
 
 elif prompt := st.chat_input("Type to ask a question or respond..."):   
@@ -153,7 +148,7 @@ elif prompt := st.chat_input("Type to ask a question or respond..."):
     with st.chat_message("user"):
         st.markdown(prompt)
 
-    if st.query_params["p"] == "5":
+    if st.query_params["p"] == "1":
         rate_limit_message = "Thank you for your question! You have been randomly assigned to a condition without a chatbot. **Please submit your interaction anyway** to get your chatbot word and proceed with the survey. Do not worry, this will not influence your compensation."
         st.session_state.messages.append(
                     {"role": "assistant", "content": rate_limit_message}
@@ -164,20 +159,22 @@ elif prompt := st.chat_input("Type to ask a question or respond..."):
     else:
         with st.chat_message("assistant"):
             try:
-                stream = client.chat.completions.create(
-                    model=st.session_state["model"],
-                    messages=[
-                        {"role": m["role"], "content": m["content"]}
-                        for m in st.session_state.messages
-                    ],
-                    stream=True
-                )
-                
-                response = st.write_stream(stream)
-    
+                with st.spinner("Typing..."):
+                    completion = st.session_state.client.chat.completions.create(
+                        model=st.session_state.user_data["MODEL_PATH"],
+                        messages=[
+                            {"role": m["role"], "content": m["content"]}
+                            for m in st.session_state.messages
+                        ],
+                        stream=False,
+                    )
+        
+                response = completion.choices[0].message.content or ""
+                st.markdown(response)
+        
                 st.session_state.messages.append(
                     {"role": "assistant", "content": response}
-                )
+
             except:
                 rate_limit_message = "An error has occurred or you've reached the maximum conversation length. Please submit the conversation."
                 st.session_state.messages.append(