from io import BytesIO

from flask import Flask, request, jsonify, render_template, send_file, Response
from bs4 import BeautifulSoup
from scraper import voicevox
import requests

app = Flask(__name__)

@app.route('/')
def index():
    return '<h1>OK</h1>'

@app.route("/voicevox")
def voicevox_resp():
    text = request.args.get("text")
    speaker = request.args.get("speaker")
    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)

        return send_file(io, as_attachment=True, mimetype="audio/mp3", download_name=f'rull-pyward_voicevox({text.strip()}).mp3')
    except Exception as e:
        return jsonify({'error': "Internal server error"}), 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,
          #  headers=headers,
          #  cookies=cookies,
          #  data=data,
            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)  # Use 'adhoc' to enable HTTPS locally