Update app.py

app.py

@@ -105,6 +105,8 @@ def footer():
 
 def main():
     st.title("Закупочный Патруль")
+    pdf_files_list = []
+    docx_files_list = []
 
     # Кнопка для загрузки PDF файла
     pdf_files = st.file_uploader("Загрузите PDF Извещения", type=["pdf"], accept_multiple_files=True)
@@ -113,6 +115,7 @@ def main():
         pdf_doc = fitz.open(stream=bytes_data, filetype="pdf")
         st.write("Файл Извещения:", pdf_file.name)
         #st.write(pdf_doc[0].get_text())
+        pdf_files_list.append(pdf_file)  # Добавляем файл в список
 
     # Кнопка для загрузки DOCX файла
     docx_files = st.file_uploader("Загрузите DOCX файл ПК", type=["docx"], accept_multiple_files=True)
@@ -121,11 +124,13 @@ def main():
         doc = Document(BytesIO(doc_bytes))
         #st.write(doc.paragraphs[0].text)
         st.write("файл ПК:", docx_file.name)
+        docx_files_list.append(docx_file)  # Добавляем файл в список
 
 
     # Кнопка начала проверки
     if st.button("Проверить", type="primary"):
-        if pdf_file is not None and docx_file is not None:
+        #if pdf_file is not None and docx_file is not None:
+        if len(pdf_files_list) > 0 and len(docx_files_list) > 0:
 
             print("=" * 80)
             st.write("_" * 80)
@@ -201,91 +206,92 @@ def main():
                 #st.write(":magenta[ERROR при сравнении, пожалуйста проверьте документ!]")
                 st.write(f"ERROR при сравнении {ex}")
                 # break
