Update src/streamlit_app.py

src/streamlit_app.py

@@ -658,51 +658,6 @@ def update_user_progress(user_id, topic_id, comment_id):
         if local_con:
             local_con.close()
 
-# Helper function to handle comment submission UI and logic
-def share_wisdom(prompt, allow_skip=False):
-    st.markdown(prompt)
-    new_comment_text = st.text_area(f"Your Insight {'that different from others above (Empty to skip)' if allow_skip else ''}", key="new_comment_input")
-    if st.button("Share Your Wisdom", key=f"share_wisdom_button_{random.randint(1, 11)}"):
-        if new_comment_text:
-            user_email = st.session_state.get('user_email', '')
-            user_id = find_or_create_user(user_email) # Ensure user exists
-            if user_id:
-                local_con = None
-                try:
-                    local_con = duckdb.connect(database=DB_PATH, read_only=False)
-                    comment_id = str(uuid.uuid4())
-                    local_con.execute("""
-                        INSERT INTO comments (id, topic_id, user_id, content)
-                        VALUES (?, ?, ?, ?)
-                    """, [comment_id, topic_id, user_id, new_comment_text])
-
-                    # Append new comment to history
-                    st.session_state.comment_history += f"\n\n💬 {new_comment_text}"
-
-                    # Get next comment (could be the one just submitted)
-                    next_comment_id, next_comment_content = get_random_unvoted_comment(user_id, topic_id)
-                    st.session_state.current_comment_id = next_comment_id
-                    st.session_state.current_comment_content = next_comment_content
-
-                    # Update progress
-                    update_user_progress(user_id, topic_id, next_comment_id)
-
-                    st.session_state.new_comment_input = "" # Clear input box
-                    st.rerun() # Rerun to update UI
-
-                except Exception as e:
-                    st.error(f"Error sharing information: {e}")
-                finally:
-                    if local_con:
-                        local_con.close()
-            else:
-                st.error("Could not find or create user.")
-        elif allow_skip:
-            return
-        else:
-            st.warning("Please enter your thought.")
-
-
 # --- Page Functions ---
 
 def home_page():
@@ -834,7 +789,7 @@ def view_topic_page():
     # Include functional information
     st.markdown(f"**Shareable Quest Scroll ID:** `{topic_id}`")
     # Construct shareable link using current app URL
-    app_url = st.query_params.get('base', ['https://huggingface.co/spaces/npc0/SteamPolis/'])[0] # Get base URL if available
+    app_url = st.query_params.get('base', ['http://localhost:8501/'])[0] # Get base URL if available
     shareable_link = f"{app_url}?topic={topic_id}" if app_url else f"?topic={topic_id}"
     st.markdown(f"**Shareable Scroll Link:** `{shareable_link}`")
 
@@ -933,8 +888,11 @@ def view_topic_page():
         if show_new_area_message == True:
             _, user_ids = get_user_cluster_label(user_id, get_ttl_hash(10))
             new_area_name, desc = name_user_group(user_ids, topic_id)
-            st.markdown(f"📚 You've collected **{len(comment_history.splitlines())}** insights this time.")
-            st.markdown(f"🗺️ And yet, your journey leads you to a new place: **{new_area_name}**! {desc}")
+            for statm in [
+                f"📚 You've collected **{len(comment_history.splitlines())}** insights this time.",
+                f"🗺️ And yet, your journey leads you to a new place: **{new_area_name}**! {desc}"]:
+                st.markdown(statm)
+                st.session_state.comment_history += f"\n\n{statm}"
             st.session_state._show_new_area_message = False
         st.markdown(f"[Collected new insight, {random_phrase}]:\n* {current_comment_content}")
 
@@ -993,7 +951,43 @@ def view_topic_page():
                             "As you walk through the streets, people gather, eager to hear your thoughts on the Emperor's dilemma. What advice do you give?"
                         ]
                         # Pass the current topic_id to share_wisdom if needed, though it's not currently used there.
