app.py

import os
import numpy as np
import librosa
import tensorflow as tf
from flask import Flask, request, render_template, send_from_directory
from werkzeug.utils import secure_filename
from sklearn.preprocessing import LabelEncoder

import os
os.environ['TF_CPP_MIN_LOG_LEVEL'] = '3'           # Suppresses INFO, WARNING, and ERROR logs
os.environ['TF_ENABLE_ONEDNN_OPTS'] = '0'          # Disables oneDNN custom ops (optional)

# Initialize Flask app
app = Flask(__name__)
app.config['UPLOAD_FOLDER'] = 'uploads'

# Load model and label encoder
model = tf.keras.models.load_model("saved_models/audio_classification.keras")

encoder = LabelEncoder()
encoder.classes_ = np.load("classes.npy", allow_pickle=True)

# Feature extraction
def extract_features(file_path):
    audio, sample_rate = librosa.load(file_path, res_type='kaiser_fast')
    mfccs = librosa.feature.mfcc(y=audio, sr=sample_rate, n_mfcc=40)
    mfccs_scaled = np.mean(mfccs.T, axis=0)
    return mfccs_scaled.reshape(1, -1)

# Routes
@app.route("/", methods=["GET", "POST"])
def index():
    if request.method == "POST":
        if "file" not in request.files:
            return render_template("index.html", prediction=None, error="No file part")

        file = request.files["file"]
        if file.filename == "" or file.filename is None:
            return render_template("index.html", prediction=None, error="No selected file")

        filename = secure_filename(file.filename)
        if not filename:
            return render_template("index.html", prediction=None, error="Invalid filename")
            
        filepath = os.path.join(app.config['UPLOAD_FOLDER'], filename)
        file.save(filepath)

        try:
            features = extract_features(filepath)
            prediction = model.predict(features)
            predicted_index = np.argmax(prediction, axis=1)
            predicted_label = encoder.inverse_transform(predicted_index)[0]

            return render_template("index.html", prediction=predicted_label, audio_file=filename)

        except Exception as e:
            return render_template("index.html", prediction=None, error=str(e))

    return render_template("index.html", prediction=None)

@app.route('/uploads/<filename>')
def uploaded_file(filename):
    return send_from_directory(app.config['UPLOAD_FOLDER'], filename)

if __name__ == "__main__":
    # Create uploads directory if it doesn't exist
    os.makedirs(app.config['UPLOAD_FOLDER'], exist_ok=True)
    # Use environment variables for production
    port = int(os.environ.get('PORT', 5000))
    app.run(host='0.0.0.0', port=port)