Update app.py

app.py

@@ -2,39 +2,35 @@ import gradio as gr
 import PyPDF2
 import os
 import io
-import fitz  # PyMuPDF
 from reportlab.pdfgen import canvas
 from reportlab.lib.pagesizes import A4
 from reportlab.lib import colors
 
-# --- हेल्पर फ़ंक्शन 1: सभी PDF से प्रीव्यू इमेज बनाने के लिए ---
-def generate_previews(main_pdf, additional_pdfs):
+# --- हेल्पर फ़ंक्शन 1: सभी PDF से पेजों की सूची बनाने के लिए (बिना इमेज के) ---
+def list_all_pages(main_pdf, additional_pdfs):
     if not main_pdf:
-        return None, None, gr.update(visible=False), "कृपया पहले मुख्य PDF अपलोड करें."
+        return None, gr.update(visible=False), "कृपया पहले मुख्य PDF अपलोड करें."
     
     all_files = [main_pdf] + (additional_pdfs if additional_pdfs else [])
-    previews, page_labels, total_page_count = [], [], 0
+    page_labels = []
+    total_page_count = 0
 
     try:
-        os.makedirs("previews", exist_ok=True)
         for file_index, pdf_file in enumerate(all_files):
-            doc = fitz.open(pdf_file.name)
-            for page_index, page in enumerate(doc):
-                total_page_count += 1
-                pix = page.get_pixmap(dpi=72)
-                img_path = f"previews/preview_{total_page_count}.png"
-                pix.save(img_path)
-                previews.append(img_path)
-                # लेबल में फ़ाइल नंबर और पेज नंबर दोनों शामिल करें
+            # PyPDF2 का उपयोग करके पेजों की संख्या प्राप्त करें
+            reader = PyPDF2.PdfReader(pdf_file.name)
+            num_pages = len(reader.pages)
+            total_page_count += num_pages
+            for page_index in range(num_pages):
                 page_labels.append(f"File{file_index+1}-Page{page_index+1}")
-            doc.close()
         
         status = f"{total_page_count} पेज {len(all_files)} फाइलों से पाए गए. जिन पेजों को रखना है, उन्हें चुनें."
-        return previews, gr.update(choices=page_labels, value=page_labels), gr.update(visible=True), status
+        # केवल चेकबॉक्स ग्रुप और सेक्शन को अपडेट करें
+        return gr.update(choices=page_labels, value=page_labels), gr.update(visible=True), status
     except Exception as e:
-        return None, None, gr.update(visible=False), f"प्रीव्यू बनाने में त्रुटि: {e}"
+        return None, gr.update(visible=False), f"फ़ाइल पढ़ने में त्रुटि: {e}. क्या यह एक वैध PDF है?"
 
-# --- मुख्य PDF प्रोसेसिंग फ़ंक्शन ---
+# --- मुख्य PDF प्रोसेसिंग फ़ंक्शन (कोई बदलाव नहीं) ---
 def process_pdf(main_pdf, additional_pdfs, selected_pages, split_pdf, style, custom_left, custom_right, x_pos, y_pos):
     if not main_pdf:
         raise gr.Error("मुख्य PDF फ़ाइल गायब है!")
