Update src/streamlit_app.py

src/streamlit_app.py

@@ -1,40 +1,210 @@
-import altair as alt
-import numpy as np
-import pandas as pd
 import streamlit as st
+import subprocess
+import os
+import tempfile
+import shutil
+from pathlib import Path
 
-"""
-# Welcome to Streamlit!
+st.set_page_config(
+    page_title="HTML to PDF Converter",
+    page_icon="📄",
+    layout="wide"
+)
 
-Edit `/streamlit_app.py` to customize this app to your heart's desire :heart:.
-If you have any questions, checkout our [documentation](https://docs.streamlit.io) and [community
-forums](https://discuss.streamlit.io).
+def convert_html_to_pdf(html_content, aspect_ratio):
+    """
+    Convert HTML content to PDF using Puppeteer
+    
+    Args:
+        html_content: String containing HTML content
+        aspect_ratio: One of "16:9", "1:1", or "9:16"
+    
+    Returns:
+        Path to generated PDF file or error message
+    """
+    try:
+        # Create temporary directory for processing
+        temp_dir = tempfile.mkdtemp()
+        
+        # Save HTML content to temporary file
+        html_file = os.path.join(temp_dir, "input.html")
+        with open(html_file, 'w', encoding='utf-8') as f:
+            f.write(html_content)
+        
+        # Run Node.js script to convert HTML to PDF
+        result = subprocess.run(
+            ['node', 'puppeteer_pdf.js', html_file, aspect_ratio],
+            capture_output=True,
+            text=True,
+            timeout=60
+        )
+        
+        if result.returncode != 0:
+            return None, f"PDF conversion failed: {result.stderr}"
+        
+        # Get the generated PDF path
+        pdf_file = html_file.replace('.html', '.pdf')
+        
+        if not os.path.exists(pdf_file):
+            return None, "PDF file was not generated"
+        
+        # Copy PDF to a permanent location
+        output_pdf = os.path.join(tempfile.gettempdir(), f"output_{os.getpid()}.pdf")
+        shutil.copy(pdf_file, output_pdf)
+        
+        # Clean up temporary directory
+        shutil.rmtree(temp_dir, ignore_errors=True)
+        
+        return output_pdf, None
+        
+    except subprocess.TimeoutExpired:
+        return None, "Error: PDF conversion timed out (60 seconds)"
+    except Exception as e:
+        return None, f"Error: {str(e)}"
 
-In the meantime, below is an example of what you can do with just a few lines of code:
-"""
+# Page header
+st.title("📄 HTML to PDF Converter")
+st.markdown("""
+Convert HTML files or HTML code to PDF using Puppeteer.  
+Supports multiple aspect ratios: **16:9** (landscape), **1:1** (square), and **9:16** (portrait).
+""")
 
-num_points = st.slider("Number of points in spiral", 1, 10000, 1100)
-num_turns = st.slider("Number of turns in spiral", 1, 300, 31)
+# Create tabs
+tab1, tab2 = st.tabs(["📤 Upload HTML File", "📝 Paste HTML Code"])
 
-indices = np.linspace(0, 1, num_points)
-theta = 2 * np.pi * num_turns * indices
-radius = indices
+# Tab 1: Upload HTML File
+with tab1:
+    col1, col2 = st.columns([1, 1])
+    
+    with col1:
+        uploaded_file = st.file_uploader(
+            "Choose an HTML file",
+            type=['html', 'htm'],
+            key="file_uploader"
+        )
+        
+        aspect_ratio_file = st.radio(
+            "Aspect Ratio",
+            options=["16:9", "1:1", "9:16"],
+            index=0,
+            key="aspect_file",
+            help="Select the page orientation and dimensions"
+        )
+        
+        convert_file_btn = st.button("🔄 Convert to PDF", key="convert_file", type="primary", use_container_width=True)
+    
+    with col2:
+        if convert_file_btn:
+            if uploaded_file is not None:
+                with st.spinner("Converting HTML to PDF..."):
+                    # Read uploaded file
+                    html_content = uploaded_file.read().decode('utf-8')
+                    
+                    # Convert to PDF
+                    pdf_path, error = convert_html_to_pdf(html_content, aspect_ratio_file)
+                    
+                    if error:
+                        st.error(error)
+                    else:
+                        st.success("✅ PDF generated successfully!")
+                        
+                        # Read PDF file
+                        with open(pdf_path, 'rb') as f:
+                            pdf_data = f.read()
+                        
+                        # Download button
+                        st.download_button(
+                            label="⬇️ Download PDF",
+                            data=pdf_data,
+                            file_name="converted.pdf",
+                            mime="application/pdf",
+                            use_container_width=True
+                        )
+            else:
+                st.warning("Please upload an HTML file first.")
 
