Update app.py

app.py

@@ -1,4 +1,4 @@
-# app.py - Fast & Reliable PDF Generator (Fixed ByteArray Error)
+# app.py - Enhanced PDF & Video Generator (HD Video with Names Below Images)
 
 import gradio as gr
 from fpdf import FPDF
@@ -6,7 +6,8 @@ import os
 import tempfile
 import re
 from datetime import datetime
-from pathlib import Path
+from moviepy.editor import ImageClip, concatenate_videoclips, TextClip, CompositeVideoClip
+from PIL import Image  # For image processing if needed
 
 class SimplePDF(FPDF):
     def footer(self):
@@ -15,53 +16,48 @@ class SimplePDF(FPDF):
         self.cell(0, 10, f'Page {self.page_no()}', 0, 0, 'C')
 
 def clean_text(text, max_length=100):
-    """Clean text for PDF compatibility"""
+    """Clean text for PDF and video compatibility"""
     if not text:
         return ""
-    
-    # Replace problematic Unicode characters
     replacements = {
         '–': '-', '—': '-', '“': '"', '”': '"', '‘': "'", '’': "'",
         '•': '-', '°': '°', '™': 'TM', '®': 'R', '©': 'C',
         '€': 'EUR', '£': 'GBP', '¥': 'JPY'
     }
-    
     for old, new in replacements.items():
         text = text.replace(old, new)
-    
-    # Remove other problematic characters
     text = re.sub(r'[^\w\s\-.,!?\(\)\[\]:;0-9°]', '', text)
     text = ' '.join(text.split())[:max_length]
     return text.strip()
 
-def generate_pdf(title, plan, names, image_files):
-    # Fast validation
+def generate_outputs(title, plan, names, image_files, duration_per_slide):
+    # Validation
     if not title or not title.strip():
-        return None, "❌ Title required!"
-    
-    if not names or not names.strip():
-        return None, "❌ Names required!"
+        return None, None, "❌ Title required!"
     
     names_list = [n.strip() for n in names.split(',') if n.strip()]
     
     if not names_list:
-        return None, "❌ At least one name needed!"
+        return None, None, "❌ At least one name needed!"
     
     if not image_files:
-        return None, "❌ Images required!"
+        return None, None, "❌ Images required!"
     
     if len(names_list) != len(image_files):
-        return None, f"❌ {len(names_list)} names vs {len(image_files)} images"
+        return None, None, f"❌ {len(names_list)} names vs {len(image_files)} images"
+    
+    # Limit to 30 items max
+    if len(names_list) > 30:
+        return None, None, "❌ Max 30 items allowed!"
     
     try:
+        # PDF Generation
         pdf = SimplePDF()
         pdf.set_margins(15, 15, 15)
         
-        # Clean inputs
         clean_title = clean_text(title)
         clean_plan = clean_text(plan, 400)
         
-        # Title Page
         pdf.add_page()
         pdf.set_font('Arial', 'B', 20)
         pdf.cell(0, 15, clean_title, ln=1, align='C')
@@ -76,10 +72,8 @@ def generate_pdf(title, plan, names, image_files):
         pdf.cell(0, 8, f"Generated: {datetime.now().strftime('%Y-%m-%d %H:%M')}", ln=1, align='C')
         pdf.cell(0, 8, f"Participants: {len(names_list)}", ln=1, align='C')
         
-        # Participant pages
         for i, (name, img_path) in enumerate(zip(names_list, image_files)):
             pdf.add_page()
-            
             clean_name = clean_text(name, 40)
             pdf.set_font('Arial', 'B', 16)
             pdf.cell(0, 12, clean_name.upper(), ln=1, align='C')
@@ -87,7 +81,6 @@ def generate_pdf(title, plan, names, image_files):
             pdf.set_font('Arial', '', 11)
             pdf.cell(0, 8, f"ID: {i+1}/{len(names_list)}", ln=1, align='C')
             
-            # Add image safely
             image_added = False
             if os.path.exists(img_path):
                 try:
@@ -99,74 +92,151 @@ def generate_pdf(title, plan, names, image_files):
                     pass
             
             if not image_added:
