Fix syntax error in avatar_app.py

Co-authored-by: Amp <amp@ampcode.com>
Amp-Thread-ID: https://ampcode.com/threads/T-a617baec-31b4-4481-b9ff-5899d702f52f

avatar_app.py

@@ -318,6 +318,29 @@ class OmniAvatarAPI:
                     logger.info(f"? SUCCESS: Downloaded models loaded - Video: {video_files}, Audio: {audio_files}")
                     return True
 
+            # Fallback: Check traditional OmniAvatar paths
+            traditional_paths = [
+                "./pretrained_models/Wan2.1-T2V-14B",
+                "./pretrained_models/OmniAvatar-14B",
+                "./pretrained_models/wav2vec2-base-960h"
+            ]
+
+            if all(os.path.exists(path) for path in traditional_paths):
+                self.model_loaded = True
+                logger.info("? SUCCESS: Traditional OmniAvatar models found")
+                return True
+
+            # No models found
+            logger.warning("?? WARNING: No models found (neither downloaded nor traditional)")
+            self.model_loaded = False
+            return True  # App can still run in TTS-only mode
+
+        except Exception as e:
+            logger.error(f"? Error in model detection: {e}")
+            self.model_loaded = False
+            return True
+
+
     async def download_file(self, url: str, suffix: str = '') -> str:
         """Download file from URL to temporary location"""
         import aiohttp
@@ -343,28 +366,6 @@ class OmniAvatarAPI:
             logger.error(f"Failed to download file from {url}: {e}")
             raise HTTPException(status_code=400, detail=f"Failed to download file: {e}")
 
-
-            # Fallback: Check traditional OmniAvatar paths
-            traditional_paths = [
-                "./pretrained_models/Wan2.1-T2V-14B",
-                "./pretrained_models/OmniAvatar-14B",
-                "./pretrained_models/wav2vec2-base-960h"
-            ]
-
-            if all(os.path.exists(path) for path in traditional_paths):
-                self.model_loaded = True
-                logger.info("? SUCCESS: Traditional OmniAvatar models found")
-                return True
-
-            # No models found
-            logger.warning("?? WARNING: No models found (neither downloaded nor traditional)")
-            self.model_loaded = False
-            return True  # App can still run in TTS-only mode
-
-        except Exception as e:
-            logger.error(f"? Error in model detection: {e}")
-            self.model_loaded = False
-            return True
     def validate_audio_url(self, url: str) -> bool:
         """Validate if URL is likely an audio file"""
         try:
@@ -766,8 +767,6 @@ async def download_video_models():
             "audio_model_path": audio_model_path,
             "status": "READY FOR VIDEO GENERATION"
         }
-
-
         
     except Exception as e:
         logger.error(f"? Model download failed: {e}")
@@ -776,21 +775,6 @@ async def download_video_models():
             "message": f"Model download failed: {str(e)}",
             "error": str(e)
         }
-        
-    except Exception as e:
-        logger.error(f"? Model download failed: {e}")
-        return {
-            "success": False,
-            "message": f"Model download failed: {str(e)}",
-            "error": str(e)
-        }
-    except Exception as e:
-        logger.error(f"? Model download failed: {e}")
-        return {
-            "success": False,
-            "message": f"Model download failed: {str(e)}",
-            "error": str(e)
-        }
 
 @app.post("/reload-models")
 async def reload_models():
@@ -819,53 +803,6 @@ async def reload_models():
             "message": f"Model reload failed: {str(e)}",
             "error": str(e)
         }
-        
-        # Download small video generation model
-        logger.info("?? Downloading text-to-video model...")
-        
-        model_path = snapshot_download(
-            repo_id="ali-vilab/text-to-video-ms-1.7b",
-            cache_dir="./downloaded_models/video",
-            local_files_only=False
-        )
-        
-        logger.info(f"? Video model downloaded: {model_path}")
-        
-        # Download audio model
-        audio_model_path = snapshot_download(
-            repo_id="facebook/wav2vec2-base-960h", 
-            cache_dir="./downloaded_models/audio",
-            local_files_only=False
-        )
-        
-        logger.info(f"? Audio model downloaded: {audio_model_path}")
-        
-        # Check final storage usage
-        _, _, free_bytes_after = shutil.disk_usage(".")
-        free_gb_after = free_bytes_after / (1024**3)
-        used_gb = free_gb - free_gb_after
-        
-        return {
-            "success": True,
-            "message": "? Video generation models downloaded successfully!",
-            "models_downloaded": [
-                "ali-vilab/text-to-video-ms-1.7b",
-                "facebook/wav2vec2-base-960h"
-            ],
-            "storage_used_gb": round(used_gb, 2),
-            "storage_remaining_gb": round(free_gb_after, 2),
-            "video_model_path": model_path,
-            "audio_model_path": audio_model_path,
-            "status": "READY FOR VIDEO GENERATION"
-        }
-        
-    except Exception as e:
-        logger.error(f"? Model download failed: {e}")
-        return {
-            "success": False,
-            "message": f"Model download failed: {str(e)}",
-            "error": str(e)
-        }
 
 @app.get("/model-status")
 async def get_model_status():