Update src/streamlit_app.py

src/streamlit_app.py

@@ -97,12 +97,12 @@ def embed_sentence(text):
 
 
 def stream_response():
-    for char in extract_output(response).split(" "):
+    for char in extract_output(st.session_state.response).split(" "):
         yield char+" "
         time.sleep(0.1)  # Simulate a delay
 
 def stream_thoughts():
-    for char in extract_thinking(response).split(" "):
+    for char in extract_thinking(st.session_state.response).split(" "):
         yield char+" "
         time.sleep(0.1)  # Simulate a delay
 
@@ -170,7 +170,8 @@ if 'embeddings' not in st.session_state:
     st.session_state.embeddings = None
 if 'chunks' not in st.session_state:
     st.session_state.chunks = None
-
+if 'response' not in st.session_state:
+    st.session_state.response=''
 # Sidebar document upload
 st.sidebar.title("Uploading your document 📄")
 uploaded_file = st.sidebar.file_uploader(
@@ -224,10 +225,12 @@ if generate_button:
                 """
                 if st.session_state.db_transaction:
                     response=respond_chat(updated_text)
+                    st.session_state.response=response
 
             else: 
                 if st.session_state.db_transaction:
                     response=respond_chat(text)
+                    st.session_state.response=response
         except Exception as e:
                     st.error(f"Error during code generation: {e}")