Update src/brand.py

src/brand.py

@@ -2,26 +2,19 @@ import base64
 import os
 import sys
 import time
+import pandas as pd
 from playwright.sync_api import sync_playwright
 from PIL import Image
-import pandas as pd
-
-def save_html_file(file_name, html_content):
-    with open(file_name, 'w') as file:
-        file.write(html_content)
 
-def encode_image_to_base64(image_path):
-    with open(image_path, "rb") as image_file:
-        return base64.b64encode(image_file.read()).decode('utf-8')
+# ✅ Define temp folder for Hugging Face Spaces
+TEMP_FOLDER = "/tmp"
+HTML_FILE = os.path.join(TEMP_FOLDER, "brand_marketing.html")
+SCREENSHOT_FILE = os.path.join(TEMP_FOLDER, "brand_marketing.png")
+PDF_FOLDER = os.path.join(TEMP_FOLDER, "template_PDF")
+os.makedirs(PDF_FOLDER, exist_ok=True)  # Ensure folder exists
 
-# Allow user to upload images for logo and product images
-logo_image_path = r"src/templates_images/Component 3.png"
-cola_image_path = r"src/templates_images/Frame 52.png"  
-# Encode images to base64
-logo_base64 = encode_image_to_base64(logo_image_path)
-cola_base64 = encode_image_to_base64(cola_image_path)
 def save_html_file(file_name, html_content):
-    with open(file_name, 'w') as file:
+    with open(file_name, "w", encoding="utf-8") as file:
         file.write(html_content)
 
 def encode_image_to_base64(image_path):
@@ -29,11 +22,44 @@ def encode_image_to_base64(image_path):
         with open(image_path, "rb") as img_file:
             return base64.b64encode(img_file.read()).decode("utf-8")
     except FileNotFoundError:
-        print(f"Image not found: {image_path}")
+        print(f"⚠️ Image not found: {image_path}")
         return ""
     except Exception as e:
-        print(f"Error encoding image {image_path}: {e}")
+        print(f"⚠️ Error encoding image {image_path}: {e}")
         return ""
+
+def capture_screenshot():
+    """Capture a screenshot using Playwright."""
+    os.system("playwright install --with-deps")  # Ensure Playwright is installed
+
+    # Delete existing screenshot
+    if os.path.exists(SCREENSHOT_FILE):
+        os.remove(SCREENSHOT_FILE)
+
+    with sync_playwright() as p:
+        browser = p.chromium.launch(headless=True)
+        page = browser.new_page()
+        page.goto(f"file://{HTML_FILE}")
+        page.screenshot(path=SCREENSHOT_FILE, full_page=True)
+        browser.close()
+    print(f"✅ Screenshot saved: {SCREENSHOT_FILE}")
+
+def convert_to_pdf():
+    """Convert the screenshot to a PDF file."""
+    try:
+        PDF_FILE = os.path.join(PDF_FOLDER, "brand_marketing.pdf")
+        img = Image.open(SCREENSHOT_FILE)
+        img.convert("RGB").save(PDF_FILE, "PDF")
+        print(f"✅ PDF saved: {PDF_FILE}")
+    except Exception as e:
+        print(f"❌ Error converting PNG to PDF: {e}")
+
+if __name__ == "__main__":
+    # Ensure previous HTML/Screenshot are removed
+    if os.path.exists(HTML_FILE):
+        os.remove(HTML_FILE)
+    if os.path.exists(SCREENSHOT_FILE):
+        os.remove(SCREENSHOT_FILE)
 # Function to generate HTML for Social Media Marketing
 # Function to generate HTML for Brand Marketing
 def generate_brand_marketing_html(product_image_base64_1, competitor_image_base64_1, product_image_base64_2, competitor_image_base64_2, donts_html, suggestions_html, company_name):
