Update app.py

app.py

@@ -6,7 +6,7 @@ from reportlab.pdfgen import canvas
 from reportlab.lib.pagesizes import A4
 from reportlab.lib import colors
 
-# --- हेल्पर फ़ंक्शन 1: सभी PDF से पेजों की सूची बनाने के लिए (बिना इमेज के) ---
+# --- हेल्पर फ़ंक्शन 1: सभी PDF से पेजों की सूची बनाने के लिए ---
 def list_all_pages(main_pdf, additional_pdfs):
     if not main_pdf:
         return None, gr.update(visible=False), "कृपया पहले मुख्य PDF अपलोड करें."
@@ -17,7 +17,6 @@ def list_all_pages(main_pdf, additional_pdfs):
 
     try:
         for file_index, pdf_file in enumerate(all_files):
-            # PyPDF2 का उपयोग करके पेजों की संख्या प्राप्त करें
             reader = PyPDF2.PdfReader(pdf_file.name)
             num_pages = len(reader.pages)
             total_page_count += num_pages
@@ -25,12 +24,11 @@ def list_all_pages(main_pdf, additional_pdfs):
                 page_labels.append(f"File{file_index+1}-Page{page_index+1}")
         
         status = f"{total_page_count} पेज {len(all_files)} फाइलों से पाए गए. जिन पेजों को रखना है, उन्हें चुनें."
-        # केवल चेकबॉक्स ग्रुप और सेक्शन को अपडेट करें
         return gr.update(choices=page_labels, value=page_labels), gr.update(visible=True), status
     except Exception as 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 फ़ाइल गायब है!")
@@ -41,27 +39,29 @@ 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]
         
-        filtered_writer = PyPDF2.PdfWriter()
+        final_writer = PyPDF2.PdfWriter()
+        new_page_counter = 0
+
+        # एक ही चरण में फ़िल्टर और नंबर करें (यह अधिक मेमोरी-कुशल है)
         for label in selected_pages:
+            new_page_counter += 1 # नई PDF के लिए पेज नंबर 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])
+            
+            if not (0 <= file_idx < len(readers) and 0 <= page_idx < len(readers[file_idx].pages)):
+                continue # अमान्य लेबल को छोड़ दें
+                
+            page = readers[file_idx].pages[page_idx]
 
-        final_writer = PyPDF2.PdfWriter()
-        filtered_pdf_stream = io.BytesIO()
-        filtered_writer.write(filtered_pdf_stream)
-        filtered_pdf_stream.seek(0)
-        
-        filtered_reader = PyPDF2.PdfReader(filtered_pdf_stream)
-        for i, page in enumerate(filtered_reader.pages):
-            new_page_number = i + 1
+            # इस पेज के लिए वॉटरमार्क बनाएं
             packet = io.BytesIO()
             can = canvas.Canvas(packet, pagesize=A4)
             if style == "Custom...":
-                page_num_text = f"{custom_left}{new_page_number}{custom_right}"
+                page_num_text = f"{custom_left}{new_page_counter}{custom_right}"
             else:
-                page_num_text = style.replace("1", str(new_page_number))
+                page_num_text = style.replace("1", str(new_page_counter))
             
             can.setFont("Helvetica-Bold", 12)
             can.setFillColor(colors.black)
@@ -69,23 +69,28 @@ def process_pdf(main_pdf, additional_pdfs, selected_pages, split_pdf, style, cus
             can.save()
             packet.seek(0)
             
-            watermark = PyPDF2.PdfReader(packet)
-            page.merge_page(watermark.pages[0])
+            # वॉटरमार्क को सीधे पेज पर मर्ज करें
+            watermark_reader = PyPDF2.PdfReader(packet)
+            page.merge_page(watermark_reader.pages[0])
+            
+            # संशोधित पेज को अंतिम राइटर में जोड़ें
             final_writer.add_page(page)
 
+        # अंतिम PDF को सहेजें
         original_filename = os.path.basename(getattr(main_pdf, 'orig_name', main_pdf.name))
-        final_numbered_path = f"Final_Merged_{original_filename}"
+        final_numbered_path = f"Final_Output_{original_filename}"
         with open(final_numbered_path, "wb") as f:
             final_writer.write(f)
 
+        # यदि चुना गया है तो विभाजित करें
         if split_pdf:
-            reader = PyPDF2.PdfReader(final_numbered_path)
+            # हम final_writer का उपयोग कर सकते हैं, फाइल को फिर से पढ़ने की जरूरत नहीं है
             odd_writer, even_writer = PyPDF2.PdfWriter(), PyPDF2.PdfWriter()
-            for i, page in enumerate(reader.pages):
+            for i, page in enumerate(final_writer.pages):
                 (odd_writer if (i + 1) % 2 != 0 else even_writer).add_page(page)
             
-            odd_path = f"odd_merged_{original_filename}" if odd_writer.pages else None
-            even_path = f"even_merged_{original_filename}" if even_writer.pages else None
+            odd_path = f"odd_output_{original_filename}" if odd_writer.pages else None
+            even_path = f"even_output_{original_filename}" if even_writer.pages else None
             if odd_path:
                 with open(odd_path, "wb") as f: odd_writer.write(f)
             if even_path:
@@ -98,23 +103,20 @@ def process_pdf(main_pdf, additional_pdfs, selected_pages, split_pdf, style, cus
     except Exception as e:
         raise gr.Error(f"एक गंभीर त्रुटि हुई: {e}")
 
-# --- Gradio इंटरफ़ेस (UI में बदलाव) ---
+# --- Gradio इंटरफ़ेस (UI में कोई बदलाव नहीं) ---
 with gr.Blocks(theme=gr.themes.Default(primary_hue="blue")) as demo:
-    gr.Markdown("# 🚀 PDF वर्कफ़्लो स्टेशन v5.0 (लाइट) 🚀\nतेज, बिना प्रीव्यू वाला संस्करण!")
+    gr.Markdown("# 🚀 PDF वर्कफ़्लो स्टेशन (स्थिर और तेज) 🚀\nPDF जोड़ें, पेज हटाएं, पुनः-नंबर करें और विभाजित करें!")
     
-    # चरण 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"])
     list_pages_button = gr.Button("📂 सभी पेजों की सूची जेनरेट करें")
     
-    # चरण 2: चयन (बिना इमेज के)
     with gr.Column(visible=False) as selection_section:
         gr.Markdown("### 2. अंतिम PDF में रखने के लिए पेज चुनें")
         gr.Markdown("<p style='color:red;'>ध्यान दें: जिन पेजों को आप अनचेक करेंगे, वे अंतिम फ़ाइल से हटा दिए जाएंगे.</p>")
         page_selector = gr.CheckboxGroup(label="पेज चुनें")
 
-    # चरण 3: सेटिंग्स और प्रोसेसिंग
     gr.Markdown("### 3. नंबरिंग स्टाइल और अंतिम आउटपुट")
     split_checkbox = gr.Checkbox(label="अंतिम PDF को सम/विषम पेजों में विभाजित करें", value=False)
     with gr.Accordion("Page Numbering Studio", open=True):
@@ -129,14 +131,12 @@ with gr.Blocks(theme=gr.themes.Default(primary_hue="blue")) as demo:
     process_button = gr.Button("🚀 PDF प्रोसेस करें", variant="primary")
     status_output = gr.Textbox(label="स्थिति", interactive=False)
     
-    # चरण 4: आउटपुट
     with gr.Column():
         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 ---
     list_pages_button.click(
         fn=list_all_pages, 
         inputs=[main_pdf_input, additional_pdfs_input],