|
|
from io import BytesIO |
|
|
|
|
|
from flask import Flask, request, jsonify, render_template, send_file, Response |
|
|
from bs4 import BeautifulSoup |
|
|
from scraper import voicevox |
|
|
from utils import mp3_to_opus |
|
|
import requests |
|
|
|
|
|
app = Flask(__name__) |
|
|
|
|
|
def isTrue(thing): |
|
|
return str(thing).lower() == "true" |
|
|
|
|
|
@app.route('/') |
|
|
def index(): |
|
|
return '<h1>OK</h1>' |
|
|
|
|
|
@app.route("/voicevox") |
|
|
def voicevox_resp(): |
|
|
text = request.args.get("text") |
|
|
speaker = request.args.get("speaker") |
|
|
opus = isTrue(request.args.get("opus")) |
|
|
if text is None: |
|
|
return jsonify({'error': "Please provided parameter 'text'"}), 400 |
|
|
if speaker is None: |
|
|
speaker = 1 |
|
|
|
|
|
|
|
|
|
|
|
try: |
|
|
data = voicevox(text, speaker=speaker) |
|
|
io = BytesIO(data) |
|
|
if opus: |
|
|
io = mp3_to_opus(io) |
|
|
mime = "audio/ogg" if opus else "audio/mpeg" |
|
|
ext = ".opus" if opus else ".mp3" |
|
|
return send_file(io, as_attachment=True, mimetype=mime, download_name=f'rull-pyward_voicevox({text.strip()}){ext}') |
|
|
except Exception as e: |
|
|
print(e) |
|
|
return jsonify({'error': str(e) }), 500 |
|
|
|
|
|
@app.route('/proxy') |
|
|
def proxy(): |
|
|
url = request.args.get('url') |
|
|
if not url: |
|
|
return jsonify({'error': 'URL parameter is missing'}), 400 |
|
|
|
|
|
try: |
|
|
headers = {key: value for (key, value) in request.headers if key != 'Host'} |
|
|
cookies = request.cookies |
|
|
method = request.method |
|
|
data = request.get_data() |
|
|
|
|
|
response = requests.request( |
|
|
method=method, |
|
|
url=url, |
|
|
|
|
|
|
|
|
|
|
|
allow_redirects=False |
|
|
) |
|
|
|
|
|
print(response.headers.items()) |
|
|
if response.headers.get('content-type', '').startswith('text/html'): |
|
|
|
|
|
soup = BeautifulSoup(response.text, 'html.parser') |
|
|
for tag in soup.find_all(href=True): |
|
|
tag['href'] = '/proxy?url=' + tag['href'] |
|
|
for tag in soup.find_all(src=True): |
|
|
tag['src'] = '/proxy?url=' + tag['src'] |
|
|
content = soup.prettify(formatter="html") |
|
|
return content, response.status_code |
|
|
else: |
|
|
return Response(response.content, status=response.status_code) |
|
|
except requests.RequestException as e: |
|
|
return jsonify({'error': str(e)}), 500 |
|
|
|
|
|
if __name__ == '__main__': |
|
|
app.run(host="0.0.0.0", port=7860) |
|
|
|
