small improvements

api_config.py

@@ -90,32 +90,32 @@ CONFIG_FOR_STANDARD_HF_USERS = {
     "max_rendering_time_per_client_per_video_in_sec": 15 * 60,
 
     "min_num_inference_steps": 2,
-    "default_num_inference_steps": 4,
-    "max_num_inference_steps": 4,
+    "default_num_inference_steps": 8,
+    "max_num_inference_steps": 8,
     
     "min_num_frames": 9, # 8 + 1
     "default_num_frames": 81, # 8*10 + 1
     "max_num_frames": 81,
 
     "min_clip_duration_seconds": 1,
-    "default_clip_duration_seconds": 2,
-    "max_clip_duration_seconds": 2,
+    "default_clip_duration_seconds": 3,
+    "max_clip_duration_seconds": 3,
 
-    "min_clip_playback_speed": 0.65,
-    "default_clip_playback_speed": 0.65,
-    "max_clip_playback_speed": 0.65,
+    "min_clip_playback_speed": 0.7,
+    "default_clip_playback_speed": 0.7,
+    "max_clip_playback_speed": 0.7,
 
     "min_clip_framerate": 8,
     "default_clip_framerate": 25,
     "max_clip_framerate": 25,
 
     "min_clip_width": 544,
-    "default_clip_width": 768, # 640,
-    "max_clip_width": 768, # 640,
+    "default_clip_width": 1216, # 768, # 640,
+    "max_clip_width": 1216, # 768, # 640,
 
     "min_clip_height": 320,
-    "default_clip_height": 448, # 416,
-    "max_clip_height": 448, # 416,
+    "default_clip_height": 640, # 448, # 416,
+    "max_clip_height": 640, # 448, # 416,
 }
 
 # Hugging Face users with a Pro may enjoy an improved experience
