Update app.py

app.py

@@ -17,22 +17,45 @@ import time
 from typing import Optional, Annotated, List
 import uvicorn
 import asyncio
+from contextlib import asynccontextmanager
+import socket
 
-# Initialize FastAPI app
+# Global model variable
+model = None
+
+@asynccontextmanager
+async def lifespan(app: FastAPI):
+    """Application lifespan management"""
+    # Startup
+    global model
+    try:
+        from sonics import HFAudioClassifier
+        print("🔄 Loading Madverse Music AI model...")
+        model = HFAudioClassifier.from_pretrained("awsaf49/sonics-spectttra-alpha-120s")
+        model.eval()
+        print("✅ Model loaded successfully!")
+    except Exception as e:
+        print(f"❌ Failed to load model: {e}")
+        raise
+    
+    yield
+    
+    # Shutdown
+    print("🔄 Shutting down...")
+
+# Initialize FastAPI app with lifespan
 app = FastAPI(
     title="Madverse Music API",
     description="AI-powered music detection API to identify AI-generated vs human-created music",
     version="1.0.0",
     docs_url="/",
-    redoc_url="/docs"
+    redoc_url="/docs",
+    lifespan=lifespan
 )
 
 # API Key Configuration
 API_KEY = os.getenv("MADVERSE_API_KEY", "madverse-music-api-key-2024")  # Default key for demo
 
-# Global model variable
-model = None
-
 async def verify_api_key(x_api_key: Annotated[str | None, Header()] = None):
     """Verify API key from header"""
     if x_api_key is None:
@@ -90,20 +113,6 @@ class ErrorResponse(BaseModel):
     error: str
     message: str
 
-@app.on_event("startup")
-async def load_model():
-    """Load the AI model on startup"""
-    global model
-    try:
-        from sonics import HFAudioClassifier
-        print("🔄 Loading Madverse Music AI model...")
-        model = HFAudioClassifier.from_pretrained("awsaf49/sonics-spectttra-alpha-120s")
-        model.eval()
-        print("✅ Model loaded successfully!")
-    except Exception as e:
-        print(f"❌ Failed to load model: {e}")
-        raise
-
 def cleanup_file(file_path: str):
     """Background task to cleanup temporary files"""
     try:
@@ -381,5 +390,31 @@ async def get_info():
         }
     }
 
+def find_available_port(start_port: int = 8000, max_attempts: int = 10) -> int:
+    """Find an available port starting from start_port"""
+    for port in range(start_port, start_port + max_attempts):
+        try:
+            # Try to bind to the port
+            with socket.socket(socket.AF_INET, socket.SOCK_STREAM) as s:
+                s.bind(('0.0.0.0', port))
+                return port
+        except OSError:
+            continue
+    
+    # If no port found, raise an exception
+    raise RuntimeError(f"No available port found in range {start_port}-{start_port + max_attempts - 1}")
+
 if __name__ == "__main__":
-    uvicorn.run(app, host="0.0.0.0", port=8000) 
+    try:
+        # Find an available port
+        port = find_available_port(8000)
+        print(f"🚀 Starting server on port {port}")
+        uvicorn.run(app, host="0.0.0.0", port=port)
+    except RuntimeError as e:
+        print(f"❌ {e}")
+        print("💡 You can also manually specify a different port by running:")
+        print("   uvicorn app:app --host 0.0.0.0 --port YOUR_PORT_NUMBER")
+    except KeyboardInterrupt:
+        print("\n🛑 Server stopped by user")
+    except Exception as e:
+        print(f"❌ Failed to start server: {e}")