Update app.py

app.py

@@ -1,4 +1,4 @@
-# app.py - Gym Workout Plan PDF (Fixed Text Display + Workout Numbers)
+# app.py - Gym Workout Plan PDF (Title + Workout Names + Images Layout)
 
 import gradio as gr
 from fpdf import FPDF
@@ -10,84 +10,82 @@ class GymWorkoutPDF(FPDF):
     def footer(self):
         self.set_y(-10)
         self.set_font('Arial', 'I', 7)
-        self.cell(0, 8, f'GYM WORKOUT PLAN | Page {self.page_no()} | {datetime.now().strftime("%d/%m/%Y")}', 0, 0, 'C')
+        self.cell(0, 8, f'GYM WORKOUT PLAN | Page {self.page_no()}', 0, 0, 'C')
 
-def clean_text_for_pdf(text, max_length=40):
-    """Fixed text cleaning - preserves more characters"""
+def clean_text_for_pdf(text, max_length=45):
+    """Clean text while preserving workout names"""
     if not text:
-        return "Workout Plan"
-    
-    # Convert to string first
+        return "Workout"
     text = str(text)
-    
-    # Preserve gym-related characters
     replacements = {
         '–': '-', '—': '-', '“': '"', '”': '"', '‘': "'", '’': "'",
-        '•': '•', '°': '°', '™': 'TM', '®': 'R', '©': 'C',
-        '\n': '\n', '\r': '', '\t': '    '  # Preserve line breaks
+        '•': '•', '°': '°', '×': 'x', '✓': 'v', '\n': '\n'
     }
-    
     for old, new in replacements.items():
         text = text.replace(old, new)
-    
-    # Keep safe characters including gym symbols
-    text = re.sub(r'[\x00-\x08\x0B\x0C\x0E-\x1F\x7F]', '', text)  # Remove control chars only
-    text = re.sub(r'\s+', ' ', text)  # Normalize spaces
-    return text.strip()[:max_length]
+    # Keep workout-related characters
+    text = re.sub(r'[\x00-\x08\x0B\x0C\x0E-\x1F\x7F]', '', text)
+    text = re.sub(r'\s+', ' ', text.strip())
+    return text[:max_length]
 
-def generate_gym_workout_pdf(title, names, image_files):
+def generate_workout_pdf(main_title, workout_names, image_files):
     # Validation
-    if not title.strip():
-        return None, "❌ Workout Plan Title Required!"
+    if not main_title.strip():
+        return None, "❌ Main Title Required!"
     
-    names_list = [n.strip() for n in names.split(',') if n.strip()]
+    names_list = [n.strip() for n in workout_names.split(',') if n.strip()]
     
     if not names_list:
-        return None, "❌ Add Gym Member Names!"
+        return None, "❌ Add Workout Names!"
     
     if len(names_list) > 30:
-        return None, "❌ Max 30 Members!"
+        return None, "❌ Max 30 Workouts!"
     
     if not image_files or len(names_list) != len(image_files):
-        return None, f"❌ Need {len(names_list)} Member Photos!"
+        return None, f"❌ Need {len(names_list)} Workout Images!"
     
     try:
         pdf = GymWorkoutPDF()
         pdf.set_margins(18, 18, 18)
         pdf.set_auto_page_break(auto=True, margin=18)
         
-        # COVER PAGE with PROPER TEXT SIZES
+        clean_main_title = clean_text_for_pdf(main_title, 50)
+        
+        # FIRST: MAIN TITLE PAGE
         pdf.add_page()
-        pdf.set_font('Arial', 'B', 18)  # Title size
+        pdf.set_font('Arial', 'B', 20)
         pdf.set_text_color(0, 128, 0)
-        pdf.cell(0, 15, "🏋️ GYM WORKOUT PROGRAM", ln=1, align='C')
+        pdf.cell(0, 18, "🏋️ GYM WORKOUT PROGRAM", ln=1, align='C')
         
-        pdf.set_font('Arial', 'B', 14)  # Subtitle size
-        clean_title = clean_text_for_pdf(title)
-        pdf.cell(0, 12, clean_title.upper(), ln=1, align='C')
+        pdf.set_font('Arial', 'B', 16)
+        pdf.cell(0, 15, clean_main_title.upper(), ln=1, align='C')
         