@@ -254,169 +280,8 @@ def generate_brand_marketing_html(product_image_base64_1, competitor_image_base6
 
     """
 
-# Function to parse "Product_output_cleaned.txt" for Don'ts and Suggestions specific to Brand Marketing
-def parse_cleaned_file_brand_marketing(file_path):
-    with open(file_path, "r") as file:
-        content = file.read()
-
-    sections = content.split("==================================================")
-    for section in sections:
-        lines = section.strip().split("\n")
-        if lines and "Brand Marketing" in lines[0]:
-            donts = []
-            suggestions = []
-            mode = None
-            for line in lines[1:]:
-                if line.startswith("Don'ts:"):
-                    mode = "donts"
-                elif line.startswith("Suggestions:"):
-                    mode = "suggestions"
-                elif mode == "donts" and line.startswith("-"):
-                    donts.append(line.lstrip("- "))
-                elif mode == "suggestions" and line.startswith("-"):
-                    suggestions.append(line.lstrip("- "))
-            return "<br>".join(donts), "<br>".join(suggestions)
-
-    return "", ""
-
-# Function to process Brand Marketing and generate HTML
-def process_brand_marketing(data, base_image_dir, output_file, cleaned_file_path, company_name):
-    # Filter for Brand Marketing category
-    brand_data = data[data["Category"] == "Brand Marketing"]
-    
-    if brand_data.empty:
-        print("No Brand Marketing data found in the provided Excel file.")
-        return
-
-    # Parse Don'ts and Suggestions
-    donts_html, suggestions_html = parse_cleaned_file_brand_marketing(cleaned_file_path)
-
-    # Ensure there are at least two rows
-    if len(brand_data) < 2:
-        print("Not enough rows for two product and competitor image comparisons.")
-        return
-
-    # Get the first two records (assuming these are needed)
-    brand_row_1 = brand_data.iloc[0]
-    brand_row_2 = brand_data.iloc[1]
-
-    product_image_path_1 = os.path.join(base_image_dir, brand_row_1["Product_Image_Name"])
-    competitor_image_path_1 = os.path.join(base_image_dir, brand_row_1["Competitor_Image_Name"])
-
-    product_image_path_2 = os.path.join(base_image_dir, brand_row_2["Product_Image_Name"])
-    competitor_image_path_2 = os.path.join(base_image_dir, brand_row_2["Competitor_Image_Name"])
-
-    # Encode images to Base64
-    product_image_base64_1 = encode_image_to_base64(product_image_path_1)
-    competitor_image_base64_1 = encode_image_to_base64(competitor_image_path_1)
-
-    product_image_base64_2 = encode_image_to_base64(product_image_path_2)
-    competitor_image_base64_2 = encode_image_to_base64(competitor_image_path_2)
-
- # Generate HTML content
-    html_content = generate_brand_marketing_html(
-        product_image_base64_1,
-        competitor_image_base64_1,
-        product_image_base64_2,
-        competitor_image_base64_2,
-        donts_html,
-        suggestions_html,
-        company_name
-    )
-
-    # Save the HTML file
-    with open(output_file, "w", encoding="utf-8") as f:
-        f.write(html_content)
-
-    print(f"HTML file for Brand Marketing has been saved as: {output_file}")
-
-# Main script for Content Marketing
-if __name__ == "__main__":
-    if len(sys.argv) > 1:
-        company_name = sys.argv[1]  # The second argument passed will be the company_name
-    else:
-        company_name = "Default_Company"  # Default value if no argument is passed
-    # Load the Excel file
-    file_path = "Output_File/excel/top_3_sd_results.xlsx"  # Replace with the path to your Excel file
-    data = pd.read_excel(file_path)
-
-    base_image_dir = ""  # Replace with the actual directory where your images are stored
-
-    # Path to the cleaned file with Don'ts and Suggestions
-    cleaned_file_path = "data/output_generated_file/Product_output_cleaned.txt"  # Replace with the path to your cleaned file
-
-    # Output HTML file
-    output_file = "src/templates/brand_marketing.html"
-
-    # Generate HTML for Content Marketing
-    process_brand_marketing(data, base_image_dir, output_file, cleaned_file_path, company_name)
-
-# Force UTF-8 encoding for terminal output
-sys.stdout.reconfigure(encoding='utf-8')
-
-def capture_screenshot_with_playwright(html_file_path, screenshot_path):
-    """
-    Capture a full-page screenshot of the HTML file directly using Playwright.
-    """
-    try:
-        # Launch Playwright in headless mode
-        with sync_playwright() as p:
-            browser = p.chromium.launch(headless=True)
-            page = browser.new_page()
-
-            # Open the HTML file in the browser
-            page.goto(f"file:///{os.path.abspath(html_file_path)}")
-            
-            # Capture the full-page screenshot
-            page.screenshot(path=screenshot_path, full_page=True)
-            print(f"Screenshot saved: {screenshot_path}")
-
-            browser.close()
-
-    except Exception as e:
-        print(f"Error capturing screenshot: {e}")
-
-import os
-from PIL import Image
-
-def convert_png_to_pdf(png_path, company_name):
-    """
-    Convert a PNG image into a PDF strictly named as 'company_name brand marketing.pdf'
-    in a cloud-friendly directory.
-    """
-    try:
-        # Use a safer output directory (e.g., /home/user or /tmp)
-        output_folder = "/tmp/template_PDF"
-        os.makedirs(output_folder, exist_ok=True)
-
-        # Create the PDF file name as 'company_name brand marketing.pdf'
-        pdf_path = os.path.join(output_folder, f"{company_name} brand marketing.pdf")  
-
-        # Convert the PNG to PDF
-        img = Image.open(png_path)
-        img.convert('RGB').save(pdf_path, "PDF")
-
-        print(f"PDF saved: {pdf_path}")
-        return pdf_path  # Return the file path for verification
-    except Exception as e:
-        print(f"Error converting PNG to PDF: {e}")
-        return None
-
-
-if __name__ == "__main__":
-    # Paths for demonstration
-    html_file_path = "src/templates/brand_marketing.html"
-    
-    # Screenshot saved in the folder: data/reports/template_ss
-    screenshot_folder = "data/reports/template_ss"
-    os.makedirs(screenshot_folder, exist_ok=True)
-    screenshot_path = os.path.join(screenshot_folder, "brand_marketing_screenshot.png")
-
-    # Ensure Playwright browsers are installed
-    os.system("playwright install")
-
-    # Capture screenshot
-    capture_screenshot_with_playwright(html_file_path, screenshot_path)
+  save_html_file(HTML_FILE, html_content)
 
-    # Convert screenshot to PDF with the company name strictly as the filename
-    convert_png_to_pdf(screenshot_path, company_name)
+    # ✅ Capture Screenshot and Convert to PDF
+    capture_screenshot()
+    convert_to_pdf()