colpali fix

pages/Multimodal_Conversational_Search.py

@@ -347,29 +347,29 @@ with st.sidebar:
     
     
     # st.subheader(":blue[Your multi-modal documents]")
-    # pdf_doc_ = st.file_uploader(
-    #     "Upload your PDFs here and click on 'Process'", accept_multiple_files=False)
+    pdf_doc_ = st.file_uploader(
+        "Upload your PDFs here and click on 'Process'", accept_multiple_files=False)
                     
                 
-    # pdf_docs = [pdf_doc_]
-    # if st.button("Process"):
-    #     with st.spinner("Processing"):
-    #         if os.path.isdir(parent_dirname+"/pdfs") == False:
-    #             os.mkdir(parent_dirname+"/pdfs")
+    pdf_docs = [pdf_doc_]
+    if st.button("Process"):
+        with st.spinner("Processing"):
+            if os.path.isdir(parent_dirname+"/pdfs") == False:
+                os.mkdir(parent_dirname+"/pdfs")
             
-    #         for pdf_doc in pdf_docs:
-    #             print(type(pdf_doc))
-    #             pdf_doc_name = (pdf_doc.name).replace(" ","_")
-    #             with open(os.path.join(parent_dirname+"/pdfs",pdf_doc_name),"wb") as f: 
-    #                 f.write(pdf_doc.getbuffer())  
+            for pdf_doc in pdf_docs:
+                print(type(pdf_doc))
+                pdf_doc_name = (pdf_doc.name).replace(" ","_")
+                with open(os.path.join(parent_dirname+"/pdfs",pdf_doc_name),"wb") as f: 
+                    f.write(pdf_doc.getbuffer())  
                     
-    #             request_ = { "bucket": s3_bucket_,"key": pdf_doc_name}
-    #             # if(st.session_state.input_copali_rerank):
-    #             #     copali.process_doc(request_)
-    #             # else:
-    #             rag_DocumentLoader.load_docs(request_)
-    #             print('lambda done')
-    #     st.success('you can start searching on your PDF')
+                request_ = { "bucket": s3_bucket_,"key": pdf_doc_name}
+                # if(st.session_state.input_copali_rerank):
+                #     copali.process_doc(request_)
+                # else:
+                rag_DocumentLoader.load_docs(request_)
+                print('lambda done')
+        st.success('you can start searching on your PDF')
         
      ############## haystach demo temporary addition ############       
     #if(pdf_doc_ is None or pdf_doc_ == ""):