Add auth login

app.py

@@ -1,20 +1,43 @@
+import os.path
+
 from flask import Flask
 from flask_apscheduler import APScheduler
+from flask_login import LoginManager
+from flask_wtf import CSRFProtect
 
-from tts_app import frontend, voice_api
-
-app = Flask(__name__)
+from tts_app import frontend, voice_api, auth, admin
+from tts_app.auth.models import users
 
-app.register_blueprint(frontend, url_prefix='/')
-app.register_blueprint(voice_api, url_prefix='/voice')
+app = Flask(__name__, template_folder=os.path.join(os.path.dirname(__file__), 'tts_app', 'templates'),
+            static_folder=os.path.join(os.path.dirname(__file__), 'tts_app', 'static'))
 
 app.config.from_pyfile("config.py")
 
+login_manager = LoginManager()
+login_manager.init_app(app)
+login_manager.login_view = 'auth.login'
+
+csrf = CSRFProtect(app)
+
+
+@login_manager.user_loader
+def load_user(user_id):
+    for user in users.values():
+        if user.get_id() == user_id:
+            return user
+    return None
+
+
 # Initialize scheduler
 scheduler = APScheduler()
 scheduler.init_app(app)
 if app.config.get("CLEAN_INTERVAL_SECONDS", 3600) > 0:
     scheduler.start()
 
+app.register_blueprint(frontend, url_prefix='/')
+app.register_blueprint(voice_api, url_prefix='/voice')
+app.register_blueprint(auth, url_prefix='/auth')
+app.register_blueprint(admin, url_prefix='/admin')
+
 if __name__ == '__main__':
     app.run(host='0.0.0.0', port=app.config.get("PORT", 23456), debug=app.config.get("DEBUG", False))

requirements.txt

@@ -40,6 +40,8 @@ flask==2.2.3
 flask_apscheduler
 pydantic==2.3.0
 werkzeug==2.3.6
+flask_login
+flask_wtf
 
 # Compression and Decompression
 py7zr

tts_app/__init__.py

@@ -1,2 +1,4 @@
 from tts_app.frontend import frontend
 from tts_app.voice_api import voice_api
+from tts_app.auth import auth
+from tts_app.admin import admin

tts_app/admin/__init__.py

@@ -0,0 +1 @@
+from tts_app.admin.views import admin

tts_app/admin/views.py

@@ -0,0 +1,9 @@
+from flask import Blueprint
+from flask_login import login_required
+
+admin = Blueprint('admin', __name__)
+
+@admin.route('/')
+@login_required
+def setting():
+    return "Hello Admin!"

tts_app/auth/__init__.py

@@ -0,0 +1 @@
+from tts_app.auth.views import auth

tts_app/auth/forms.py

@@ -0,0 +1,8 @@
+from flask_wtf import FlaskForm
+from wtforms import StringField, PasswordField, SubmitField
+from wtforms.validators import DataRequired, Length
+
+class LoginForm(FlaskForm):
+    username = StringField('Username', validators=[DataRequired(), Length(1, 64)])
+    password = PasswordField('Password', validators=[DataRequired()])
+    submit = SubmitField('Login')

tts_app/auth/models.py

@@ -0,0 +1,19 @@
+class User:
+    def __init__(self, id, username, password):
+        self.id = id
+        self.username = username
+        self.password = password
+
+    def is_authenticated(self):
+        return True
+
+    def is_active(self):
+        return True
+
+    def is_anonymous(self):
+        return False
+
+    def get_id(self):
+        return str(self.id)
+
+users = {'admin': User(1, 'admin', 'password')}

tts_app/auth/views.py

@@ -0,0 +1,29 @@
+from flask import render_template, redirect, url_for, flash, Blueprint
+from flask_login import login_user, logout_user, login_required
+
+from tts_app.auth.forms import LoginForm
+from tts_app.auth.models import users
+
+auth = Blueprint('auth', __name__)
+
+
+@auth.route('/login', methods=['GET', 'POST'])
+def login():
+    form = LoginForm()
+    if form.validate_on_submit():
+        user = users.get(form.username.data)
+        if user and user.password == form.password.data:
+            login_user(user)
+            flash('Logged in successfully.')
+            return redirect(url_for('admin.setting'))
+        flash('Wrong username or password.')
+    return render_template('login.html', form=form)
+
+@auth.route('/logout')
+@login_required
+def logout():
+    logout_user()
+    flash('You have been logged out.')
+    return redirect(url_for('auth.login'))
+
+

tts_app/static/css/login.css

@@ -0,0 +1,33 @@
+body {
+    font-family: Arial, sans-serif;
+    margin: 50px;
+}
+
+.container {
+    max-width: 300px;
+    margin: 0 auto;
+    padding: 20px;
+    border: 1px solid #ccc;
+    border-radius: 5px;
+}
+
+label, input {
+    width: 100%;
+    padding: 8px;
+    margin-bottom: 10px;
+    box-sizing: border-box;
+}
+
+button {
+    width: 100%;
+    padding: 10px;
+    background-color: #007BFF;
+    color: #fff;
+    border: none;
+    border-radius: 3px;
+    cursor: pointer;
+}
+
+button:hover {
+    background-color: #0056b3;
+}

{templates → tts_app/templates}/index.html

@@ -4,8 +4,8 @@
     <meta charset="UTF-8"/>
     <meta name="viewport" content="width=device-width, initial-scale=1.0"/>
     <title>vits-simple-api</title>
-    <link rel="stylesheet" href="/static/css/style.css">
-    <link rel="stylesheet" href="/static/css/bootstrap.min.css"/>
+    <link rel="stylesheet" href="{{ url_for('static', filename='css/style.css') }}">
+    <link rel="stylesheet" href="{{ url_for('static', filename='css/bootstrap.min.css') }}">
 </head>
 <body>
 <main class="main-container">
@@ -323,8 +323,8 @@
 
 </main>
 
-<script src="/static/js/jquery.slim.min.js"></script>
-<script src="/static/js/bootstrap.bundle.min.js"></script>
+<script src="{{ url_for('static', filename='js/jquery.slim.min.js') }}"></script>
+<script src="{{ url_for('static', filename='js/bootstrap.bundle.min.js') }}"></script>
 
 <script>
     $(function () {

tts_app/templates/login.html

@@ -0,0 +1,25 @@
+<!DOCTYPE html>
+<html lang="en">
+<head>
+    <meta charset="UTF-8">
+    <meta name="viewport" content="width=device-width, initial-scale=1.0">
+    <title>Login</title>
+    <link rel="stylesheet" href="{{ url_for('static', filename='css/login.css') }}">
+    <link rel="stylesheet" href="{{ url_for('static', filename='css/bootstrap.min.css') }}">
+</head>
+<body>
+    <div class="container">
+        <h2>Login</h2>
+        <form action="{{ url_for('auth.login') }}" method="post">
+             {{ form.hidden_tag() }}
+            <label for="username">Username:</label>
+            <input type="text" id="username" name="username" required>
+
+            <label for="password">Password:</label>
+            <input type="password" id="password" name="password" required>
+
+            <button type="submit">Login</button>
+        </form>
+    </div>
+</body>
+</html>

tts_app/voice_api/views.py

@@ -7,7 +7,7 @@ from werkzeug.utils import secure_filename
 
 from logger import logger
 from contants import ModelType
-from tts_app.auth import require_api_key
+from tts_app.voice_api.auth import require_api_key
 from tts_app.model_manager import model_manager, tts_manager
 from tts_app.voice_api.utils import *
 from utils.data_utils import check_is_none