Integrating chat to llm response

app.py

@@ -36,12 +36,15 @@ print(result['result'])
 
 
 # Streamed response emulator
-def response_generator():
-    response = random.choice(
-        [
-            result['result']
-        ]
-    )
+def response_generator(prompt):
+    # response = random.choice(
+    #     [
+    #         result['result']
+    #     ]
+    # )
+
+    response = qa_chain({"query": prompt})['result']   
+    
     for word in response.split():
         yield word + " "
         time.sleep(0.05)
@@ -68,6 +71,6 @@ if prompt := st.chat_input("What is up?"):
 
     # Display assistant response in chat message container
     with st.chat_message("assistant"):
-        response = st.write_stream(response_generator())
+        response = st.write_stream(response_generator(prompt))
     # Add assistant response to chat history
     st.session_state.messages.append({"role": "assistant", "content": response})