Upload 2 files

main.py

@@ -0,0 +1,51 @@
+from pdf2image import convert_from_path
+from PIL import Image
+import os
+import time
+import shutil
+
+def convert_pdf_to_webp(pdf_path):
+    images = convert_from_path(pdf_path)
+    if not os.path.exists('img'):
+        os.makedirs('img')
+    for i, image in enumerate(images):
+        image_path = os.path.join('img', f"page_{i}.webp")
+        image.save(image_path, "WEBP")
+
+def convert_webp_to_pdf(output_pdf):
+    start_time = time.time()
+    image_files = [f for f in os.listdir('img') if f.lower().endswith(('.png', '.jpg', '.jpeg', '.bmp', '.gif', '.webp'))]
+    if not image_files:
+        print("هیچ فایل تصویری در پوشه یافت نشد.")
+        return
+    image_files.sort()
+    image_paths = [os.path.join('img', f) for f in image_files]
+    images = [Image.open(image_path) for image_path in image_paths]
+    images[0].save(output_pdf, "PDF", resolution=100.0, save_all=True, append_images=images[1:])
+    print(f"فایل PDF با موفقیت ایجاد شد: {output_pdf}")
+    elapsed_time = time.time() - start_time
+    print(f"WebP to PDF conversion took {elapsed_time:.2f} seconds")
+
+def convert_images_to_webp(image_paths):
+    if not os.path.exists('img'):
+        os.makedirs('img')
+    for image_path in image_paths:
+        try:
+            img = Image.open(image_path)
+            filename = os.path.basename(image_path)
+            output_filename = os.path.splitext(filename)[0] + '.webp'
+            output_path = os.path.join('img', output_filename)
+            img.save(output_path, 'WEBP')
+            print(f'Image {filename} converted to {output_filename}')
+        except Exception as e:
+            print(f'Error converting {image_path}: {e}')
+
+def filig(file_path):
+	file_size_bytes = os.path.getsize(file_path)
+	file_siz = file_size_bytes / (1024 * 1024)
+	return file_siz
+
+def Percentage(file1,file2):
+	f =  filig(file1) - filig(file2)
+	g = int((f / filig(file1)) * 100)
+	return str(g) + "%"

تست.py

@@ -0,0 +1,57 @@
+#pip install streamlit pdf2image pillow
+import streamlit as st
+from main import convert_pdf_to_webp, convert_webp_to_pdf, convert_images_to_webp,Percentage,filig
+import os
+import shutil
+
+def main():
+    st.title("تبدیل فایل‌ها")
+    st.write("فایل های PDF تا 80 درصد حجم کمتر!")
+
+    option = st.sidebar.selectbox("انتخاب عملیات", ("تبدیل تصاویر به PDF کم حجم", "کاهش حجم PDF"))
+
+    if option == "تبدیل تصاویر به PDF کم حجم":
+        st.subheader("تبدیل تصاویر به PDF کم حجم")
+        uploaded_files = st.file_uploader("تصاویر خود را آپلود کنید", type=["png", "jpg", "jpeg", "bmp", "gif"], accept_multiple_files=True)
+
+        if uploaded_files:
+            image_paths = []
+            kol = 0
+            for uploaded_file in uploaded_files:
+                image_path = os.path.join("temp", uploaded_file.name)
+                kol += filig(uploaded_file)
+                with open(image_path, "wb") as f:
+                    f.write(uploaded_file.getbuffer())
+                image_paths.append(image_path)
+
+            if st.button("تبدیل به PDF کم حجم"):
+                convert_images_to_webp(image_paths)
+                output_pdf = "output.pdf"
+                convert_webp_to_pdf(output_pdf)
+                with open(output_pdf, "rb") as f:
+                    st.download_button("دانلود PDF", f, file_name=output_pdf)
+                dar = Percentage(kol,output_pdf)
+                st.success(f"حجم PDF نسبت به حجم تصاویر {dar} کاهش یافته است!")
+
+    elif option == "کاهش حجم PDF":
+        st.subheader("کاهش حجم PDF")
+        uploaded_file = st.file_uploader("فایل PDF خود را آپلود کنید", type=["pdf"], accept_multiple_files=True)
+
+        if uploaded_file:
+            pdf_path = os.path.join("temp", uploaded_file.name)
+            with open(pdf_path, "wb") as f:
+                f.write(uploaded_file.getbuffer())
+
+            if st.button("کاهش حجم PDF"):
+                convert_pdf_to_webp(pdf_path)
+                output_pdf = "output_converted.pdf"
+                convert_webp_to_pdf(output_pdf)
+                with open(output_pdf, "rb") as f:
+                    st.download_button("دانلود PDF", f, file_name=output_pdf)
+                dar = Percentage(uploaded_file,output_pdf)
+                st.success(f"حجم pdf {dar} کاهش یافت!")
+
+if __name__ == "__main__":
+    if not os.path.exists('temp'):
+        os.makedirs('temp')
+    main()