Update app.py

app.py

@@ -86,8 +86,7 @@ def embedding(chunk,query):
     chain = get_conversational_chain()
     response = chain({"input_documents": doc, "question": query}, return_only_outputs=True)
     print(response)
-    return response["output_text"]
-    #st.write("Reply: ", response["output_text"])
+    st.write("Reply: ", response["output_text"])
 
 
     
@@ -97,8 +96,6 @@ def main():
     st.title("Menu:")
     pdf_docs = st.file_uploader("Upload your PDF File and Click on the Submit & Process Button", accept_multiple_files=False, key="pdf_uploader")
     query = st.text_input("Ask a Question from the PDF File", key="query")
-    if 'messages' not in st.session_state:
-        st.session_state.messages = [{'role': 'assistant', "content": 'Hello! Upload a PDF and ask me anything about its content.'}]
     if st.button("Submit & Process", key="process_button"):
         with st.spinner("Processing..."):
             raw_text = get_pdf(pdf_docs)
@@ -106,14 +103,7 @@ def main():
             if query:
                 embedding(text_chunks,query)
             st.success("Done")
-    if query:
-        st.session_state.messages.append({'role': 'user', "content": query})
-        response = embedding(text_chunks,query)
-        st.session_state.messages.append({'role': 'assistant', "content": response})
-
-    for message in st.session_state.messages:
-        with st.chat_message(message['role']):
-            st.write(message['content'])
+
 
 if __name__ == "__main__":
     main()