-        pdf.ln(10)
-        pdf.set_font('Arial', 'B', 11)
-        pdf.cell(0, 8, f"{'='*45}", ln=1, align='C')
-        pdf.set_font('Arial', '', 10)
-        pdf.cell(0, 8, f"📋 MEMBERS: {len(names_list)}", ln=1, align='C')
-        pdf.cell(0, 8, f"📅 DATE: {datetime.now().strftime('%d/%m/%Y %H:%M')}", ln=1, align='C')
-        pdf.set_font('Arial', 'B', 11)
-        pdf.cell(0, 8, f"{'='*45}", ln=1, align='C')
+        pdf.ln(15)
+        pdf.set_font('Arial', 'B', 12)
+        pdf.cell(0, 10, f"{'='*50}", ln=1, align='C')
+        pdf.set_font('Arial', '', 11)
+        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 %H:%M')}", ln=1, align='C')
+        pdf.set_font('Arial', 'B', 12)
+        pdf.cell(0, 10, f"{'='*50}", ln=1, align='C')
         
-        # MEMBER WORKOUT PAGES
-        for i, (name, img_path) in enumerate(zip(names_list, image_files)):
+        # NEXT: INDIVIDUAL WORKOUT PAGES
+        for i, (workout_name, img_path) in enumerate(zip(names_list, image_files)):
             pdf.add_page()
             
-            # WORKOUT NUMBER HEADER (NOT MEMBER #)
-            pdf.set_font('Arial', 'B', 14)
+            # WORKOUT NUMBER + NAME HEADER
+            pdf.set_font('Arial', 'B', 16)
             pdf.set_text_color(0, 128, 0)
-            pdf.cell(0, 12, f"WORKOUT {i+1}", ln=1, align='C')
+            pdf.cell(0, 14, f"WORKOUT {i+1}", ln=1, align='C')
             
-            # PHOTO SECTION (Medium size)
-            pdf.ln(8)
-            img_width = 100
-            img_height = 120
+            pdf.set_font('Arial', 'B', 14)
+            clean_workout_name = clean_text_for_pdf(workout_name, 40)
+            pdf.multi_cell(0, 12, clean_workout_name.upper(), align='C')
+            
+            # IMAGE FOR WORKOUT (Centered)
+            pdf.ln(10)
+            img_width = 110
+            img_height = 130
             page_width = pdf.w - 36
             x_pos = (page_width - img_width) / 2 + 18
             
@@ -103,193 +101,182 @@ def generate_gym_workout_pdf(title, names, image_files):
                     pass
             
             if not image_success:
+                # Workout placeholder image
                 pdf.set_fill_color(240, 248, 240)
                 pdf.set_draw_color(0, 128, 0)
                 pdf.rect(x_pos, pdf.get_y(), img_width, img_height, 'FD')
-                pdf.set_font('Arial', 'B', 10)
+                pdf.set_font('Arial', 'B', 11)
                 pdf.set_text_color(0, 128, 0)
-                pdf.set_xy(x_pos + 25, pdf.get_y() + 45)
+                pdf.set_xy(x_pos + 25, pdf.get_y() + 50)
                 pdf.cell(img_width - 50, 10, "WORKOUT", ln=1, align='C')
-                pdf.set_xy(x_pos + 25, pdf.get_y() + 60)
-                pdf.cell(img_width - 50, 10, "PHOTO", ln=1, align='C')
+                pdf.set_xy(x_pos + 25, pdf.get_y() + 70)
+                pdf.cell(img_width - 50, 10, f"IMAGE {i+1}", ln=1, align='C')
             
-            # MEMBER NAME (CORRECT SIZE & POSITION)
+            # WORKOUT DETAILS SECTION (Below Image)
             pdf.set_y(pdf.get_y() + img_height + 15)
-            pdf.set_font('Arial', 'B', 15)  # Medium name size
-            pdf.set_text_color(0, 0, 0)
-            clean_name = clean_text_for_pdf(name, 30)
-            pdf.multi_cell(0, 12, clean_name.upper(), align='C')  # Use multi_cell for better text
-            
-            # WORKOUT INFO SECTION
-            pdf.ln(8)
-            pdf.set_font('Arial', 'B', 11)
+            pdf.set_font('Arial', 'B', 12)
             pdf.set_text_color(0, 128, 0)
