Update src/streamlit_app.py

src/streamlit_app.py

@@ -70,18 +70,18 @@ def get_current_system_prompt():
         "Static: dense paragraph only",
     )
 
+    
+    # static
     if mode.startswith("Static"):
-        # Always use dense paragraph prompt
         st.session_state["assigned_prompt_type"] = "dense"
         return SYSTEM_PROMPT_DENSE
 
-    # Random assignment mode
-    # if "assigned_prompt_type" not in st.session_state:
-    #     # Randomly assign this session to 'dense' or 'bullet'
-    #     st.session_state["assigned_prompt_type"] = random.choice(["dense", "bullet"])
+    # Random assignment
+    if "assigned_prompt_type" not in st.session_state:
+        st.session_state["assigned_prompt_type"] = random.choice(["dense", "bullet"])
 
-    # if st.session_state["assigned_prompt_type"] == "dense":
-    #     return SYSTEM_PROMPT_DENSE
+    if st.session_state["assigned_prompt_type"] == "dense":
+        return SYSTEM_PROMPT_DENSE
     else:
         return SYSTEM_PROMPT_BULLET