fix HF API key setup

api_config.py

@@ -181,4 +181,4 @@ CONFIG_FOR_ADMIN_HF_USERS = {
     "max_clip_height": 640, # 512, # 448, # 416,
 }
 
-CONFIG_FOR_ANONYMOUS_USERS
+CONFIG_FOR_ADMIN_HF_USERS = CONFIG_FOR_PRO_HF_USERS

api_core.py

@@ -258,16 +258,24 @@ class VideoGenerationAPI:
                 lambda: self.hf_api.whoami(token=token)
             )
             
-            logger.info(f"Token valid for user: {user_info.name}")
+            # Handle both object and dict response formats from whoami
+            username = user_info.get('name') if isinstance(user_info, dict) else getattr(user_info, 'name', None)
+            is_pro = user_info.get('is_pro') if isinstance(user_info, dict) else getattr(user_info, 'is_pro', False)
+            
+            if not username:
+                logger.error(f"Could not determine username from user_info: {user_info}")
+                return 'anon'
+                
+            logger.info(f"Token valid for user: {username}")
             
             # Determine the user role based on the information
             user_role: UserRole
             
             # Check if the user is an admin
-            if user_info.name in ADMIN_ACCOUNTS:
+            if username in ADMIN_ACCOUNTS:
                 user_role = 'admin'
             # Check if the user has a pro account
-            elif hasattr(user_info, 'is_pro') and user_info.is_pro:
+            elif is_pro:
                 user_role = 'pro'
             else:
                 user_role = 'normal'
@@ -276,7 +284,7 @@ class VideoGenerationAPI:
             self.user_role_cache[token] = {
                 'role': user_role,
                 'timestamp': current_time,
-                'username': user_info.name
+                'username': username
             }
             
             return user_role

build/web/flutter_bootstrap.js

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

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": "2e5faa3ab3c446c3c3eaa2ce252f366a",
+const RESOURCES = {"flutter_bootstrap.js": "1345e31de2b3abaeb866a96f3ed33f5f",
 "version.json": "b5eaae4fc120710a3c35125322173615",
 "index.html": "cd2094e3989e3eb0424e47d7d188c298",
 "/": "cd2094e3989e3eb0424e47d7d188c298",
-"main.dart.js": "14a165f5350b8bcff180f1cf376cf819",
+"main.dart.js": "f8d70b55b2c132f06a67b5857e1a76a9",
 "flutter.js": "83d881c1dbb6d6bcd6b42e274605b69c",
 "aitube.svg": "26140ba0d153b213b122bc6ebcc17f6c",
 "favicon.png": "c8a183c516004e648a7bac7497c89b97",

build/web/index.html

@@ -156,7 +156,7 @@
   </script>
   
   <!-- Add version parameter for cache busting -->
-  <script src="flutter_bootstrap.js?v=1746641479" async></script>
+  <script src="flutter_bootstrap.js?v=1746643716" async></script>
   
   <!-- Add cache busting script -->
   <script>

lib/screens/settings_screen.dart

@@ -67,11 +67,48 @@ class _SettingsScreenState extends State<SettingsScreen> {
                     obscureText: true,
                     onChanged: (value) async {
                       await _settingsService.setHuggingfaceApiKey(value);
+                      
+                      // Show a snackbar to indicate the API key was saved
+                      if (context.mounted) {
+                        ScaffoldMessenger.of(context).showSnackBar(
+                          const SnackBar(
+                            content: Text('API Key saved. Reconnecting...'),
+                            duration: Duration(seconds: 2),
+                          ),
+                        );
+                      }
+                      
                       // Reinitialize the websocket connection when the API key changes
                       final websocket = WebSocketApiService();
-                      if (websocket.isConnected) {
+                      try {
+                        // First dispose the current connection
                         await websocket.dispose();
+                        
+                        // Then create a new connection with the new API key
                         await websocket.connect();
+                        
+                        // Finally, initialize the connection completely
+                        await websocket.initialize();
+                        
+                        // Show success message
+                        if (context.mounted) {
+                          ScaffoldMessenger.of(context).showSnackBar(
+                            const SnackBar(
+                              content: Text('Connected successfully with new API key'),
+                              backgroundColor: Colors.green,
+                            ),
+                          );
+                        }
+                      } catch (e) {
+                        // Show error message if connection fails
+                        if (context.mounted) {
+                          ScaffoldMessenger.of(context).showSnackBar(
+                            SnackBar(
+                              content: Text('Failed to connect: $e'),
+                              backgroundColor: Colors.red,
+                            ),
+                          );
+                        }
                       }
                     },
                   ),

lib/screens/video_screen.dart

@@ -511,7 +511,7 @@ class _VideoScreenState extends State<VideoScreen> {
           IconButton(
             icon: const Icon(Icons.share, color: AiTubeColors.primary),
             onPressed: _shareVideo,
-            tooltip: 'Share video',
+            tooltip: 'Share prompt',
           ),
         ],
       ),