@@ -45,17 +41,13 @@ def process_pdf(main_pdf, additional_pdfs, selected_pages, split_pdf, style, cus
         all_files = [main_pdf] + (additional_pdfs if additional_pdfs else [])
         readers = [PyPDF2.PdfReader(f.name) for f in all_files]
         
-        # चरण 1: चुने हुए पेजों को फ़िल्टर और मर्ज करें
         filtered_writer = PyPDF2.PdfWriter()
         for label in selected_pages:
-            # लेबल को पार्स करें: 'FileX-PageY' -> file_idx=X-1, page_idx=Y-1
             parts = label.replace("File", "").split("-Page")
             file_idx, page_idx = int(parts[0]) - 1, int(parts[1]) - 1
-            
             if 0 <= file_idx < len(readers) and 0 <= page_idx < len(readers[file_idx].pages):
                 filtered_writer.add_page(readers[file_idx].pages[page_idx])
 
-        # चरण 2: इस नई, फ़िल्टर की गई PDF पर पुनः-नंबरिंग करें
         final_writer = PyPDF2.PdfWriter()
         filtered_pdf_stream = io.BytesIO()
         filtered_writer.write(filtered_pdf_stream)
@@ -86,7 +78,6 @@ def process_pdf(main_pdf, additional_pdfs, selected_pages, split_pdf, style, cus
         with open(final_numbered_path, "wb") as f:
             final_writer.write(f)
 
-        # चरण 3: यदि चुना गया है तो इस अंतिम PDF को विभाजित करें
         if split_pdf:
             reader = PyPDF2.PdfReader(final_numbered_path)
             odd_writer, even_writer = PyPDF2.PdfWriter(), PyPDF2.PdfWriter()
@@ -107,20 +98,20 @@ def process_pdf(main_pdf, additional_pdfs, selected_pages, split_pdf, style, cus
     except Exception as e:
         raise gr.Error(f"एक गंभीर त्रुटि हुई: {e}")
 
-# --- Gradio इंटरफ़ेस ---
+# --- Gradio इंटरफ़ेस (UI में बदलाव) ---
 with gr.Blocks(theme=gr.themes.Default(primary_hue="blue")) as demo:
-    gr.Markdown("# 🚀 PDF वर्कफ़्लो स्टेशन v4.0 🚀\nPDF जोड़ें, पेज हटाएं, पुनः-नंबर करें और विभाजित करें!")
+    gr.Markdown("# 🚀 PDF वर्कफ़्लो स्टेशन v5.0 (लाइट) 🚀\nतेज, बिना प्रीव्यू वाला संस्करण!")
     
     # चरण 1: अपलोड
     gr.Markdown("### 1. अपनी PDF फाइलें अपलोड करें")
     main_pdf_input = gr.File(label="मुख्य PDF", file_types=[".pdf"])
     additional_pdfs_input = gr.File(label="(वैकल्पिक) जोड़ने के लिए और PDF", file_count="multiple", file_types=[".pdf"])
-    generate_preview_button = gr.Button("🔍 सभी पेजों का प्रीव्यू जेनरेट करें")
+    list_pages_button = gr.Button("📂 सभी पेजों की सूची जेनरेट करें")
     
-    # चरण 2: प्रीव्यू और चयन
-    with gr.Column(visible=False) as preview_section:
+    # चरण 2: चयन (बिना इमेज के)
+    with gr.Column(visible=False) as selection_section:
         gr.Markdown("### 2. अंतिम PDF में रखने के लिए पेज चुनें")
-        preview_gallery = gr.Gallery(label="सभी पेजों का प्रीव्यू", show_label=True, columns=8, object_fit="contain", height="auto")
+        gr.Markdown("<p style='color:red;'>ध्यान दें: जिन पेजों को आप अनचेक करेंगे, वे अंतिम फ़ाइल से हटा दिए जाएंगे.</p>")
         page_selector = gr.CheckboxGroup(label="पेज चुनें")
 
     # चरण 3: सेटिंग्स और प्रोसेसिंग
@@ -140,16 +131,16 @@ with gr.Blocks(theme=gr.themes.Default(primary_hue="blue")) as demo:
     
     # चरण 4: आउटपुट
     with gr.Column():
-        pdf_output_final = gr.File(label="अंतिम PDF डाउनलोड करें (मर्ज और नंबर की हुई)")
+        pdf_output_final = gr.File(label="अंतिम PDF डाउनलोड करें")
         with gr.Row():
             pdf_output_odd = gr.File(label="विषम पेज (Odd) डाउनलोड करें")
             pdf_output_even = gr.File(label="सम पेज (Even) डाउनलोड करें")
 
     # --- UI Logic ---
-    generate_preview_button.click(
-        fn=generate_previews, 
+    list_pages_button.click(
+        fn=list_all_pages, 
         inputs=[main_pdf_input, additional_pdfs_input], 
-        outputs=[preview_gallery, page_selector, preview_section, status_output]
+        outputs=[page_selector, selection_section, status_output]
     )
     
     def toggle_custom_style(style): return gr.update(visible=(style == "Custom..."))