talksik
74 lines
2.0 KiB
Python
74 lines
2.0 KiB
Python
import whisper
|
|
import tts
|
|
from tempfile import NamedTemporaryFile
|
|
from flask import Flask, request, abort, send_file
|
|
from dotenv import load_dotenv
|
|
load_dotenv()
|
|
|
|
|
|
# Load the Whisper model:
|
|
model = whisper.load_model("base")
|
|
|
|
app = Flask(__name__)
|
|
|
|
|
|
@app.route("/")
|
|
def helloWorld():
|
|
return "Hello, World!"
|
|
|
|
|
|
@app.route("/transcribe", methods=["POST"])
|
|
def transcribe():
|
|
if not request.files:
|
|
# If the user didn't submit any files, return a 400 (Bad Request) error.
|
|
abort(400, "No files submitted!")
|
|
|
|
# For each file, let's store the results in a list of dictionaries.
|
|
results = []
|
|
|
|
# Loop over every file that the user submitted.
|
|
for filename, handle in request.files.items():
|
|
# Create a temporary file.
|
|
# The location of the temporary file is available in `temp.name`.
|
|
temp = NamedTemporaryFile()
|
|
# Write the user's uploaded file to the temporary file.
|
|
# The file will get deleted when it drops out of scope.
|
|
handle.save(temp)
|
|
# Let's get the transcript of the temporary file.
|
|
result = model.transcribe(temp.name)
|
|
# Now we can store the result object for this file.
|
|
results.append(
|
|
{
|
|
"filename": filename,
|
|
"transcript": result["text"],
|
|
}
|
|
)
|
|
|
|
# This will be automatically converted to JSON.
|
|
return {"results": results}
|
|
|
|
|
|
# endpoint to convert text to speech and return the file
|
|
@app.route("/tts", methods=["POST"])
|
|
def text_to_speech():
|
|
# get the text from the request json
|
|
content = request.get_json()
|
|
text = content['text']
|
|
|
|
if not text:
|
|
abort(400, "Please provide text to convert to speech!")
|
|
|
|
# call the text_to_wav function from tts.py
|
|
fileName = tts.text_to_wav(
|
|
voice_name="en-GB-Neural2-B",
|
|
text=text,
|
|
)
|
|
|
|
# return the file
|
|
return send_file(f"{fileName}", mimetype="audio/wav")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
# 0.0.0.0 opens up to all ipv4 addresses
|
|
app.run(host="0.0.0.0", port=5001, debug=True)
|