feat: minimal working inplace feedback

cards.py

@@ -6,15 +6,5 @@ def chat_card():
     # st.chat_message("assistant").write("Hello, user!")
     # st.chat_input("Type something")
 
-# def status_card():
-#     st.page_link("status.py", label="Status", icon=":material/error:")
-#     cols = st.columns(2)
-#     cols[0].error("Error")
-#     cols[0].warning("Warning")
-#     cols[1].info("Info")
-#     cols[1].success("Success")
-#     cols = st.columns(2)
-#     if cols[0].button("Toast!"):
-#         st.toast("Toast message", icon=":material/notifications:")
-#     if cols[1].button("Balloons!"):
-#         st.balloons()
+def inplace_chat_card():
+    st.page_link("inplace_chat.py", label="In-place Feedback Chat", icon=":material/edit_document:")

chat.py

@@ -4,7 +4,7 @@ from huggingface_hub import InferenceClient
 import os
 import json
 
-st.header("In-place feedback demo")
+st.header("Multi-turn feedback chatbot demo")
 
 # import the inference client
 hf_token = os.getenv("HF_TOKEN")
@@ -27,13 +27,14 @@ if "chat_history" not in st.session_state:
 for message in st.session_state.chat_history:
     with st.chat_message(message["role"]):
         st.markdown(message["content"])
-        if "reasoning" in message:
-            with st.expander("Show reasoning"):
-                st.markdown(message["reasoning"])
+        # if "reasoning" in message:
+        #     with st.expander("Show reasoning"):
+        #         st.markdown(message["reasoning"])
 
 if prompt := st.chat_input("Send a message"):
     # add message to chat history
     st.session_state.chat_history.append({"role": "user", "content": prompt})
+    
     # show message
     with st.chat_message("user"):
         st.markdown(prompt)     # for markdown of user text
@@ -63,10 +64,10 @@ if prompt := st.chat_input("Send a message"):
                 st.markdown(reply)
                 print(reply)
                 
