Update app.py

app.py

@@ -145,10 +145,8 @@ def get_duration(
     duration_seconds, guidance_scale, guidance_scale_2,
     seed, randomize_seed, progress,
 ):
-    # --- CRITICAL FIX: Handle NoneType for input_image ---
     if input_image is None:
-        return 120 # Return default duration if image is missing to prevent crash
-    # -----------------------------------------------------
+        return 120
     BASE_FRAMES_HEIGHT_WIDTH = 81 * 832 * 624
     BASE_STEP_DURATION = 15
     
@@ -205,7 +203,7 @@ def generate_video(
 # =========================================================
 with gr.Blocks(theme=gr.themes.Soft()) as demo:
     
-    # --- ADVERTISEMENT BANNER FOR DREAM HUB PRO ---
+    # --- TOP ADVERTISEMENT BANNER ---
     gr.HTML("""
     <div style="background: linear-gradient(90deg, #4f46e5, #9333ea); color: white; padding: 15px; border-radius: 10px; text-align: center; margin-bottom: 10px; box-shadow: 0 4px 15px rgba(0,0,0,0.1);">
         <div style="display: flex; align-items: center; justify-content: center; gap: 20px; flex-wrap: wrap;">
@@ -222,23 +220,40 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
     </div>
     """)
     
-    # --- NEW: PROFESSIONAL YOUTUBE CHANNEL BANNER ---
+    # --- PROFESSIONAL YOUTUBE EMBED SECTION ---
+    # This section now includes the embedded video as an expert display
     gr.HTML("""
-    <div style="background: linear-gradient(90deg, #cc0000, #ff0000); color: white; padding: 12px; border-radius: 10px; text-align: center; margin-bottom: 20px; box-shadow: 0 4px 15px rgba(255,0,0,0.2);">
-        <div style="display: flex; align-items: center; justify-content: center; gap: 20px; flex-wrap: wrap;">
-            <div style="display: flex; align-items: center; gap: 10px;">
-                <span style="font-size: 28px;">▶️</span>
+    <div style="background: linear-gradient(135deg, #b90000 0%, #ff0000 100%); color: white; padding: 25px; border-radius: 16px; text-align: center; margin-bottom: 25px; box-shadow: 0 10px 30px rgba(185, 0, 0, 0.3);">
+        <div style="display: flex; align-items: center; justify-content: center; gap: 25px; flex-wrap: wrap; margin-bottom: 20px;">
+            <div style="display: flex; align-items: center; gap: 15px;">
+                <div style="background: white; width: 50px; height: 50px; border-radius: 50%; display: flex; align-items: center; justify-content: center; box-shadow: 0 4px 8px rgba(0,0,0,0.2);">
+                    <span style="font-size: 24px;">▶️</span>
+                </div>
                 <div style="text-align: left;">
-                    <h3 style="margin: 0; font-weight: bold; font-size: 18px;">Imagination Engineering</h3>
-                    <p style="margin: 2px 0 0 0; opacity: 0.9; font-size: 13px;">Tutorials, AI Engineering & Creative Tech</p>
+                    <h3 style="margin: 0; font-weight: 800; font-size: 22px; letter-spacing: 0.5px;">Imagination Engineering</h3>
+                    <p style="margin: 4px 0 0 0; opacity: 0.95; font-size: 14px; font-weight: 400;">Mastering AI & Creative Tech</p>
                 </div>
             </div>
             <a href="https://www.youtube.com/@ImaginationEngineering" target="_blank" style="text-decoration: none;">
-                <button style="background-color: white; color: #cc0000; border: none; padding: 8px 25px; border-radius: 25px; font-weight: bold; cursor: pointer; transition: transform 0.2s; font-size: 15px; box-shadow: 0 2px 5px rgba(0,0,0,0.2);">
-                    Subscribe & Watch 📺
+                <button style="background-color: white; color: #cc0000; border: none; padding: 10px 28px; border-radius: 30px; font-weight: 700; cursor: pointer; transition: transform 0.2s, box-shadow 0.2s; font-size: 15px; box-shadow: 0 4px 12px rgba(0,0,0,0.2);">
+                    SUBSCRIBE & WATCH 📺
                 </button>
             </a>
         </div>
+        
+        <!-- VIDEO EMBED CONTAINER -->
+        <div style="width: 100%; max-width: 650px; margin: 0 auto; border-radius: 12px; overflow: hidden; box-shadow: 0 15px 40px rgba(0,0,0,0.4); border: 4px solid rgba(255,255,255,0.15);">
+            <div style="position: relative; padding-bottom: 56.25%; height: 0; overflow: hidden;">
+                <iframe 
+                    src="https://www.youtube.com/embed/w_7wL_i3f1k?rel=0&modestbranding=1" 
+                    style="position: absolute; top: 0; left: 0; width: 100%; height: 100%;" 
+                    frameborder="0" 
+                    allow="accelerometer; autoplay; clipboard-write; encrypted-media; gyroscope; picture-in-picture; web-share" 
+                    allowfullscreen 
+                    title="Imagination Engineering Feature">
+                </iframe>
+            </div>
+        </div>
     </div>
     """)
     # ---------------------------------------------