-                        share_wisdom(random.choice(prompts), allow_skip=True)
+                        st.markdown(random.choice(prompts))
+                        new_comment_text = st.text_area("Your Insight that different from others above (Empty to skip)", key="tmp_new_comment_input")
+                        if st.button("Share Your Wisdom"):
+                            if new_comment_text and len(new_comment_text.strip()):
+                                user_email = st.session_state.get('user_email', '')
+                                user_id = find_or_create_user(user_email) # Ensure user exists
+                                if user_id:
+                                    local_con = None
+                                    try:
+                                        local_con = duckdb.connect(database=DB_PATH, read_only=False)
+                                        comment_id = str(uuid.uuid4())
+                                        local_con.execute("""
+                                            INSERT INTO comments (id, topic_id, user_id, content)
+                                            VALUES (?, ?, ?, ?)
+                                        """, [comment_id, topic_id, user_id, new_comment_text])
+
+                                        # Append new comment to history
+                                        st.session_state.comment_history += f"\n\n💬 {new_comment_text}"
+
+                                        # Get next comment (could be the one just submitted)
+                                        next_comment_id, next_comment_content = get_random_unvoted_comment(user_id, topic_id)
+                                        st.session_state.current_comment_id = next_comment_id
+                                        st.session_state.current_comment_content = next_comment_content
+
+                                        # Update progress
+                                        update_user_progress(user_id, topic_id, next_comment_id)
+
+                                        st.session_state.tmp_new_comment_input = "" # Clear input box
+                                        st.rerun() # Rerun to update UI
+
+                                    except Exception as e:
+                                        st.error(f"Error sharing information: {e}")
+                                    finally:
+                                        if local_con:
+                                            local_con.close()
+                                else:
+                                    st.error("Could not find or create user.")
 
                 # Get next comment
                 # This should always get the next unvoted comment for the user in this topic.
@@ -1039,7 +1033,46 @@ def view_topic_page():
 
     # --- Comment Submission ---
     with st.expander("Offer Your Counsel to the Emperor", expanded=False):
-        share_wisdom("Having heard the thoughts of others, what wisdom do you wish to share regarding the Emperor's concern?")
+        st.markdown("Having heard the thoughts of others, what wisdom do you wish to share regarding the Emperor's concern?")
+        new_comment_text = st.text_area(f"Your Insight", key="new_comment_input")
+        if st.button("Share Your Wisdom"):
+            if new_comment_text:
+                user_email = st.session_state.get('user_email', '')
+                user_id = find_or_create_user(user_email) # Ensure user exists
+                if user_id:
+                    local_con = None
+                    try:
+                        local_con = duckdb.connect(database=DB_PATH, read_only=False)
+                        comment_id = str(uuid.uuid4())
+                        local_con.execute("""
+                            INSERT INTO comments (id, topic_id, user_id, content)
+                            VALUES (?, ?, ?, ?)
+                        """, [comment_id, topic_id, user_id, new_comment_text])
+
+                        # Append new comment to history
+                        st.session_state.comment_history += f"\n\n💬 {new_comment_text}"
+
+                        # Get next comment (could be the one just submitted)
+                        next_comment_id, next_comment_content = get_random_unvoted_comment(user_id, topic_id)
+                        st.session_state.current_comment_id = next_comment_id
+                        st.session_state.current_comment_content = next_comment_content
+
+                        # Update progress
+                        update_user_progress(user_id, topic_id, next_comment_id)
+
+                        st.session_state.new_comment_input = "" # Clear input box
+                        st.rerun() # Rerun to update UI
+
+                    except Exception as e:
+                        st.error(f"Error sharing information: {e}")
+                    finally:
+                        if local_con:
+                            local_con.close()
+                else:
+                    st.error("Could not find or create user.")
+            else:
+                st.warning("Please enter your thought.")
+
     st.markdown("---")