-                # Placeholder if image fails
                 pdf.set_font('Arial', '', 12)
                 pdf.cell(0, 10, "[IMAGE PLACEHOLDER]", ln=1, align='C')
-                pdf.set_draw_color(200, 200, 200)
-                pdf.rect(50, 50, 100, 100, 'D')
+        
+        pdf_path = f"pdf_{int(datetime.now().timestamp())}.pdf"
+        pdf.output(pdf_path)
+        
+        # Video Generation - Full HD 1920x1080
+        clips = []
+        hd_width, hd_height = 1920, 1080
+        text_height = 150  # Space for name below image
+        
+        for i, (name, img_path) in enumerate(zip(names_list, image_files)):
+            clean_name = clean_text(name, 40)
             
-            # Add name below image
-            pdf.ln(120)
-            pdf.set_font('Arial', '', 10)
-            pdf.cell(0, 8, f"Name: {clean_name}", ln=1, align='C')
+            # Load image
+            if os.path.exists(img_path):
+                try:
+                    # Create image clip
+                    img_clip = ImageClip(img_path).resize(height=hd_height - text_height - 50)  # Resize to fit above text
+                    img_clip = img_clip.set_position(("center", 25))  # Position at top with margin
+                    
+                    # Text clip for name below
+                    text_clip = TextClip(
+                        f"{clean_name.upper()}\nID: {i+1}",
+                        fontsize=80, color='white', font='Arial-Bold',
+                        size=(hd_width, text_height), bg_color='black', method='caption'
+                    ).set_position(("center", hd_height - text_height - 25))
+                    
+                    # Composite
+                    composite = CompositeVideoClip([img_clip, text_clip], size=(hd_width, hd_height))
+                    composite = composite.set_duration(duration_per_slide)
+                    clips.append(composite)
+                except Exception as e:
+                    # Fallback placeholder clip
+                    text_clip = TextClip(
+                        f"{clean_name.upper()}\nID: {i+1}\n[Image Not Available]",
+                        fontsize=60, color='white', bg_color='black'
+                    ).set_duration(duration_per_slide)
+                    clips.append(text_clip)
+            else:
+                # Placeholder if no image
+                text_clip = TextClip(
+                    f"{clean_name.upper()}\nID: {i+1}\n[No Image]",
+                    fontsize=60, color='white', bg_color='black'
+                ).set_duration(duration_per_slide)
+                clips.append(text_clip)
+        
+        # Concatenate all clips into video
+        video = concatenate_videoclips(clips, method="compose")
+        
+        # Add title intro slide if title
+        if clean_title:
+            title_clip = TextClip(
+                clean_title.upper(),
+                fontsize=100, color='white', bg_color='blue',
+                size=(hd_width, hd_height)
+            ).set_duration(3)  # 3 sec intro
+            video = concatenate_videoclips([title_clip, video])
         
-        # FIXED: Direct file output (no BytesIO complications)
-        output_path = f"output_{int(datetime.now().timestamp())}.pdf"
-        pdf.output(output_path)
+        video_path = f"video_{int(datetime.now().timestamp())}.mp4"
+        video.write_videofile(video_path, fps=24, codec='libx264', audio=False)  # HD silent video
         
-        success_msg = f"✅ SUCCESS!\n📄 {clean_title}\n👥 {len(names_list)} participants"
-        return output_path, success_msg
+        success_msg = f"✅ Generated!\n📄 PDF & 🎥 HD Video (1920x1080)\n👥 {len(names_list)} items (IDs 1 to {len(names_list)})"
+        return pdf_path, video_path, success_msg
         
     except Exception as e:
-        return None, f"❌ Error: {str(e)}"
+        return None, None, f"❌ Error: {str(e)}"
 
