Create app.py

app.py

@@ -0,0 +1,57 @@
+from flask import Flask, request, Response, stream_with_context, jsonify
+from kokoro_onnx import Kokoro
+import soundfile as sf
+import io
+
+app = Flask(__name__)
+
+print("--- STARTUP ---")
+# Because we set the ENV variable in Dockerfile, we don't need arguments here!
+print("Loading Kokoro...")
+kokoro = Kokoro("kokoro-v1.0.int8.onnx", "voices-v1.0.bin")
+print("SUCCESS: Model Loaded!")
+
+@app.route("/", methods=["GET"])
+def home():
+    return "Kokoro TTS API is Online!"
+
+# Universal Route (Accepts /v1 or no /v1)
+@app.route("/audio/speech", methods=["POST"])
+@app.route("/v1/audio/speech", methods=["POST"])
+def generate_speech():
+    try:
+        data = request.json
+        text = data.get("input", "")
+        voice = data.get("voice", "af_bella")
+        speed = float(data.get("speed", 1.0))
+
+        def generate_stream():
+            samples, sample_rate = kokoro.create(
+                text, 
+                voice=voice, 
+                speed=speed, 
+                lang="en-us"
+            )
+            buffer = io.BytesIO()
+            sf.write(buffer, samples, sample_rate, format='WAV')
+            buffer.seek(0)
+            yield buffer.read()
+
+        return Response(stream_with_context(generate_stream()), mimetype="audio/wav")
+        
+    except Exception as e:
+        print(f"Error: {e}")
+        return jsonify({"error": str(e)}), 500
+
+@app.route("/v1/models", methods=["GET"])
+@app.route("/models", methods=["GET"])
+def list_models():
+    return jsonify({"data": [{"id": "kokoro", "object": "model"}]})
+
+@app.route("/v1/audio/voices", methods=["GET"])
+@app.route("/audio/voices", methods=["GET"])
+def list_voices():
+    return jsonify({"voices": list(kokoro.get_voices())})
+
+if __name__ == "__main__":
+    app.run(host="0.0.0.0", port=7860)