-x = radius * np.cos(theta)
-y = radius * np.sin(theta)
+# Tab 2: Paste HTML Code
+with tab2:
+    col1, col2 = st.columns([1, 1])
+    
+    with col1:
+        html_code = st.text_area(
+            "HTML Content",
+            value="""<!DOCTYPE html>
+<html>
+<head>
+    <title>Sample Document</title>
+    <style>
+        body {
+            font-family: Arial, sans-serif;
+            margin: 40px;
+            background: linear-gradient(135deg, #667eea 0%, #764ba2 100%);
+            color: white;
+        }
+        h1 {
+            font-size: 48px;
+            margin-bottom: 20px;
+        }
+        p {
+            font-size: 18px;
+            line-height: 1.6;
+        }
+    </style>
+</head>
+<body>
+    <h1>Hello, PDF World! 🌍</h1>
+    <p>This is a sample HTML document converted to PDF.</p>
+    <p>You can customize this HTML with your own content!</p>
+</body>
+</html>""",
+            height=400,
+            key="html_code"
+        )
+        
+        aspect_ratio_text = st.radio(
+            "Aspect Ratio",
+            options=["16:9", "1:1", "9:16"],
+            index=0,
+            key="aspect_text",
+            help="Select the page orientation and dimensions"
+        )
+        
+        convert_text_btn = st.button("🔄 Convert to PDF", key="convert_text", type="primary", use_container_width=True)
+    
+    with col2:
+        if convert_text_btn:
+            if html_code and html_code.strip():
+                with st.spinner("Converting HTML to PDF..."):
+                    # Convert to PDF
+                    pdf_path, error = convert_html_to_pdf(html_code, aspect_ratio_text)
+                    
+                    if error:
+                        st.error(error)
+                    else:
+                        st.success("✅ PDF generated successfully!")
+                        
+                        # Read PDF file
+                        with open(pdf_path, 'rb') as f:
+                            pdf_data = f.read()
+                        
+                        # Download button
+                        st.download_button(
+                            label="⬇️ Download PDF",
+                            data=pdf_data,
+                            file_name="converted.pdf",
+                            mime="application/pdf",
+                            use_container_width=True
+                        )
+            else:
+                st.warning("Please enter HTML content first.")
 
-df = pd.DataFrame({
-    "x": x,
-    "y": y,
-    "idx": indices,
-    "rand": np.random.randn(num_points),
-})
-
-st.altair_chart(alt.Chart(df, height=700, width=700)
-    .mark_point(filled=True)
-    .encode(
-        x=alt.X("x", axis=None),
-        y=alt.Y("y", axis=None),
-        color=alt.Color("idx", legend=None, scale=alt.Scale()),
-        size=alt.Size("rand", legend=None, scale=alt.Scale(range=[1, 150])),
-    ))
+# Footer with tips
+st.markdown("---")
+st.markdown("""
+### 💡 Tips:
+- **16:9** - Best for presentations and landscape documents (297mm × 210mm)
+- **1:1** - Square format (210mm × 210mm)
+- **9:16** - Portrait format, standard A4 (210mm × 297mm)
+- Include inline CSS for styling
+- External resources should use absolute URLs
+""")