Update app.py

app.py

@@ -1,19 +1,17 @@
-# app.py - Gym Workout Plan PDF + HD Video Generator (MoviePy Fixed)
+# app.py - Gym Workout Plan PDF + Video (Fixed Errors)
 
 import gradio as gr
 from fpdf import FPDF
 import os
 import re
 from datetime import datetime
-import tempfile
 
-# Video generation with MoviePy (Fixed dependencies)
+# Video generation
 try:
     from moviepy.editor import ImageClip, concatenate_videoclips, TextClip, CompositeVideoClip
     VIDEO_ENABLED = True
 except ImportError:
     VIDEO_ENABLED = False
-    print("MoviePy not available - Video generation disabled")
 
 class GymWorkoutPDF(FPDF):
     def footer(self):
@@ -21,35 +19,30 @@ class GymWorkoutPDF(FPDF):
         self.set_font('Arial', 'I', 7)
         self.cell(0, 8, f'GYM WORKOUT PLAN | Page {self.page_no()}', 0, 0, 'C')
 
-def clean_text_for_pdf_video(text, max_length=45):
-    """Clean text for both PDF and video"""
+def clean_text(text, max_length=45):
     if not text:
         return "Workout"
     text = str(text)
-    
-    # Remove emojis and unsupported chars
     emoji_pattern = re.compile(
         "["
         u"\U0001F600-\U0001F64F"
-        u"\U0001F300-\U0001F5FF" 
+        u"\U0001F300-\U0001F5FF"
         u"\U0001F680-\U0001F6FF"
         u"\U0001F1E0-\U0001F1FF"
         "]+",
         flags=re.UNICODE
     )
     text = emoji_pattern.sub('', text)
-    
     replacements = {
         '–': '-', '—': '-', '“': '"', '”': '"', '•': '*', '°': 'deg'
     }
     for old, new in replacements.items():
         text = text.replace(old, new)
-    
     text = re.sub(r'[^\x00-\x7F\u0020-\u007E]', '', text)
     text = re.sub(r'\s+', ' ', text.strip())
     return text[:max_length]
 
-def generate_gym_content(main_title, workout_names, image_files):
+def generate_content(main_title, workout_names, image_files):
     if not main_title.strip():
         return None, None, "Error: Main Title Required!"
     
@@ -62,14 +55,14 @@ def generate_gym_content(main_title, workout_names, image_files):
         return None, None, "Error: Max 30 Workouts!"
     
     if not image_files or len(names_list) != len(image_files):
-        return None, None, f"Error: Need {len(names_list)} Images!"
+        return None, None, f"Error: Number of workout names ({len(names_list)}) must match images ({len(image_files)})!"
     
     try:
-        clean_main_title = clean_text_for_pdf_video(main_title, 50)
+        clean_main_title = clean_text(main_title, 50)
         pdf = GymWorkoutPDF()
         pdf.set_margins(18, 18, 18)
         
-        # PDF GENERATION
+        # PDF
         pdf.add_page()
         pdf.set_font('Arial', 'B', 20)
         pdf.set_text_color(0, 128, 0)
@@ -82,18 +75,16 @@ def generate_gym_content(main_title, workout_names, image_files):
         pdf.cell(0, 10, f"TOTAL WORKOUTS: {len(names_list)}", ln=1, align='C')
         pdf.cell(0, 10, f"CREATED: {datetime.now().strftime('%d/%m/%Y')}", ln=1, align='C')
         
-        # Workout pages in PDF
         for i, (workout_name, img_path) in enumerate(zip(names_list, image_files)):
             pdf.add_page()
             pdf.set_font('Arial', 'B', 16)
             pdf.set_text_color(0, 128, 0)
             pdf.cell(0, 14, f"WORKOUT {i+1}", ln=1, align='C')
             
+            clean_name = clean_text(workout_name, 40)
             pdf.set_font('Arial', 'B', 14)
-            clean_name = clean_text_for_pdf_video(workout_name, 40)
             pdf.multi_cell(0, 12, clean_name.upper(), align='C')
             