-            pdf.cell(0, 10, "💪 WORKOUT DETAILS", ln=1, align='C')
+            pdf.cell(0, 10, "💪 WORKOUT EXECUTION", ln=1, align='C')
             
             pdf.set_draw_color(0, 128, 0)
             pdf.line(18, pdf.get_y(), pdf.w - 18, pdf.get_y())
-            pdf.ln(5)
+            pdf.ln(8)
             
-            # Workout plan template with proper spacing
+            # Workout plan template
             pdf.set_font('Arial', '', 9)
             pdf.set_text_color(60, 60, 60)
             
-            workout_lines = [
-                "📋 EXERCISES: _______________________________________________",
-                "⚡ SETS/REPS: ______________________________________________",
-                "⏱️ REST TIME: _______________________________________________",
-                "💧 WATER BREAKS: __________________________________________",
-                "📈 PROGRESS NOTES: ________________________________________"
+            details = [
+                f"📋 EXERCISES FOR {clean_workout_name.upper()}:",
+                "⚡ SETS × REPS: _______________________________",
+                "⏱️  REST PERIODS: ____________________________",
+                "🔥 INTENSITY LEVEL: _________________________",
+                "💧 HYDRATION BREAKS: ________________________",
+                "📈 PROGRESS TRACKING: ______________________",
+                "👟 EQUIPMENT NEEDED: _______________________"
             ]
             
-            for line in workout_lines:
+            for detail in details:
                 pdf.ln(6)
-                pdf.cell(0, 8, line, ln=1, align='L')
+                pdf.cell(0, 8, detail, ln=1, align='L')
             
-            # Workout number footer
-            pdf.ln(10)
-            pdf.set_font('Arial', 'B', 10)
+            # Workout summary footer
+            pdf.ln(12)
+            pdf.set_font('Arial', 'B', 11)
             pdf.set_text_color(0, 128, 0)
-            pdf.cell(0, 8, f"WORKOUT SESSION #{i+1} - TRAIN HARD!", ln=1, align='C')
+            pdf.cell(0, 10, f"WORKOUT {i+1}: {clean_workout_name.upper()}", ln=1, align='C')
+            pdf.set_font('Arial', '', 8)
+            pdf.cell(0, 8, "PRINT AND TRAIN HARD! 💪", ln=1, align='C')
         
         # Save PDF
         timestamp = int(datetime.now().timestamp())
-        pdf_path = f"gym_workout_{timestamp}.pdf"
+        pdf_path = f"workout_plan_{timestamp}.pdf"
         pdf.output(pdf_path)
         
         success_msg = f"""
-✅ **GYM WORKOUT PLAN READY!**
-🏋️ **Title**: {clean_title[:30]}...
-💪 **{len(names_list)} Workouts** (Workout 1, Workout 2, etc.)
-📄 **Medium Sizes**: Photos 100×120px | Names 15pt
-📝 **TEXT FIXED**: All input text displays correctly
-📑 **{len(names_list) + 1} Pages** - Perfect gym layout!
+✅ **WORKOUT PLAN GENERATED!**
+🏋️ **Main Title**: {clean_main_title[:30]}...
+💪 **{len(names_list)} Workouts**: 
+   • Workout 1: {clean_text_for_pdf(names_list[0])[:20]}...
+   • Workout {len(names_list)}: {clean_text_for_pdf(names_list[-1])[:20]}...
+📄 **Layout**: Title Page → Workout Names + Images
+📏 **Perfect Sizes**: Images 110×130px | Names 14pt
+📑 **{len(names_list) + 1} Pages** - Ready for gym!
         """
         return pdf_path, success_msg
         
     except Exception as e:
         return None, f"❌ Error: {str(e)}"
 
-# ENHANCED GYM UI/UX
+# WORKOUT-FOCUSED UI
 with gr.Blocks(
     title="Gym Workout Plan Generator",
     theme=gr.themes.Soft(primary_hue="green"),
     css="""
-    .gradio-container {max-width: 950px !important;}
+    .gradio-container {max-width: 1000px !important;}
     .header {background: linear-gradient(135deg, #16a34a, #15803d); color: white; padding: 25px; border-radius: 15px; text-align: center;}
     .input-box {background: #f0fdf4; padding: 18px; border-radius: 10px; border: 2px solid #dcfce7; margin: 10px 0;}
-    .btn-generate {background: linear-gradient(45deg, #16a34a, #15803d) !important; font-size: 17px; padding: 12px;}
+    .btn-generate {background: linear-gradient(45deg, #16a34a, #15803d) !important; font-size: 18px; padding: 12px;}
     .status {padding: 15px; border-radius: 10px; font-size: 15px; margin: 12px 0;}
     .success {background: #dcfce7; color: #166534; border-left: 5px solid #16a34a;}
-    .gym-info {background: #ecfdf5; padding: 12px; border-radius: 8px; border-left: 4px solid #16a34a; margin: 8px 0;}
-    .preview-text {font-family: monospace; background: #f3f4f6; padding: 10px; border-radius: 5px; font-size: 12px;}
+    .workout-info {background: #ecfdf5; padding: 12px; border-radius: 8px; border-left: 4px solid #16a34a;}
+    .preview-workouts {background: #f0fdf4; padding: 10px; border-radius: 6px; font-family: monospace;}
     """
 ) as demo:
     
-    # Enhanced Header
+    # Header
     gr.HTML("""
     <div class="header">
-        <h2>🏋️‍♂️ Professional Gym Workout Generator</h2>
-        <p><strong>Fixed Text Display • Workout 1, 2, 3... • Perfect Sizes</strong></p>
+        <h2>🏋️‍♂️ Workout Plan Generator</h2>
+        <p><strong>Main Title → Workout Names → Images | Perfect Gym Layout</strong></p>
     </div>
     """)
     
     gr.Markdown("""
-    ### 🚀 Features:
-    - **✅ TEXT FIXED**: All your input text displays correctly in PDF
-    - **📝 Workout Numbers**: "Workout 1", "Workout 2", etc. (not member #)
-    - **📏 Medium Sizes**: Photos 100×120px | Names 15pt | Perfect fit
-    - **💪 Gym Template**: Workout details + trainer notes space
-    - **🎨 Green Theme**: Professional gym colors
+    ### 📋 Structure:
+    1. **FIRST**: Main Program Title Page
+    2. **NEXT**: Individual Workout Pages
+       - Workout 1 Name + Image
+       - Workout 2 Name + Image  
+       - Workout 3 Name + Image...
+    3. **Each Page**: Workout name, exercise image, workout template
     """)
     
-    gr.HTML('<div class="gym-info">💡 <strong>Input Text Preview:</strong> Your title and names will display exactly as entered (with cleaning for PDF compatibility).</div>')
+    gr.HTML('<div class="workout-info">💡 <strong>Workout Names Example:</strong> "Chest Day", "Legs & Core", "Full Body HIIT", "Arms & Shoulders"</div>')
     
-    # Enhanced Inputs with Preview
+    # Inputs - Clear Structure
     with gr.Row():
         with gr.Column(scale=2):
-            title_input = gr.Textbox(
-                label="🏋️ Workout Plan Title *",
-                placeholder="e.g., '6 Week Strength Program' or 'Beginner Muscle Building'",
+            main_title_input = gr.Textbox(
+                label="🎯 MAIN PROGRAM TITLE *",
+                placeholder="e.g., '12 Week Muscle Building Program' or 'Beginner Strength Cycle'",
                 lines=2,
                 elem_classes="input-box"
             )
             
-            title_preview = gr.Markdown("", elem_classes="preview-text")
-            
-            names_input = gr.Textbox(
-                label="👥 Member Names * (Comma separated)",
-                placeholder="John Smith, Jane Doe, Mike Johnson, Sarah Wilson",
-                lines=3,
+            workout_names_input = gr.Textbox(
+                label="💪 WORKOUT NAMES * (Comma separated)",
+                placeholder="Chest Day, Leg Day, Back & Biceps, Cardio Burn, Full Body",
+                lines=4,
                 elem_classes="input-box"
             )
             
-            names_preview = gr.Markdown("", elem_classes="preview-text")
+            workout_preview = gr.Markdown("", elem_classes="preview-workouts")
         
         with gr.Column(scale=1):
             images_input = gr.File(
-                label="📸 Member Photos *",
+                label="🖼️ Workout Images *",
                 file_count="multiple",
                 file_types=[".jpg", ".jpeg", ".png", ".webp"],
-                elem_classes="input-box"
+                elem_classes="input-box",
+                info="Upload images for each workout (exercise demo photos)"
             )
     
     # Controls
     with gr.Row():
-        generate_btn = gr.Button("🚀 Generate Gym Workout PDF", variant="primary", elem_classes="btn-generate")
+        generate_btn = gr.Button("🚀 Generate Workout Plan PDF", variant="primary", elem_classes="btn-generate")
         clear_btn = gr.Button("🗑️ Clear All", variant="secondary")
     
     # Outputs
-    pdf_output = gr.File(label="📥 Download Gym Workout PDF", elem_classes="input-box")
+    pdf_output = gr.File(label="📥 Download Workout Plan PDF", elem_classes="input-box")
     status_output = gr.Markdown("Ready to generate workout plan...", elem_classes="status success")
     
-    # Events with Text Preview
-    def preview_title(title):
-        clean = clean_text_for_pdf(title)
-        return f"📄 PDF Title: '{clean}'"
-    
-    def preview_names(names):
-        names_list = [clean_text_for_pdf(n.strip()) for n in names.split(',') if n.strip()]
+    # Events
+    def preview_workouts(workout_names):
+        names_list = [clean_text_for_pdf(n.strip()) for n in workout_names.split(',') if n.strip()]
         if names_list:
-            return f"👥 Names in PDF:\n" + " | ".join(names_list)
-        return "👥 No names detected"
+            preview = "📋 **Workouts in PDF:**\n"
+            for i, name in enumerate(names_list[:5], 1):  # Show first 5
+                preview += f"• **Workout {i}**: {name}\n"
+            if len(names_list) > 5:
+                preview += f"... and {len(names_list)-5} more"
+            return preview
+        return "📋 No workouts detected"
     
-    title_input.change(
-        fn=preview_title,
-        inputs=[title_input],
-        outputs=[title_preview]
-    )
+    def validate_inputs(main_title, workout_names, images):
+        if not main_title.strip():
+            return "❌ Main title required!", gr.update(visible=False)
+        names_list = [n.strip() for n in workout_names.split(',') if n.strip()]
+        if len(names_list) != len(images or []):
+            return f"❌ {len(names_list)} workouts need {len(names_list)} images!", gr.update(visible=False)
+        return "✅ Ready to generate!", gr.update(visible=True)
     
-    names_input.change(
-        fn=preview_names,
-        inputs=[names_input],
-        outputs=[names_preview]
+    main_title_input.change(fn=lambda x: x, inputs=[main_title_input], outputs=[status_output])
+    workout_names_input.change(
+        fn=preview_workouts,
+        inputs=[workout_names_input],
+        outputs=[workout_preview]
     )
     
-    def count_workouts(names):
-        names_list = [n.strip() for n in names.split(',') if n.strip()]
-        count = len(names_list)
-        if count == 0:
-            return "❌ Add members (1-30)"
-        elif count > 30:
-            return f"⚠️ {count} workouts - MAX 30!"
-        return f"✅ {count} workouts ready (Workout 1 to {count})"
-    
     generate_btn.click(
-        fn=generate_gym_workout_pdf,
-        inputs=[title_input, names_input, images_input],
+        fn=generate_workout_pdf,
+        inputs=[main_title_input, workout_names_input, images_input],
         outputs=[pdf_output, status_output]
     )
     
     clear_btn.click(
-        fn=lambda: ("", None, "", None, None, "", "Form cleared!"),
-        outputs=[title_input, title_preview, names_input, images_input, names_preview, pdf_output, status_output]
-    )
-    
-    names_input.change(
-        fn=count_workouts,
-        inputs=[names_input],
-        outputs=[status_output]
+        fn=lambda: ("", "", None, "", "Form cleared!"),
+        outputs=[main_title_input, workout_names_input, images_input, workout_preview, status_output]
     )
 
 if __name__ == "__main__":