Update frontend_streamlit.py

frontend_streamlit.py

@@ -1,129 +1,96 @@
-import streamlit as st
-from chat_langraph import system, workflow, HumanMessage, AIMessage, get_all_chat_ids, ToolMessage
-import uuid
-from chat_title_giver import give_me_the_title
-import os
-import base64  
-import re
-
-st.title("College chatbot")
-st.set_page_config(layout='wide')
-
-def set_title(messages):
-    if messages:
-        title = give_me_the_title(messages)
-        st.session_state.chat_dict[st.session_state.current_chat_id] = title
-
-def set_config():
-    return {'configurable': {'thread_id': st.session_state.current_chat_id}}
-
-
-def load_session_state():
-    if "chats" not in st.session_state:
-        st.session_state.chats = get_all_chat_ids()
-    if "current_chat_id" not in st.session_state:
-        if len(st.session_state.chats) > 0:
-            st.session_state.current_chat_id = st.session_state.chats[-1]
-        else:
-           
-            new_id = str(uuid.uuid4())
-            st.session_state.chats.append(new_id)
-            st.session_state.current_chat_id = new_id
-    if "chat_dict" not in st.session_state:
-        st.session_state.chat_dict = {}
-    if "uploaded_image" not in st.session_state:
-        st.session_state.uploaded_image = None
-
-def render_sidebar():
-    with st.sidebar:
-        st.title("Chats")
-        if st.button("➕ New Chat"):
-            new_id = str(uuid.uuid4())
-            st.session_state.chats.append(new_id)
-            st.session_state.current_chat_id = new_id
-            config = {'configurable': {'thread_id': new_id}}
-            workflow.update_state(config, {"messages": [system]})
-            st.session_state.chat_dict[new_id] = "New Chat"
-            st.rerun()  
-        for chat_id in st.session_state.chats:
-            if st.button(st.session_state.chat_dict.get(chat_id, "New Chat"), key=chat_id):
-                st.session_state.current_chat_id = chat_id
-
-def loadchats():
-    if "current_chat_id" not in st.session_state:
-        return []
-    config = {'configurable': {'thread_id': st.session_state.current_chat_id}}
-    state = workflow.get_state(config)
-    messages = state.values.get("messages", [])
-    for message in messages:
-        if isinstance(message, HumanMessage):
-            if message.content and message.content.strip():
-                with st.chat_message("human"):
-                    st.write(message.content)
-        elif isinstance(message, AIMessage):
-            content = st.expander("Thinking...", expanded=False)
-            if message.content and message.content.strip():
-                text = message.content
-                m = re.search(r'</\s*think\s*>', text, re.IGNORECASE)
-                if m:
-                    thinking_text = text[:m.start()].strip()
-                    rest = text[m.end():].strip()
-                    if thinking_text:
-                        with content:
-                            st.write(thinking_text)
-                    if rest:
-                        with st.chat_message("assistant"):
-                            st.write(rest)
-                else:
-                    with st.chat_message("assistant"):
-                        st.write(text)
-        elif isinstance(message, ToolMessage):
-            with st.chat_message("assistant"):
-                st.info(f"Used a tool")
-    return messages
-
-load_session_state()
-render_sidebar()
-if("current_chat_id" in st.session_state):
-    loadchats()
-    user_input = st.chat_input("Your message: ")
-    if(user_input is not None):
-        response_placeholder = st.empty() 
-        with st.chat_message("human"):
-            st.write(user_input)
-        
-        full_response = ""
-        answer_response = ""
-        is_thinking = True
-        with st.chat_message("assistant"):
-            content = st.expander("Thinking...", expanded=True)
-            resp = st.empty() 
-            resp2 = None
-            with content:
-                resp2 = st.empty()
-            for messages,metadata in workflow.stream({"messages":[system,HumanMessage(user_input)]} , config={'configurable': {'thread_id': st.session_state.current_chat_id}} , stream_mode="messages"):
-                if(isinstance(messages , AIMessage) and messages.content.strip()):
-                    if(messages.content.startswith("<think>")):
-                        is_thinking = True
-                        resp2.markdown(full_response + "|")
-                    if is_thinking:
-                        full_response += messages.content
-                        with content:
-                            resp2.markdown(full_response + "|")
-                        if "</think>" in full_response:
-                            is_thinking = False
-                            full_response = full_response.replace("</think>", "")
-                            resp2.markdown(full_response)
-                    else:
-                        answer_response += messages.content
-                        resp.markdown(answer_response + "|")
-                if(isinstance(messages ,ToolMessage) and messages.content.strip()):
-                    st.info(f"Using appropriate tool")
-
-        st.rerun()    
-
-        
-
-
-
-
+import streamlit as st
+from chat_langraph import system, workflow, HumanMessage, AIMessage, get_all_chat_ids, ToolMessage
+import uuid
+from chat_title_giver import give_me_the_title
+import re
+
+st.title("College chatbot")
+st.set_page_config(layout='wide')
+
+
+def set_title(messages):
+    if messages:
+        title = give_me_the_title(messages)
+        st.session_state.chat_dict[st.session_state.current_chat_id] = title
+
+
+def set_config():
+    return {'configurable': {'thread_id': st.session_state.current_chat_id}}
+
+
+def load_session_state():
+    if "chats" not in st.session_state:
+        st.session_state.chats = get_all_chat_ids()
+    if "current_chat_id" not in st.session_state:
+        if len(st.session_state.chats) > 0:
+            st.session_state.current_chat_id = st.session_state.chats[-1]
+        else:
+            new_id = str(uuid.uuid4())
+            st.session_state.chats.append(new_id)
+            st.session_state.current_chat_id = new_id
+    if "chat_dict" not in st.session_state:
+        st.session_state.chat_dict = {}
+
+
+def render_sidebar():
+    with st.sidebar:
+        st.title("Chats")
+        if st.button("➕ New Chat"):
+            new_id = str(uuid.uuid4())
+            st.session_state.chats.append(new_id)
+            st.session_state.current_chat_id = new_id
+            config = {'configurable': {'thread_id': new_id}}
+            workflow.update_state(config, {"messages": [system]})
+            st.session_state.chat_dict[new_id] = "New Chat"
+            st.experimental_rerun()
+        for chat_id in st.session_state.chats:
+            if st.button(st.session_state.chat_dict.get(chat_id, "New Chat"), key=chat_id):
+                st.session_state.current_chat_id = chat_id
+
+
+def loadchats():
+    if "current_chat_id" not in st.session_state:
+        return []
+    config = {'configurable': {'thread_id': st.session_state.current_chat_id}}
+    state = workflow.get_state(config)
+    messages = state.values.get("messages", [])
+    for message in messages:
+        if isinstance(message, HumanMessage):
+            if message.content and message.content.strip():
+                with st.chat_message("human"):
+                    st.write(message.content)
+        elif isinstance(message, AIMessage):
+            if message.content and message.content.strip():
+                with st.chat_message("assistant"):
+                    st.write(message.content)
+        elif isinstance(message, ToolMessage):
+            with st.chat_message("assistant"):
+                st.info(f"Used a tool")
+    return messages
+
+
+load_session_state()
+render_sidebar()
+
+if "current_chat_id" in st.session_state:
+    loadchats()
+    user_input = st.chat_input("Your message: ")
+    if user_input is not None:
+        with st.chat_message("human"):
+            st.write(user_input)
+
+        with st.chat_message("assistant"):
+            response_placeholder = st.empty()
+            full_response = ""
+            for messages, metadata in workflow.stream(
+                {"messages": [system, HumanMessage(user_input)]},
+                config={'configurable': {'thread_id': st.session_state.current_chat_id}},
+                stream_mode="messages"
+            ):
+                if isinstance(messages, AIMessage) and messages.content.strip():
+                    full_response += messages.content
+                if isinstance(messages, ToolMessage) and messages.content.strip():
+                    st.info(f"Using appropriate tool")
+            response_placeholder.markdown(full_response)
+
+        st.experimental_rerun()