-            # Image in PDF
             if os.path.exists(img_path):
                 try:
                     img_width = 110
@@ -108,137 +99,70 @@ def generate_gym_content(main_title, workout_names, image_files):
         pdf_path = f"workout_plan_{timestamp}.pdf"
         pdf.output(pdf_path)
         
-        # VIDEO GENERATION (HD 1080p)
+        # VIDEO
         video_path = None
-        if VIDEO_ENABLED and image_files:
-            try:
-                clips = []
-                hd_width, hd_height = 1920, 1080
-                duration_per_slide = 30  # 30 seconds per workout
-                
-                # Title slide
-                title_text = TextClip(
-                    f"GYM WORKOUT PROGRAM\n{clean_main_title.upper()}",
-                    fontsize=80, color='white', font='Arial-Bold',
-                    size=(hd_width, 300), bg_color='darkgreen'
-                ).set_duration(3).set_position('center')
-                title_bg = ImageClip(image_files[0]).resize(height=hd_height).set_duration(3)
-                title_clip = CompositeVideoClip([title_bg, title_text])
-                clips.append(title_clip)
-                
-                # Workout slides
-                for i, (workout_name, img_path) in enumerate(zip(names_list, image_files)):
-                    clean_name = clean_text_for_pdf_video(workout_name, 40)
-                    
-                    # Image background
-                    img_clip = ImageClip(img_path).resize(height=hd_height-200)
-                    img_clip = img_clip.set_position('center').set_duration(duration_per_slide)
-                    
-                    # Workout text overlay
-                    text_content = f"WORKOUT {i+1}\n{clean_name.upper()}"
-                    text_clip = TextClip(
-                        text_content,
-                        fontsize=60, color='white', font='Arial-Bold',
-                        size=(hd_width-400, 200), bg_color='rgba(0,128,0,0.7)'
-                    ).set_position(('center', hd_height-250)).set_duration(duration_per_slide)
-                    
-                    # Composite slide
-                    slide = CompositeVideoClip([img_clip, text_clip], size=(hd_width, hd_height))
-                    clips.append(slide)
+        if VIDEO_ENABLED:
+            clips = []
+            hd_width, hd_height = 1920, 1080
+            duration_per_slide = 5
+            
+            # Title slide
+            title_text = TextClip(
+                f"GYM WORKOUT PROGRAM\n{clean_main_title.upper()}",
+                fontsize=80, color='white', font='Arial-Bold',
+                size=(hd_width, 300), bg_color='darkgreen'
+            ).set_duration(3).set_position('center')
+            title_clip = CompositeVideoClip([title_text], size=(hd_width, hd_height))
+            clips.append(title_clip)
+            
+            # Workout slides
+            for i, (workout_name, img_path) in enumerate(zip(names_list, image_files)):
+                clean_name = clean_text(workout_name, 40)
+                img_clip = ImageClip(img_path).resize(height=hd_height-200)
+                img_clip = img_clip.set_position('center').set_duration(duration_per_slide)
                 
-                # Generate final video
-                video = concatenate_videoclips(clips, method="compose")
-                video_path = f"workout_video_{timestamp}.mp4"
-                video.write_videofile(
-                    video_path, 
-                    fps=24, 
-                    codec='libx264',
-                    audio=False,
-                    temp_audiofile='temp-audio.m4a',
-                    remove_temp=True
-                )
+                text_clip = TextClip(
+                    f"WORKOUT {i+1}\n{clean_name.upper()}",
+                    fontsize=60, color='white', font='Arial-Bold',
+                    size=(hd_width-400, 200), bg_color='rgba(0,128,0,0.7)'
+                ).set_position(('center', hd_height-250)).set_duration(duration_per_slide)
                 