-        print("=" * 80)
-        st.write("_" * 80)
-        """
-            ЧАСТЬ 2 "Обеспечение гарантийных обязательств"
-        """
-
-        # Поле для вывода комментария после проверки
-        # result_comment_area = st.text_area(f"Результат проверки : {result_comment}")
-
-        # достаем параграфы текстовые из документов
-        paragraphs = get_doc_blocks(doc)
-        pdf_blocks = get_pdf_blocks(pdf_doc)
-
-        try:
-            garant_pdf = detect_garant_pdf(pdf_blocks, debug_print=False)
-            garant_docx = detect_garant_doc(paragraphs, debug_print=False)
-
-            MCKs = get_mck(pdf_blocks, debug_print=False)
-            price_doc = detect_garant_doc_price(paragraphs, debug_print=False)
-
-        #print(colored("ERROR при прочтении, пожалуйста проверьте документ!", 'magenta'))
-        except Exception as ex:
-            # st.write(f"ERROR при прочтении ЧАСТЬ 2, ")
-            st.write(f":red[ERROR при прочтении ЧАСТЬ 2, {ex}]")
-
-        check_garant(
-            garant_pdf=garant_pdf,
-            garant_docx=garant_docx,
-            price_doc=price_doc,
-            MCKs=MCKs
-        )
-        print("=" * 80)
-        st.write("_" * 80)
-        """
-            ЧАСТЬ 3 "ИКЗ"
-        """
-
-        ikz_pdf = get_ikz_pdf(pdf_blocks)
-        ikz_docx = get_ikz_doc(doc)
-        if ikz_docx and ikz_pdf:
-            ikz_docx_clean = set(
-                [el.replace("-", '') for el in ikz_docx]
-            )
-            ikz_pdf_clean = set(
-                [el.replace("-", '') for el in ikz_pdf]
+            print("=" * 80)
+            st.write("_" * 80)
+            """
+                ЧАСТЬ 2 "Обеспечение гарантийных обязательств"
+            """
+    
+            # Поле для вывода комментария после проверки
+            # result_comment_area = st.text_area(f"Результат проверки : {result_comment}")
+    
+            # достаем параграфы текстовые из документов
+            paragraphs = get_doc_blocks(doc)
+            pdf_blocks = get_pdf_blocks(pdf_doc)
+    
+            try:
+                garant_pdf = detect_garant_pdf(pdf_blocks, debug_print=False)
+                garant_docx = detect_garant_doc(paragraphs, debug_print=False)
+    
+                MCKs = get_mck(pdf_blocks, debug_print=False)
+                price_doc = detect_garant_doc_price(paragraphs, debug_print=False)
+    
+            #print(colored("ERROR при прочтении, пожалуйста проверьте документ!", 'magenta'))
+            except Exception as ex:
+                # st.write(f"ERROR при прочтении ЧАСТЬ 2, ")
+                st.write(f":red[ERROR при прочтении ЧАСТЬ 2, {ex}]")
+    
+            check_garant(
+                garant_pdf=garant_pdf,
+                garant_docx=garant_docx,
+                price_doc=price_doc,
+                MCKs=MCKs
             )
-            if ikz_docx_clean == ikz_pdf_clean:
-              #  print(colored("РАВНЫ", 'green'))
-                st.write(":green[РАВНЫ]")
-               # st.write("РАВНЫ")
+            print("=" * 80)
+            st.write("_" * 80)
+            """
+                ЧАСТЬ 3 "ИКЗ"
+            """
+    
+            ikz_pdf = get_ikz_pdf(pdf_blocks)
+            ikz_docx = get_ikz_doc(doc)
+            if ikz_docx and ikz_pdf:
+                ikz_docx_clean = set(
+                    [el.replace("-", '') for el in ikz_docx]
+                )
+                ikz_pdf_clean = set(
+                    [el.replace("-", '') for el in ikz_pdf]
+                )
+                if ikz_docx_clean == ikz_pdf_clean:
+                  #  print(colored("РАВНЫ", 'green'))
+                    st.write(":green[РАВНЫ]")
+                   # st.write("РАВНЫ")
+                else:
+                  #  print(colored("ОШИБКА", 'magenta'))
+                    st.write(":red[ОШИБКА]")
+                   # st.write("ОШИБКА")
+                    for el in sorted(ikz_docx_clean):
+                        if el not in ikz_pdf_clean:
+                           # print(f"ИКЗ {el} присутствует в .docx, но не в .pdf")
+                            st.write(f"ИКЗ {el} присутствует в .docx, но не в .pdf")
+                    print("-" * 60)
+                    st.write("-" * 60)
+                    for el in sorted(ikz_pdf_clean):
+                        if el not in ikz_docx_clean:
+                           # print(f"ИКЗ {el} присутствует в .pdf, но не в .docx")
+                            st.write(f"ИКЗ {el} присутствует в .pdf, но не в .docx")
             else:
-              #  print(colored("ОШИБКА", 'magenta'))
+               # print(colored("ПУСТОЕ ЗНАЧЕНИЕ", 'magenta'))
+                st.write(":red[ПУСТОЕ ЗНАЧЕНИЕ]")
+               # st.write("ПУСТОЕ ЗНАЧЕНИЕ")
+    
+            print("=" * 80)
+            st.write("_" * 80)
+            """
+                ЧАСТЬ 4 "часть 1 статьи 93 ФЗ-44 в ПК"
+            """
+    
+            fz93_docx = get_fz93_doc(doc)
+            if fz93_docx:  # проверка на пустоту, если есть ст.93 - ОШИБКА
+                print(colored("ОШИБКА", 'magenta'))
                 st.write(":red[ОШИБКА]")
-               # st.write("ОШИБКА")
-                for el in sorted(ikz_docx_clean):
-                    if el not in ikz_pdf_clean:
-                       # print(f"ИКЗ {el} присутствует в .docx, но не в .pdf")
-                        st.write(f"ИКЗ {el} присутствует в .docx, но не в .pdf")
-                print("-" * 60)
-                st.write("-" * 60)
-                for el in sorted(ikz_pdf_clean):
-                    if el not in ikz_docx_clean:
-                       # print(f"ИКЗ {el} присутствует в .pdf, но не в .docx")
-                        st.write(f"ИКЗ {el} присутствует в .pdf, но не в .docx")
-        else:
-           # print(colored("ПУСТОЕ ЗНАЧЕНИЕ", 'magenta'))
-            st.write(":red[ПУСТОЕ ЗНАЧЕНИЕ]")
-           # st.write("ПУСТОЕ ЗНАЧЕНИЕ")
-
-        print("=" * 80)
-        st.write("_" * 80)
-        """
-            ЧАСТЬ 4 "часть 1 статьи 93 ФЗ-44 в ПК"
-        """
-
-        fz93_docx = get_fz93_doc(doc)
-        if fz93_docx:  # проверка на пустоту, если есть ст.93 - ОШИБКА
-            print(colored("ОШИБКА", 'magenta'))
-            st.write(":red[ОШИБКА]")
-            print("ч. 1 ст. 93 ФЗ в ПК")
-            st.write("ч. 1 ст. 93 ФЗ в ПК")
+                print("ч. 1 ст. 93 ФЗ в ПК")
+                st.write("ч. 1 ст. 93 ФЗ в ПК")
+            else:
+                print(colored("ВЕРНО", 'green'))
+                st.write(":green[ВЕРНО]")
         else:
-            print(colored("ВЕРНО", 'green'))
-            st.write(":green[ВЕРНО]")
-
+            st.write("Нет документов для проверки")  # Сообщение, если нет загруженных файлов