Update app.py

app.py

@@ -82,26 +82,26 @@ if query:
     )
 
     chain=load_qa_chain(llm, chain_type="stuff")
-    with get_openai_callback() as cb:
-        response = chain.run(input_documents=search, question=query)
-        print(cb)
+    
+    response = chain.run(input_documents=search, question=query)
+
         # st.write(response)
 
-        st.session_state['messages'].append({"role": "user", "content": query})
-        st.session_state['messages'].append({"role": "assistant", "content": response})
+    st.session_state['messages'].append({"role": "user", "content": query})
+    st.session_state['messages'].append({"role": "assistant", "content": response})
 
-        response_container = st.container()
-        # container for text box
-        container = st.container()
+    response_container = st.container()
+    # container for text box
+    container = st.container()
 
-        with container:
-            if query:
-                output = response
-                st.session_state['past'].append(query)
-                st.session_state['generated'].append(output)
+    with container:
+        if query:
+            output = response
+            st.session_state['past'].append(query)
+            st.session_state['generated'].append(output)
         
-        if st.session_state['generated']:
-            with response_container:
-                for i in range(len(st.session_state['generated'])):
-                    message(st.session_state["past"][i], is_user=True, key=str(i) + '_user')
-                    message(st.session_state["generated"][i], key=str(i))
+    if st.session_state['generated']:
+        with response_container:
+            for i in range(len(st.session_state['generated'])):
+                message(st.session_state["past"][i], is_user=True, key=str(i) + '_user')
+                message(st.session_state["generated"][i], key=str(i))