Update src/Report/Report.py

src/Report/Report.py

@@ -2,89 +2,107 @@ import os
 import PyPDF2
 from reportlab.lib.pagesizes import letter
 from reportlab.pdfgen import canvas
-import shutil
+from screeninfo import get_monitors
+import sys
+import codecs
 
-def ensure_directory_exists(directory):
-    """Ensure the given directory exists."""
-    os.makedirs(directory, exist_ok=True)
+sys.stdout = codecs.getwriter("utf-8")(sys.stdout.buffer)
 
-def merge_pdfs(pdf_list, output_path, elongated_files=[]):
-    """Merge multiple PDFs into a single PDF while handling missing files."""
+def get_screen_size():
+    """Get the primary monitor screen size."""
+    try:
+        monitor = get_monitors()[0]  # Get primary monitor
+        screen_width = monitor.width
+        screen_height = monitor.height
+        return screen_width, int(screen_height * 1.5)  # Scale height
+    except Exception as e:
+        print(f"Error getting screen size: {e}")
+    return 1000, 1500  # Default size
+
+def add_padding_to_pdf(input_pdf, output_pdf, padding):
+    """Add padding to a PDF file."""
+    try:
+        reader = PyPDF2.PdfReader(input_pdf)
+        writer = PyPDF2.PdfWriter()
+
+        for page in reader.pages:
+            original_width = page.mediabox.width
+            original_height = page.mediabox.height
+
+            # New dimensions with padding
+            new_width = original_width + 2 * padding
+            new_height = original_height + 2 * padding
+
+            # Create a new canvas
+            packet = canvas.Canvas(output_pdf, pagesize=(new_width, new_height))
+            packet.translate(padding, padding)  # Center original content
+            packet.save()
+
+            writer.add_page(page)
+
+        with open(output_pdf, "wb") as out_file:
+            writer.write(out_file)
+
+        print(f"✅ Padding added: {input_pdf} -> {output_pdf}")
+
+    except Exception as e:
+        print(f"⚠️ Error adding padding to {input_pdf}: {e}")
+
+def merge_pdfs(pdf_list, output_path, elongated_files=[], normalize_size=True):
+    """Merge multiple PDFs into a single PDF."""
     pdf_merger = PyPDF2.PdfMerger()
+
+    # Resolve absolute paths
     pdf_list = [os.path.abspath(pdf) for pdf in pdf_list]
 
     for pdf in pdf_list:
         if not os.path.exists(pdf):
-            print(f"⚠️ Skipping missing file: {pdf}")
+            print(f"⚠️ File not found: {pdf}")
             continue
 
         try:
-            print(f"✅ Adding: {pdf}")
+            if pdf in elongated_files:
+                print(f"🔍 Processing elongated file: {pdf}")
+                # Add additional elongation logic here if needed
             pdf_merger.append(pdf)
+            print(f"✅ Added: {pdf}")
         except Exception as e:
             print(f"⚠️ Could not add {pdf}: {e}")
 
     try:
         pdf_merger.write(output_path)
-        print(f"📄 Merged PDF saved: {output_path}")
+        print(f"📄 Merged PDF saved as: {output_path}")
     except Exception as e:
-        print(f"❌ Error saving merged PDF: {e}")
+        print(f"Error saving merged PDF: {e}")
     finally:
         pdf_merger.close()
 
-def convert_png_to_pdf(png_path, output_pdf):
-    """Convert a PNG image into a PDF and save it in a cloud-friendly location."""
-    from PIL import Image
-
-    try:
-        ensure_directory_exists(os.path.dirname(output_pdf))
-
-        img = Image.open(png_path)
-        img.convert('RGB').save(output_pdf, "PDF")
-
-        print(f"📄 PDF saved: {output_pdf}")
-    except Exception as e:
-        print(f"❌ Error converting PNG to PDF: {e}")
-
 if __name__ == "__main__":
-    # ✅ Define cloud-safe directories
-    base_dir = "/home/user/app"  # Hugging Face Spaces safe storage
-    template_dir = os.path.join(base_dir, "data/reports/template_PDF")
-    report_stats_dir = os.path.join(base_dir, "data/reports/report_stats")
-    output_dir = os.path.join(base_dir, "src/Report")
-
-    # ✅ Ensure directories exist
-    ensure_directory_exists(template_dir)
-    ensure_directory_exists(report_stats_dir)
-    ensure_directory_exists(output_dir)
-
-    # ✅ List of PDF files to merge
+    # List of PDF files to merge
     pdf_files = [
-        os.path.join(base_dir, "src/Report/1updated.pdf"),
-        os.path.join(report_stats_dir, "2.pdf"),
-        os.path.join(report_stats_dir, "3.pdf"),
-        os.path.join(report_stats_dir, "objective.pdf"),
-        os.path.join(template_dir, "brand marketing.pdf"),
-        os.path.join(template_dir, "content marketing.pdf"),
-        os.path.join(template_dir, "social media marketing.pdf"),
-        os.path.join(report_stats_dir, "last.pdf")
+        "src/Report/1updated.pdf",
+        "data/reports/report_stats/2.pdf",
+        "data/reports/report_stats/3.pdf",
+        "data/reports/report_stats/objective.pdf",
+        "data/reports/template_PDF/brand marketing.pdf",
+        "data/reports/template_PDF/content marketing.pdf",
+        "data/reports/template_PDF/social media marketing.pdf",
+        "data/reports/report_stats/last.pdf"
     ]
 
-    # ✅ Define elongated PDFs (for special processing)
     elongated_pdfs = [
-        os.path.join(template_dir, "brand marketing.pdf"),
-        os.path.join(template_dir, "content marketing.pdf"),
-        os.path.join(template_dir, "social media marketing.pdf")
+        "data/reports/template_PDF/brand marketing.pdf",
+        "data/reports/template_PDF/content marketing.pdf",
+        "data/reports/template_PDF/social media marketing.pdf"
     ]
 
-    # ✅ Strictly enforce output file name and location
-    output_file = os.path.join(output_dir, "report.pdf")
+    # Define output directory and strict file name
+    output_dir = "src/Report"
+    os.makedirs(output_dir, exist_ok=True)  # Create the directory if it doesn't exist
+    output_file = os.path.join(output_dir, "report.pdf")  # Enforce strict file name as report.pdf
 
-    # ✅ Merge PDFs
+    # Merge the PDFs and save the result in the specified directory
     merge_pdfs(pdf_files, output_file, elongated_files=elongated_pdfs)
 
-    # ✅ Make the report available for download
-    visible_output = os.path.join(base_dir, "report.pdf")
-    shutil.copy(output_file, visible_output)
 
-    print(f"📂 Download your report from: {visible_output}")
+