Upload 4 files

app.py

@@ -8,6 +8,7 @@ import func
 from apscheduler.schedulers.background import BackgroundScheduler
 import requests
 import time
+from flask_httpauth import HTTPBasicAuth
 
 os.environ['TZ'] = 'Asia/Shanghai'
 app = Flask(__name__)
@@ -24,6 +25,12 @@ handler.setFormatter(formatter)
 
 logger.addHandler(handler) 
 
+auth = HTTPBasicAuth()
+
+@auth.verify_password
+def verify_password(username, password):
+    return password == os.environ.get('password')
+
 safety_settings = [
     {
         "category": "HARM_CATEGORY_HARASSMENT",
@@ -111,12 +118,11 @@ function copyLink(event) {
 </html>
     """
     return render_template_string(html_template, main_content=main_content)
+
 @app.route('/hf/v1/chat/completions', methods=['POST'])
+@auth.login_required
 def chat_completions():
     global current_api_key
-    is_authenticated, auth_error, status_code = func.authenticate_request(request)
-    if not is_authenticated:
-        return auth_error if auth_error else jsonify({'error': '未授权'}), status_code if status_code else 401
     try:
         request_data = request.get_json()
         messages = request_data.get('messages', [])
@@ -244,11 +250,11 @@ def chat_completions():
             }
         }), 500
 
+
+
 @app.route('/hf/v1/models', methods=['GET'])
+@auth.login_required
 def list_models():
-    is_authenticated, auth_error, status_code = func.authenticate_request(request)
-    if not is_authenticated:
-        return auth_error if auth_error else jsonify({'error': '未授权'}), status_code if status_code else 401
     response = {"object": "list", "data": GEMINI_MODELS}
     return jsonify(response)
 

func.py

@@ -8,26 +8,6 @@ import re
 import os
 logger = logging.getLogger(__name__)
 
-password = os.environ.get('password')
-
-def authenticate_request(request):
-    auth_header = request.headers.get('Authorization')
-
-    if not auth_header:
-        return False, jsonify({'error': '缺少Authorization请求头'}), 401
-    try:
-        auth_type, pass_word = auth_header.split(' ', 1)
-    except ValueError:
-        return False, jsonify({'error': 'Authorization请求头格式错误'}), 401
-
-    if auth_type.lower() != 'bearer':
-        return False, jsonify({'error': 'Authorization类型必须为Bearer'}), 401
-
-    if pass_word != password:
-        return False, jsonify({'error': '未授权'}), 401
-
-    return True, None, None
-
 def process_messages_for_gemini(messages):
     
     gemini_history = []

requirements.txt

@@ -5,4 +5,5 @@ Werkzeug==2.0.3
 google==3.0.0
 google-generativeai==0.8.3
 pillow==10.4.0
-apscheduler
+apscheduler
+Flask-HTTPAuth==4.8.0