Update app.py

app.py

@@ -14,18 +14,19 @@ def main():
     st.set_page_config(page_title="Chat PDF")
     st.header("Chat PDF 💬")
 
-    pdf = st.file_uploader("Upload your PDF file", type="pdf")
+    pdfs = st.file_uploader("Upload your PDF files", type="pdf", accept_multiple_files=True)
 
-    if pdf is not None:
+    text = ""
+    if pdfs:
         try:
-            pdf_reader = PdfReader(pdf)
-            text = ""
-            for page in pdf_reader.pages:
-                page_text = page.extract_text() or ""
-                text += page_text
+            for pdf in pdfs:
+                pdf_reader = PdfReader(pdf)
+                for page in pdf_reader.pages:
+                    page_text = page.extract_text() or ""
+                    text += page_text
 
             if not text:
-                st.write("No text could be extracted from the PDF.")
+                st.write("No text could be extracted from the PDFs.")
                 return
 
             char_text_splitter = CharacterTextSplitter(separator="\n", chunk_size=1000,
@@ -33,47 +34,20 @@ def main():
             text_chunks = char_text_splitter.split_text(text)
 
             embeddings = OpenAIEmbeddings()
-            docsearch = FAISS.from_texts(text_chunks, embeddings) 
-            llm = OpenAI() 
+            docsearch = FAISS.from_texts(text_chunks, embeddings)
+            llm = OpenAI()
 
-            
-            chain = load_qa_chain(llm, chain_type="stuff")
+            chain = load_qa_chain(llm, chain_type="your_chain_type_here")  # Replace "your_chain_type_here" with the actual chain type
 
-            while True:
-                # Ask a question
-                query = st.text_input("Type your question:")
-                
-                    #except Exception as e:
-                    #st.error(f"An error occurred: {e}")
+            query = st.text_input("Type your question:")
 
-            # Check if the user wants to exit
-                if query.lower() == 'exit':
-                    break
-
-            # Search documents for the query
-            #docs = docsearch.similarity_search(query)
             if query:
-                    docs = docsearch.similarity_search(query)
-                    response = chain.run(input_documents=docs, question=query)
-                    st.write(response)
-
-            # Get the response
-            response = chain.run(input_documents=docs, question=query)
-
-            # Print the question and response
-            print(" ")
-            print("Question:", query)
-            print("Answer:", response)
-            print("-" * 50) 
+                docs = docsearch.similarity_search(query)
+                response = chain.run(input_documents=docs, question=query)
+                st.write(response)
 
-    
-            #query = st.text_input("Type your question:")
-            #if query:
-             #   docs = docsearch.similarity_search(query)
-              #  response = chain.run(input_documents=docs, question=query)
-               # st.write(response)
         except Exception as e:
             st.error(f"An error occurred: {e}")
 
-if __name__ == '__main__':
+if __name__ == "__main__":
     main()