|
|
from flask import Flask, request, jsonify |
|
|
from api import UrduWhisper |
|
|
import uuid |
|
|
import os |
|
|
|
|
|
app = Flask(__name__) |
|
|
model = UrduWhisper() |
|
|
|
|
|
UPLOAD = "uploads" |
|
|
os.makedirs(UPLOAD, exist_ok=True) |
|
|
|
|
|
@app.route("/transcribe", methods=["POST"]) |
|
|
def transcribe_audio(): |
|
|
if "file" not in request.files: |
|
|
return jsonify({"error": "No file uploaded"}), 400 |
|
|
|
|
|
file = request.files["file"] |
|
|
filename = f"{UPLOAD}/{uuid.uuid4()}.wav" |
|
|
file.save(filename) |
|
|
|
|
|
text = model.transcribe(filename) |
|
|
os.remove(filename) |
|
|
|
|
|
return jsonify({"text": text}) |
|
|
|
|
|
@app.route("/") |
|
|
def home(): |
|
|
return {"message": "Custom Urdu Whisper API Running!"} |
|
|
|
|
|
if __name__ == "__main__": |
|
|
app.run(host="0.0.0.0", port=5000) |