-                reasoning = getattr(response.choices[0], "reasoning", None)
-                if reasoning:
-                    with st.expander("Show reasoning"):
-                        st.markdown(reasoning)
+                # reasoning = getattr(response.choices[0], "reasoning", None)
+                # if reasoning:
+                #     with st.expander("Show reasoning"):
+                #         st.markdown(reasoning)
                     
                 # Add assistant response to chat history
                 st.session_state.chat_history.append(

inplace_chat.py

@@ -0,0 +1,182 @@
+# inplace_chat.py
+import os
+import json
+import difflib
+import streamlit as st
+from huggingface_hub import InferenceClient
+
+# === Page header ===
+st.header("In-place feedback chatbot demo")
+
+# === Theme primary color (from .streamlit/config.toml) ===
+PRIMARY = "#cb785c"  # keep in sync with [theme] primaryColor
+
+# HF Inference client 
+hf_token = os.getenv("HF_TOKEN")
+client = InferenceClient(model="openai/gpt-oss-20b", token=hf_token)
+
+####### initialize session states ######
+if "chat_history" not in st.session_state:
+    st.session_state.chat_history = []  # [{"role": "user", ...}, {"role": "assistant", ...}]
+if "original_user_prompt" not in st.session_state:
+    st.session_state.original_user_prompt = ""
+if "editable" not in st.session_state:
+    st.session_state.editable = False
+if "prev_text" not in st.session_state:
+    st.session_state.prev_text = ""
+if "edited_text" not in st.session_state:
+    st.session_state.edited_text = ""
+if "prev_msgs" not in st.session_state:
+    st.session_state.prev_msgs = []
+
+# === Utility: compact "removed vs added" summary for the inplace_prefix ===
+### later separate this into parse_diff.py
+def summarize_edit(old: str, new: str) -> tuple[str, str]:
+    """
+    Produce coarse 'removed_text' and 'edited_text' by joining all deletions and insertions.
+    Good enough for a single bullet like: - Replaced {removed_text} with {edited_text}
+    """
+    sm = difflib.SequenceMatcher(a=old, b=new)
+    removed_chunks, added_chunks = [], []
+    for tag, i1, i2, j1, j2 in sm.get_opcodes():
+        if tag in ("delete", "replace"):
+            chunk = old[i1:i2].strip()
+            if chunk:
+                removed_chunks.append(chunk)
+        if tag in ("insert", "replace"):
+            chunk = new[j1:j2].strip()
+            if chunk:
+                added_chunks.append(chunk)
+    removed_text = " / ".join(removed_chunks) if removed_chunks else "(none)"
+    edited_text = " / ".join(added_chunks) if added_chunks else "(none)"
+    return removed_text, edited_text
+
+# === Render current conversation ===
+for i, msg in enumerate(st.session_state.chat_history):
+    # Hide the last assistant message while the editor is open
+        if st.session_state.editable and i == len(st.session_state.chat_history) - 1 and msg["role"] == "assistant":
+            continue
+        with st.chat_message(msg["role"]):
+            st.markdown(msg["content"])
+
+if prompt := st.chat_input("Send a message"):
+    st.session_state.original_user_prompt = prompt
+    st.session_state.chat_history.append({"role": "user", "content": prompt})
+    with st.chat_message("user"):
+        st.markdown(prompt)
+
+    messages_for_api = [
+        {"role": "system", "content": "You are a helpful assistant."},
+        {"role": "user", "content": prompt},
+    ]
+
+    with st.chat_message("assistant"):
+        with st.spinner("Thinking…"):
+            try:
+                response = client.chat_completion(
+                    messages=messages_for_api,
+                    max_tokens=60000,
+                    temperature=0.7,
+                    top_p=0.95,
+                    stream=False,
+                )
+            
+                reply = response.choices[0].message["content"]
+                st.markdown(reply)
+                st.session_state.chat_history.append(
+                    {"role": "assistant", "content": reply}
+                )
+                st.session_state.prev_text = reply
+                # save previous messages for reference
+                st.session_state.prev_msgs.append(reply)
+                
+            except Exception as e:
+                st.error(f"Error: {e}")
+
+# Edit features
+if st.session_state.chat_history and st.session_state.chat_history[-1]["role"] == "assistant":
+    # button
+    with st.chat_message("assistant"):
+        col_spacer, col_edit = st.columns([0.93, 0.07])
+        with col_edit:
+            if st.button("✏️", key="edit_btn", help="Edit response"):
+                st.session_state.editable = True
+
+    ######## Edit mode #########
+    if st.session_state.editable:
+        st.markdown('<div class="editable-on">', unsafe_allow_html=True)
+        with st.chat_message("assistant"):
+            st.caption("Editing the last response…")
+            st.session_state.edited_text = st.text_area(
+                " ",
+                value=st.session_state.prev_text,
+                height=500,
+                label_visibility="collapsed",
+                key="edit_textarea",
+            )
+            # Bottom-right finish button
+            right = st.columns([1, 0.25])[1]
+            with right:
+                finished_edit = st.button(
+                    "Finish edit",
+                    icon=":material/edit:",
+                    type="primary",
+                    use_container_width=True,
+                    key="finish_edit_btn",
+                )
+        st.markdown("</div>", unsafe_allow_html=True)
+    else:
+        finished_edit = False
+
+    # === Handle edit submission (backend: in-place continuation) ===
+    if finished_edit:
+        st.session_state.chat_history.pop()
+        removed_text, added_text = summarize_edit(st.session_state.prev_text, st.session_state.edited_text)
+
+        system_prompt = "The user edited your previous answer. Please continue reasoning from the edited version below."
+        original_user_prompt = st.session_state.original_user_prompt
+
+        inplace_prefix = (
+            "This is the previous user correction to the reasoning step:\n"
+            f"- Replaced {json.dumps(removed_text)} with {json.dumps(added_text)}\n"
+            "---\n"
+            f"{st.session_state.edited_text}\n"
+            "### Further reasoning and validation\n"
+        )
+        
+         # Exit edit mode before generation
+        st.session_state.editable = False
+
+        with st.chat_message("assistant"):
+            with st.spinner("Continuing from your edit…"):
+                try:
+                    resp = client.chat_completion(
+                        messages=[
+                            {"role": "system", "content": system_prompt},
+                            {
+                                "role": "user",
+                                "content": original_user_prompt
+                                        + "\n\nUser has edited your previous response. Continue from the corrected text below:\n\n"
+                                        + inplace_prefix
+                            },
+                        ],
+                        max_tokens=60000,
+                        temperature=0.7,
+                        top_p=0.95,
+                    )
+
+                    generated = resp.choices[0].message["content"]
+                    st.session_state.prev_text = generated
+                    print("generated text after feedback")
+                    print(generated)
+                    st.session_state.chat_history.append(
+                        {"role": "assistant", "content": generated}
+                    )
+                    st.rerun()
+
+                except Exception as e:
+                    st.error(f"Error while continuing from edit: {e}")
+                    st.session_state.chat_history.append(
+                        {"role": "assistant", "content": st.session_state.prev_text}
+                    )
+                    st.session_state.editable = False

streamlit_app.py

@@ -12,21 +12,25 @@ from cards import (
     # media_card,
     # layouts_card,
     chat_card,
+    inplace_chat_card,
     # status_card
 )
 
 if "init" not in st.session_state:
-    # st.session_state.chart_data = pd.DataFrame(
-    #     np.random.randn(20, 3), columns=["a", "b", "c"]
-    # )
-    # st.session_state.map_data = pd.DataFrame(
-    #     np.random.randn(1000, 2) / [50, 50] + [37.76, -122.4],
-    #     columns=["lat", "lon"],
-    # )
     st.session_state.init = True
 
 
 pages = [
+    st.Page(
+        "inplace_chat.py",
+        title="In-place Feedback Chat",
+        icon=":material/edit_document:"
+    ),
+    st.Page(
+        "chat.py",
+        title="Chat",
+        icon=":material/chat:"
+    ),
     # st.Page(
     #     "home.py",
     #     title="Home",
@@ -62,11 +66,6 @@ pages = [
     #     title="Layouts",
     #     icon=":material/dashboard:"
     # ),
-    st.Page(
-        "chat.py",
-        title="Chat",
-        icon=":material/chat:"
-    ),
     # st.Page(
     #     "status.py",
     #     title="Status",
@@ -78,11 +77,7 @@ page = st.navigation(pages)
 page.run()
 
 with st.sidebar.container(height=310):
-    if page.title == "Chat":
-        chat_card()
-    else:
-        st.page_link("home.py", label="Home", icon=":material/home:")
-        st.write("Welcome to the home page!")
+    st.write("This is a simple demo for CSED499 Inplace feedback.")
 
 st.sidebar.caption(
     "This app uses [Space Grotesk](https://fonts.google.com/specimen/Space+Grotesk) "