Update app.py

app.py

@@ -6,26 +6,13 @@ import tempfile
 import shutil
 from typing import Dict, List
 import base64
+import subprocess
+import sys
 
 import pandas as pd
 import streamlit as st
 from num2words import num2words
 
-# Install Playwright browsers on first run (for Hugging Face Spaces)
-import subprocess
-import sys
-
-if not os.path.exists('/tmp/playwright_installed'):
-    try:
-        print("🔧 Installing Playwright browsers...")
-        subprocess.run([sys.executable, "-m", "playwright", "install", "chromium"], check=True)
-        with open('/tmp/playwright_installed', 'w') as f:
-            f.write('installed')
-        print("✅ Playwright browsers installed!")
-    except Exception as e:
-        print(f"⚠️ Failed to install Playwright browsers: {e}")
-        print("App will continue but PDF generation may not work")
-
 # Try to import Playwright for PDF conversion
 PLAYWRIGHT_AVAILABLE = False
 
@@ -35,6 +22,17 @@ try:
 except ImportError:
     pass
 
+# Install Playwright browsers on first run
+_playwright_flag = os.path.join(tempfile.gettempdir(), 'playwright_installed')
+if PLAYWRIGHT_AVAILABLE and not os.path.exists(_playwright_flag):
+    try:
+        subprocess.run([sys.executable, "-m", "playwright", "install", "chromium"], check=True)
+        with open(_playwright_flag, 'w') as f:
+            f.write('installed')
+        print("✅ Playwright browsers installed!")
+    except Exception as e:
+        print(f"⚠️ Failed to install Playwright browsers: {e}")
+
 # Try to import PyPDF for PDF page extraction
 PYPDF_AVAILABLE = False
 
@@ -59,7 +57,7 @@ def format_amount(x):
         num = float(x)
         if num == 0:
             return "0.00"
-            return f"{num:,.2f}"
+        return f"{num:,.2f}"
     except Exception:
         return "0.00"
 
@@ -310,7 +308,7 @@ if __name__ == "__main__":
                 
                 if result.returncode == 0 and os.path.exists(pdf_path):
                     return True
-        else:
+                else:
                     print(f"Subprocess approach failed: {result.stderr}")
                     
             except Exception as e2:
@@ -345,7 +343,7 @@ def extract_first_page_pdf(input_pdf_path: str, output_pdf_path: str) -> bool:
         reader = PdfReader(input_pdf_path)
         
         # Create a new PDF writer
-    writer = PdfWriter()
+        writer = PdfWriter()
         
         # Add only the first page (if it exists)
         if len(reader.pages) > 0:
@@ -530,7 +528,7 @@ with col2:
 
 # Processing section
 if template_file:
-st.markdown("---")
+    st.markdown("---")
     st.subheader("🚀 Generate Invoices")
     
     col3, col4 = st.columns(2)
@@ -544,8 +542,8 @@ st.markdown("---")
                     help="Select Inephos template to enable this button"):
             if not data_file:
                 st.error("Please upload customer data file.")
-        else:
-            with st.spinner("Generating Inephos invoices..."):
+            else:
+                with st.spinner("Generating Inephos invoices..."):
                     try:
                         zip_buffer, count = process_invoices_html(data_file, "inephos.html", "inephos")
                         st.success(f"✅ Generated {count} Inephos invoices successfully!")
@@ -571,8 +569,8 @@ st.markdown("---")
                     help="Select Srestham template to enable this button"):
             if not data_file:
                 st.error("Please upload customer data file.")
-        else:
-            with st.spinner("Generating Srestham invoices..."):
+            else:
+                with st.spinner("Generating Srestham invoices..."):
                     try:
                         zip_buffer, count = process_invoices_html(data_file, "srestham.html", "srestham")
                         st.success(f"✅ Generated {count} Srestham invoices successfully!")