-# Fast & Clean UI
-with gr.Blocks(title="Fast PDF Generator") as demo:
-    gr.Markdown("# 🚀 Fast PDF Generator")
-    gr.Markdown("**Title + Plan + Names + Images → Professional PDF**")
+# Enhanced UI/UX with Better Layout & Instructions
+with gr.Blocks(
+    theme=gr.themes.Soft(primary_hue="blue", secondary_hue="gray", neutral_hue="slate"),
+    css="""
+    .gradio-container {max-width: 1200px; margin: auto; padding: 20px;}
+    .input-label {font-weight: bold; font-size: 1.2em;}
+    .output-section {margin-top: 20px; padding: 15px; background: #f0f4f8; border-radius: 10px;}
+    .status {font-size: 1.1em; color: #333;}
+    """
+) as demo:
+    gr.Markdown(
+        """
+        # 🚀 HD PDF & Video Generator
+        **Create Professional PDF & Full HD Video Slideshow**
+        
+        ### Instructions:
+        - **Title**: Document/Video title (required)
+        - **Plan**: Optional description
+        - **Names**: Comma-separated (e.g., "John Doe, Jane Smith") - Max 30
+        - **Images**: Upload matching images (uploaded below names input for UX)
+        - **Slide Duration**: Seconds per slide in video
+        - Video: Full HD 1920x1080, name below image, IDs 1 to N
+        - Click Generate for PDF & Video downloads
+        """
+    )
     
-    with gr.Row():
+    with gr.Row(equal_height=True):
         with gr.Column(scale=2):
             title_input = gr.Textbox(
-                label="📛 Title *",
-                placeholder="Document Title",
-                lines=1
+                label="📛 Title (Required)",
+                placeholder="Enter title here...",
+                lines=1,
+                elem_classes="input-label"
             )
             plan_input = gr.Textbox(
-                label="📋 Plan (optional)",
-                placeholder="Description or plan details",
-                lines=3
+                label="📋 Plan/Description (Optional)",
+                placeholder="Enter details here...",
+                lines=4
             )
             names_input = gr.Textbox(
-                label="👥 Names *",
-                placeholder="Name1, Name2, Name3",
-                lines=2
+                label="👥 Names (Comma-Separated, Required)",
+                placeholder="Name1, Name2, Name3 (up to 30)",
+                lines=3
             )
-        
-        with gr.Column(scale=1):
             images_input = gr.File(
-                label="🖼️ Images *",
+                label="🖼️ Upload Images (Match Names Count)",
                 file_count="multiple",
-                file_types=[".jpg", ".jpeg", ".png"]
+                file_types=["image"]
+            )
+        
+        with gr.Column(scale=1):
+            duration_input = gr.Slider(
+                minimum=1,
+                maximum=30,
+                step=1,
+                value=5,
+                label="⏱️ Duration per Slide (Seconds)"
             )
+            generate_btn = gr.Button("🚀 Generate PDF & Video", variant="primary", size="lg")
+            clear_btn = gr.Button("🗑️ Clear All", variant="secondary")
     
-    with gr.Row():
-        generate_btn = gr.Button("🚀 GENERATE PDF", variant="primary")
-        clear_btn = gr.Button("🗑️ Clear", variant="secondary")
+    with gr.Row(elem_classes="output-section"):
+        pdf_output = gr.File(label="📄 Download PDF")
+        video_output = gr.File(label="🎥 Download HD Video (MP4)")
     
-    status_output = gr.Markdown("Ready...")
-    pdf_output = gr.File(label="📥 Download PDF")
+    status_output = gr.Markdown("Ready to generate...", elem_classes="status")
     
-    # Events
+    # Event handlers
     generate_btn.click(
-        fn=generate_pdf,
-        inputs=[title_input, plan_input, names_input, images_input],
-        outputs=[pdf_output, status_output]
+        fn=generate_outputs,
+        inputs=[title_input, plan_input, names_input, images_input, duration_input],
+        outputs=[pdf_output, video_output, status_output]
     )
     
     clear_btn.click(
-        fn=lambda: ("", "", "", None, "Cleared!"),
-        outputs=[title_input, plan_input, names_input, images_input, status_output]
+        fn=lambda: ("", "", "", None, 5, None, None, "Cleared!"),
+        outputs=[title_input, plan_input, names_input, images_input, duration_input, pdf_output, video_output, status_output]
     )
 
 if __name__ == "__main__":