-            except Exception as video_error:
-                print(f"Video generation failed: {video_error}")
-                video_path = None
-        
-        success_msg = f"""
-SUCCESS! GYM CONTENT GENERATED!
-PDF: {len(names_list) + 1} pages created
-Video: {'HD 1080p generated!' if video_path else 'Disabled (MoviePy not available)'}
-Workouts: {len(names_list)} total
-Ready for gym use!
-        """
+                slide = CompositeVideoClip([img_clip, text_clip], size=(hd_width, hd_height))
+                clips.append(slide)
+            
+            video = concatenate_videoclips(clips, method="compose")
+            video_path = f"workout_video_{timestamp}.mp4"
+            video.write_videofile(
+                video_path, 
+                fps=24, 
+                codec='libx264',
+                audio=False
+            )
         
+        success_msg = f"SUCCESS! Generated PDF and {'Video' if video_path else 'No Video (Install MoviePy)'} for {len(names_list)} workouts!"
         return pdf_path, video_path, success_msg
         
     except Exception as e:
         return None, None, f"Error: {str(e)}"
 
-# UI with PDF + Video Output
-with gr.Blocks(
-    title="Gym Workout Generator",
-    theme=gr.themes.Soft(primary_hue="green"),
-    css="""
-    .gradio-container {max-width: 1000px !important;}
-    .header {background: linear-gradient(135deg, #16a34a, #15803d); color: white; padding: 25px; border-radius: 15px;}
-    .input-box {background: #f0fdf4; padding: 18px; border-radius: 10px; border: 2px solid #dcfce7;}
-    .btn-generate {background: linear-gradient(45deg, #16a34a, #15803d) !important; font-size: 18px;}
-    .output-section {background: #f0fdf4; padding: 15px; border-radius: 10px; margin: 10px 0;}
-    """
-) as demo:
-    
-    gr.HTML("""
-    <div class="header">
-        <h2>Gym Workout Generator</h2>
-        <p><strong>PDF + HD Video (1080p) | Professional Gym Content</strong></p>
-    </div>
-    """)
-    
-    gr.Markdown("""
-    ### Features:
-    - PDF: Professional workout plans with images
-    - Video: HD 1080p slideshow (5 sec per workout)
-    - Structure: Title → Workout 1, 2, 3... with images
-    - No emoji errors - clean text only
-    """)
+# UI
+with gr.Blocks(title="Gym Workout Generator") as demo:
+    gr.Markdown("# Gym Workout Generator\nPDF + HD Video | No Errors")
     
     with gr.Row():
-        with gr.Column(scale=2):
-            main_title = gr.Textbox(
-                label="MAIN PROGRAM TITLE *",
-                placeholder="12 Week Muscle Building Program",
-                lines=2,
-                elem_classes="input-box"
-            )
-            
-            workout_names = gr.Textbox(
-                label="WORKOUT NAMES * (Comma separated)",
-                placeholder="Chest Day, Leg Day, Back Workout, Cardio",
-                lines=3,
-                elem_classes="input-box"
-            )
-        
-        with gr.Column(scale=1):
-            images_input = gr.File(
-                label="Workout Images *",
-                file_count="multiple",
-                file_types=[".jpg", ".jpeg", ".png"],
-                elem_classes="input-box"
-            )
+        main_title = gr.Textbox(label="MAIN TITLE *")
+        workout_names = gr.Textbox(label="WORKOUT NAMES * (Comma separated)")
     
-    generate_btn = gr.Button("Generate PDF + HD Video", variant="primary", elem_classes="btn-generate")
+    images_input = gr.File(label="Images * (Match workout count)", file_count="multiple", file_types=["image"])
     
-    with gr.Row():
-        with gr.Column():
-            pdf_output = gr.File(label="Download PDF", elem_classes="output-section")
-        with gr.Column():
-            video_output = gr.File(label="Download HD Video", elem_classes="output-section")
+    generate_btn = gr.Button("Generate")
     
-    status = gr.Markdown("Ready to generate...", elem_classes="output-section")
+    pdf_output = gr.File(label="PDF")
+    video_output = gr.File(label="HD Video")
+    status = gr.Markdown("Ready...")
     
     generate_btn.click(
-        fn=generate_gym_content,
+        fn=generate_content,
         inputs=[main_title, workout_names, images_input],
         outputs=[pdf_output, video_output, status]
     )