moved chat loop into chat_loop() following major refactoring

app.py

@@ -1,13 +1,14 @@
 import streamlit as st
 import yaml
-from chat_client import chat
 from components.sidebar import sidebar
 from components.prompt_engineering_dashboard import prompt_engineering_dashboard
-from components.stream_handler import stream_handler
-from components.show_source import show_source
 from components.chat_box import chat_box
-from components.generate_chat_stream import generate_chat_stream
 from components.init_state import init_state
+from components.chat_loop import chat_loop
+
+# from components.show_source import show_source
+# from components.stream_handler import stream_handler
+# from components.generate_chat_stream import generate_chat_stream
 
 
 with open("config.yaml", "r") as file:
@@ -27,21 +28,6 @@ prompt_engineering_dashboard(st.session_state, config)
 
 sidebar(st.session_state, config)
 
-chat_box(st.session_state)
-
-if prompt := st.chat_input("Generate Ebook"):
-    st.chat_message("user").markdown(prompt)
-    st.session_state.messages.append({"role": "user", "content": prompt})
-
-    chat_stream, links = generate_chat_stream(st.session_state, prompt, config)
-
-    with st.chat_message("assistant"):
-        placeholder = st.empty()
-        full_response = stream_handler(
-            st.session_state, chat_stream, prompt, placeholder
-        )
-        if st.session_state.rag_enabled:
-            show_source(links)
+chat_box(st.session_state, config)
 
-    st.session_state.history.append([prompt, full_response])
-    st.session_state.messages.append({"role": "assistant", "content": full_response})
+chat_loop(st.session_state, config)

components/chat_box.py

@@ -1,7 +1,7 @@
 import streamlit as st
 
 
-def chat_box(session_state):
+def chat_box(session_state, config):
     for message in session_state.messages:
         with st.chat_message(message["role"]):
             st.markdown(message["content"])

components/chat_loop.py

@@ -0,0 +1,23 @@
+import streamlit as st
+from components.generate_chat_stream import generate_chat_stream
+from components.stream_handler import stream_handler
+from components.show_source import show_source
+
+
+def chat_loop(session_state, config):
+    if prompt := st.chat_input("Generate Ebook"):
+        st.chat_message("user").markdown(prompt)
+        session_state.messages.append({"role": "user", "content": prompt})
+
+        chat_stream, links = generate_chat_stream(session_state, prompt, config)
+
+        with st.chat_message("assistant"):
+            placeholder = st.empty()
+            full_response = stream_handler(
+                session_state, chat_stream, prompt, placeholder
+            )
+            if session_state.rag_enabled:
+                show_source(links)
+
+        session_state.history.append([prompt, full_response])
+        session_state.messages.append({"role": "assistant", "content": full_response})