@@ -123,20 +123,20 @@ CONFIG_FOR_PRO_HF_USERS = {
     "max_rendering_time_per_client_per_video_in_sec": 20 * 60,
 
     "min_num_inference_steps": 2,
-    "default_num_inference_steps": 4,
-    "max_num_inference_steps": 4,
+    "default_num_inference_steps": 8,
+    "max_num_inference_steps": 8,
     
     "min_num_frames": 9, # 8 + 1
     "default_num_frames": 81, # 8*10 + 1
     "max_num_frames": 81,
 
     "min_clip_duration_seconds": 1,
-    "default_clip_duration_seconds": 2,
-    "max_clip_duration_seconds": 2,
+    "default_clip_duration_seconds": 3,
+    "max_clip_duration_seconds": 3,
 
-    "min_clip_playback_speed": 0.65,
-    "default_clip_playback_speed": 0.65,
-    "max_clip_playback_speed": 0.65,
+    "min_clip_playback_speed": 0.7,
+    "default_clip_playback_speed": 0.7,
+    "max_clip_playback_speed": 0.7,
 
     "min_clip_framerate": 8,
     "default_clip_framerate": 25,
@@ -167,8 +167,8 @@ CONFIG_FOR_ADMIN_HF_USERS = {
     "max_clip_duration_seconds": 4,
 
     "min_clip_playback_speed": 0.7,
-    "default_clip_playback_speed": 0.8,
-    "max_clip_playback_speed": 0.9,
+    "default_clip_playback_speed": 0.7,
+    "max_clip_playback_speed": 1.0,
 
     "min_clip_framerate": 8,
     "default_clip_framerate": 30,

api_core.py

@@ -353,7 +353,7 @@ title: \""""
                 
                 # Check if the description still contains placeholder tags like <LOCATION>, <GENDER>, etc.
                 if re.search(r'<[A-Z_]+>', description):
-                    logger.warning(f"Description still contains placeholder tags: {description}")
+                    #logger.warning(f"Description still contains placeholder tags: {description}")
                     if current_attempt < max_attempts:
                         # Try again with a higher temperature
                         current_attempt += 1
@@ -615,8 +615,7 @@ Your caption:"""
         frame_rate = self.get_config_value(user_role, 'clip_framerate', options)
         
         # Log the user role and config values being used
-        logger.info(f"Generating video for user with role: {user_role}")
-        logger.info(f"Using config values: width={width}, height={height}, num_frames={num_frames}, steps={num_inference_steps}, fps={frame_rate}")
+        logger.info(f"Using config values: width={width}, height={height}, num_frames={num_frames}, steps={num_inference_steps}, fps={frame_rate} | role: {user_role}")
         
         json_payload = {
             "inputs": {

assets/config/aitube.yaml

@@ -7,21 +7,24 @@ render_queue:
   buffer_size: 3
 
   # how many requests for clips can be run in parallel
-  max_concurrent_generations: 3
+  max_concurrent_generations: 2
 
   # start playback as soon as we have 1 video over 3 (25%)
   minimum_buffer_percent_to_start_playback: 5
 
+# it's OK to use high values here,
+# because some of those values are limited by the backend config,
+# such as the resoltuion or number of frames
 video:
   # default negative prompt to filter harmful content
-  default_negative_prompt: "gore, sex, blood, nudity, nude, porn, erotic, worst quality, deformed, distorted, disfigured, blurry, text, watermark"
+  default_negative_prompt: "pixelated, deformed, distorted, disfigured, blurry, text, watermark, low quality, gore, sex, blood, nudity, nude, porn, erotic"
 
   # transition time between each clip
   # the exit (older) clip will see its playback time reduced by this amount
   transition_buffer_duration_ms: 300
 
   # how long a generated clip should be, in Duration
-  original_clip_duration_seconds: 4
+  original_clip_duration_seconds: 3
 
   # The model works on resolutions that are divisible by 32
   # and number of frames that are divisible by 8 + 1 (e.g. 257).
@@ -34,7 +37,7 @@ video:
   # number of frames below 257.
 
   # number of inference steps
-  num_inference_steps: 8
+  num_inference_steps: 4
 
   guidance_scale: 1.0
 
@@ -42,8 +45,8 @@ video:
   # in frames per second (so an integer)
   original_clip_frame_rate: 25
 
-  original_clip_width: 768 # 544
-  original_clip_height: 448 # 320
+  original_clip_width: 1216 # 544
+  original_clip_height: 704 # 320
 
   # to do more with less, we slow down the videos (a 3s video will become a 4s video)
   # but if you are GPU rich feel feel to play them back at 100% of their speed!

assets/config/default.yaml

@@ -14,7 +14,7 @@ ui:
 
 video:
   # default negative prompt to filter harmful content
-  default_negative_prompt: "gore, sex, blood, nudity, nude, porn, erotic, worst quality, deformed, distorted, disfigured, blurry, text, watermark"
+  default_negative_prompt: "pixelated, deformed, distorted, disfigured, blurry, text, watermark, low quality, gore, sex, blood, nudity, nude, porn, erotic"
 
   # transition time between each clip
   # the exit (older) clip will see its playback time reduced by this amount

build/web/assets/AssetManifest.bin

@@ -1,8 +1 @@
-aitube.yamlaitube.yaml
-assets/config/README.md

-
assetassets/config/README.mdassets/config/aitube_high.yaml

-
assetassets/config/aitube_high.yamlassets/config/aitube_low.yaml

-
assetassets/config/aitube_low.yamlassets/config/default.yaml

-
assetassets/config/default.yamlassets/config/private.yaml

-
assetassets/config/private.yaml2packages/cupertino_icons/assets/CupertinoIcons.ttf

-
asset2packages/cupertino_icons/assets/CupertinoIcons.ttf
+

build/web/assets/AssetManifest.bin.json

@@ -1 +1 @@
-"DQYHF2Fzc2V0cy9jb25maWcvUkVBRE1FLm1kDAENAQcFYXNzZXQHF2Fzc2V0cy9jb25maWcvUkVBRE1FLm1kBx5hc3NldHMvY29uZmlnL2FpdHViZV9oaWdoLnlhbWwMAQ0BBwVhc3NldAceYXNzZXRzL2NvbmZpZy9haXR1YmVfaGlnaC55YW1sBx1hc3NldHMvY29uZmlnL2FpdHViZV9sb3cueWFtbAwBDQEHBWFzc2V0Bx1hc3NldHMvY29uZmlnL2FpdHViZV9sb3cueWFtbAcaYXNzZXRzL2NvbmZpZy9kZWZhdWx0LnlhbWwMAQ0BBwVhc3NldAcaYXNzZXRzL2NvbmZpZy9kZWZhdWx0LnlhbWwHGmFzc2V0cy9jb25maWcvcHJpdmF0ZS55YW1sDAENAQcFYXNzZXQHGmFzc2V0cy9jb25maWcvcHJpdmF0ZS55YW1sBzJwYWNrYWdlcy9jdXBlcnRpbm9faWNvbnMvYXNzZXRzL0N1cGVydGlub0ljb25zLnR0ZgwBDQEHBWFzc2V0BzJwYWNrYWdlcy9jdXBlcnRpbm9faWNvbnMvYXNzZXRzL0N1cGVydGlub0ljb25zLnR0Zg=="
+"DQUHF2Fzc2V0cy9jb25maWcvUkVBRE1FLm1kDAENAQcFYXNzZXQHF2Fzc2V0cy9jb25maWcvUkVBRE1FLm1kBxlhc3NldHMvY29uZmlnL2FpdHViZS55YW1sDAENAQcFYXNzZXQHGWFzc2V0cy9jb25maWcvYWl0dWJlLnlhbWwHGWFzc2V0cy9jb25maWcvY3VzdG9tLnlhbWwMAQ0BBwVhc3NldAcZYXNzZXRzL2NvbmZpZy9jdXN0b20ueWFtbAcaYXNzZXRzL2NvbmZpZy9kZWZhdWx0LnlhbWwMAQ0BBwVhc3NldAcaYXNzZXRzL2NvbmZpZy9kZWZhdWx0LnlhbWwHMnBhY2thZ2VzL2N1cGVydGlub19pY29ucy9hc3NldHMvQ3VwZXJ0aW5vSWNvbnMudHRmDAENAQcFYXNzZXQHMnBhY2thZ2VzL2N1cGVydGlub19pY29ucy9hc3NldHMvQ3VwZXJ0aW5vSWNvbnMudHRm"

build/web/assets/AssetManifest.json

@@ -1 +1 @@
-{"assets/config/README.md":["assets/config/README.md"],"assets/config/aitube_high.yaml":["assets/config/aitube_high.yaml"],"assets/config/aitube.yaml":["assets/config/aitube.yaml"],"assets/config/default.yaml":["assets/config/default.yaml"],"assets/config/private.yaml":["assets/config/private.yaml"],"packages/cupertino_icons/assets/CupertinoIcons.ttf":["packages/cupertino_icons/assets/CupertinoIcons.ttf"]}
+{"assets/config/README.md":["assets/config/README.md"],"assets/config/aitube.yaml":["assets/config/aitube.yaml"],"assets/config/custom.yaml":["assets/config/custom.yaml"],"assets/config/default.yaml":["assets/config/default.yaml"],"packages/cupertino_icons/assets/CupertinoIcons.ttf":["packages/cupertino_icons/assets/CupertinoIcons.ttf"]}

build/web/assets/assets/config/{aitube_high.yaml → aitube.yaml}

@@ -1,24 +1,30 @@
 ui:
   product_name: AiTube
-  
+  showChatInVideoView: false
+
 render_queue:
   # how many clips should be stored in advance
-  buffer_size: 5
+  buffer_size: 3
 
   # how many requests for clips can be run in parallel
-  max_concurrent_generations: 4
+  max_concurrent_generations: 2
 
-  # start playback as soon as we have about 1 video over 3 (30~33%)
-  minimum_buffer_percent_to_start_playback: 30
+  # start playback as soon as we have 1 video over 3 (25%)
+  minimum_buffer_percent_to_start_playback: 5
 
+# it's OK to use high values here,
+# because some of those values are limited by the backend config,
+# such as the resoltuion or number of frames
 video:
+  # default negative prompt to filter harmful content
+  default_negative_prompt: "pixelated, deformed, distorted, disfigured, blurry, text, watermark, low quality, gore, sex, blood, nudity, nude, porn, erotic"
 
   # transition time between each clip
   # the exit (older) clip will see its playback time reduced by this amount
   transition_buffer_duration_ms: 300
 
   # how long a generated clip should be, in Duration
-  original_clip_duration_seconds: 4
+  original_clip_duration_seconds: 3
 
   # The model works on resolutions that are divisible by 32
   # and number of frames that are divisible by 8 + 1 (e.g. 257).
@@ -31,20 +37,17 @@ video:
   # number of frames below 257.
 
   # number of inference steps
-  # this has a direct impact in performance obviously,
-  # you can try to go to low values like 12 or 14 on "safe bet" prompts,
-  # but if you need a more uncommon topic, you need to go to 18 steps or more
   num_inference_steps: 4
 
   guidance_scale: 1.0
-  
+
   # original frame-rate of each clip (before we slow them down)
   # in frames per second (so an integer)
   original_clip_frame_rate: 25
 
-  original_clip_width: 736 # 544
-  original_clip_height: 448 # 320
+  original_clip_width: 1216 # 544
+  original_clip_height: 704 # 320
 
   # to do more with less, we slow down the videos (a 3s video will become a 4s video)
   # but if you are GPU rich feel feel to play them back at 100% of their speed!
-  clip_playback_speed: 1.0 # 0.7
+  clip_playback_speed: 0.7

build/web/assets/assets/config/{aitube_low.yaml → custom.yaml}

@@ -1,5 +1,5 @@
 ui:
-  product_name: AiTube
+  product_name: Custom
   showChatInVideoView: false
 
 render_queue:
@@ -13,6 +13,8 @@ render_queue:
   minimum_buffer_percent_to_start_playback: 5
 
 video:
+  # default negative prompt to filter harmful content
+  default_negative_prompt: "pixelated, deformed, distorted, disfigured, blurry, text, watermark, low quality, gore, sex, blood, nudity, nude, porn, erotic"
 
   # transition time between each clip
   # the exit (older) clip will see its playback time reduced by this amount
@@ -45,4 +47,4 @@ video:
 
   # to do more with less, we slow down the videos (a 3s video will become a 4s video)
   # but if you are GPU rich feel feel to play them back at 100% of their speed!
-  clip_playback_speed: 0.7
+  clip_playback_speed: 0.65

build/web/assets/assets/config/default.yaml

@@ -13,6 +13,8 @@ ui:
   minimum_buffer_percent_to_start_playback: 25
 
 video:
+  # default negative prompt to filter harmful content
+  default_negative_prompt: "pixelated, deformed, distorted, disfigured, blurry, text, watermark, low quality, gore, sex, blood, nudity, nude, porn, erotic"
 
   # transition time between each clip
   # the exit (older) clip will see its playback time reduced by this amount

build/web/assets/assets/config/private.yaml

@@ -1,3 +0,0 @@
-ui:
-  product_name: Private
-  

build/web/flutter_bootstrap.js

@@ -39,6 +39,6 @@ _flutter.buildConfig = {"engineRevision":"382be0028d370607f76215a9be322e5514b263
 
 _flutter.loader.load({
   serviceWorkerSettings: {
-    serviceWorkerVersion: "794883524"
+    serviceWorkerVersion: "2953326891"
   }
 });

build/web/flutter_service_worker.js

@@ -3,11 +3,11 @@ const MANIFEST = 'flutter-app-manifest';
 const TEMP = 'flutter-temp-cache';
 const CACHE_NAME = 'flutter-app-cache';
 
-const RESOURCES = {"flutter_bootstrap.js": "a1978150e41f01842b52df7eb3c181aa",
+const RESOURCES = {"flutter_bootstrap.js": "473f7db41bcf45ab1477c0d952890dcd",
 "version.json": "b5eaae4fc120710a3c35125322173615",
 "index.html": "f34c56fffc6b38f62412a5db2315dec8",
 "/": "f34c56fffc6b38f62412a5db2315dec8",
-"main.dart.js": "aaaecee8e349d51552b33c8f81812c5c",
+"main.dart.js": "e505129ff53a6863def8471d833dbb92",
 "flutter.js": "83d881c1dbb6d6bcd6b42e274605b69c",
 "favicon.png": "5dcef449791fa27946b3d35ad8803796",
 "icons/Icon-192.png": "ac9a721a12bbc803b44f645561ecb1e1",
@@ -15,19 +15,18 @@ const RESOURCES = {"flutter_bootstrap.js": "a1978150e41f01842b52df7eb3c181aa",
 "icons/Icon-maskable-512.png": "301a7604d45b3e739efc881eb04896ea",
 "icons/Icon-512.png": "96e752610906ba2a93c65f8abe1645f1",
 "manifest.json": "7dc942a630334c1017089988a6ca07d4",
-"assets/AssetManifest.json": "b2202b29892f6fb630f40ab2f4950300",
+"assets/AssetManifest.json": "51a53d0237971d07d6d88304c41bf6fb",
 "assets/NOTICES": "f0cfae681e209e19b2b144a9f062a96f",
 "assets/FontManifest.json": "dc3d03800ccca4601324923c0b1d6d57",
-"assets/AssetManifest.bin.json": "b6a02db76c67a9b7f4f2e1623f91dcb7",
+"assets/AssetManifest.bin.json": "f7a7f5619c8ca1575d7a5e0c6cd4d529",
 "assets/packages/cupertino_icons/assets/CupertinoIcons.ttf": "33b7d9392238c04c131b6ce224e13711",
 "assets/shaders/ink_sparkle.frag": "ecc85a2e95f5e9f53123dcaf8cb9b6ce",
-"assets/AssetManifest.bin": "5894fe5676e62dc22403a833f2313e43",
-"assets/fonts/MaterialIcons-Regular.otf": "98b0ef6d6144942ca89cc207f3e08986",
-"assets/assets/config/private.yaml": "97a9ec367206bea5dce64faf94b66332",
+"assets/AssetManifest.bin": "6c597105edcadb9c676bdc998c88545a",
+"assets/fonts/MaterialIcons-Regular.otf": "f1343247a767efee2447b3f63c9498b4",
 "assets/assets/config/README.md": "07a87720dd00dd1ca98c9d6884440e31",
-"assets/assets/config/aitube_high.yaml": "c030f221344557ecf05aeef30f224502",
-"assets/assets/config/default.yaml": "e98187e5a53a8b0d8bf2c5cd9dd0a365",
-"assets/assets/config/aitube.yaml": "462167d380f349ce42a272b7d61e2220",
+"assets/assets/config/custom.yaml": "da18366f4b2f24a3bc911d86bd773a79",
+"assets/assets/config/aitube.yaml": "29ed15827ee8364e390a3b446535067a",
+"assets/assets/config/default.yaml": "ba11c9ae686f1317a29bce114f0f9fc9",
 "canvaskit/skwasm.js": "ea559890a088fe28b4ddf70e17e60052",
 "canvaskit/skwasm.js.symbols": "9fe690d47b904d72c7d020bd303adf16",
 "canvaskit/canvaskit.js.symbols": "27361387bc24144b46a745f1afe92b50",

lib/screens/settings_screen.dart

@@ -161,7 +161,14 @@ class _SettingsScreenState extends State<SettingsScreen> {
                           final confirmed = await showDialog<bool>(
                             context: context,
                             builder: (context) => AlertDialog(
-                              title: const Text('Clear Cache'),
+                              title: const Text(
+                                'Clear Cache',
+                                style: TextStyle(
+                                  color: AiTubeColors.onBackground,
+                                  fontSize: 20,
+                                  fontWeight: FontWeight.bold,
+                                ),
+                              ),
                               content: const Text(
                                 'Are you sure you want to clear all cached data? '
                                 'This will remove all saved search results and videos.',

lib/screens/video_screen.dart

@@ -1,6 +1,7 @@
 // lib/screens/video_screen.dart
 import 'dart:async';
 
+import 'package:aitube2/screens/home_screen.dart';
 import 'package:aitube2/widgets/chat_widget.dart';
 import 'package:aitube2/widgets/search_box.dart';
 import 'package:aitube2/widgets/web_utils.dart';
@@ -323,8 +324,28 @@ class _VideoScreenState extends State<VideoScreen> {
         final isWideScreen = constraints.maxWidth >= 900;
         
         return Scaffold(
-          appBar: AppBar(
-            titleSpacing: 0,
+          appBar: PreferredSize(
+            preferredSize: const Size.fromHeight(kToolbarHeight + 16),
+            child: Padding(
+              padding: const EdgeInsets.only(top: 16),
+              child: AppBar(
+                leading: IconButton(
+                  icon: Navigator.canPop(context) 
+                    ? const Icon(Icons.arrow_back, color: AiTubeColors.onBackground)
+                    : const Icon(Icons.home, color: AiTubeColors.onBackground),
+                  onPressed: () {
+                    if (Navigator.canPop(context)) {
+                      Navigator.pop(context);
+                    } else {
+                      // Navigate to home screen if we can't go back
+                      Navigator.pushReplacement(
+                        context,
+                        MaterialPageRoute(builder: (context) => const HomeScreen()),
+                      );
+                    }
+                  },
+                ),
+                titleSpacing: 0,
             title: Padding(
               padding: const EdgeInsets.all(8),
               child: SearchBox(
@@ -346,6 +367,8 @@ class _VideoScreenState extends State<VideoScreen> {
                 onPressed: _isConnected ? null : _initializeConnection,
               ),
             ],
+              ),
+            ),
           ),
           body: SafeArea(
             child: isWideScreen

lib/widgets/ai_content_disclaimer.dart

@@ -110,7 +110,7 @@ class AiContentDisclaimer extends StatelessWidget {
                       ),
                     ],
                   ),
-                  const SizedBox(height: 10),
+                  const SizedBox(height: 20),
                   Row(
                     mainAxisAlignment: MainAxisAlignment.center,
                     crossAxisAlignment: CrossAxisAlignment.baseline,
@@ -135,7 +135,7 @@ class AiContentDisclaimer extends StatelessWidget {
                       ),
                     ],
                   ),
-                  const SizedBox(height: 10),
+                  const SizedBox(height: 20),
                   Text(
                     'artificial intelligence',
                     style: GoogleFonts.arimo(
@@ -153,7 +153,7 @@ class AiContentDisclaimer extends StatelessWidget {
                       ],
                     ),
                   ),
-                  const SizedBox(height: 10),
+                  const SizedBox(height: 20),
                   Text(
                     'and may contain hallucinations or factual inaccuracies.',
                     style: GoogleFonts.arimo(

lib/widgets/video_player_widget.dart

@@ -613,7 +613,8 @@ class _VideoPlayerWidgetState extends State<VideoPlayerWidget> with WidgetsBindi
   Widget _buildPlaceholder() {
     // Use our new AI Content Disclaimer widget as the placeholder
     if (widget.initialThumbnailUrl?.isEmpty ?? true) {
-      return const AiContentDisclaimer(isInteractive: false);
+      // Set isInteractive to true as we generate content on-the-fly
+      return const AiContentDisclaimer(isInteractive: true);
     }
 
     try {