Spaces:

Moonfanz
/

gemini-rproxy

Running

App Files Files Community

Moonfanz commited on Dec 19, 2024

Commit

7910d5f

verified ·

1 Parent(s): ae34437

Upload 4 files

Browse files

Files changed (2) hide show

app.py +4 -8
func.py +8 -6

app.py CHANGED Viewed

@@ -14,8 +14,6 @@ app = Flask(__name__)
 app.secret_key = os.urandom(24)
-PASSWORD = os.environ['password']
 formatter = logging.Formatter('%(message)s')
 logger = logging.getLogger(__name__)
@@ -76,7 +74,7 @@ GEMINI_MODELS = [
 ]
 @app.route('/')
 def index():
-    main_content = "Moonfanz Gemini" # 替换成您的实际动态内容
     html_template = """
 <!DOCTYPE html>
 <html>
@@ -116,7 +114,7 @@ function copyLink(event) {
 @app.route('/hf/v1/chat/completions', methods=['POST'])
 def chat_completions():
     global current_api_key
-    is_authenticated, auth_error, status_code = func.authenticate_request(PASSWORD, request)
     if not is_authenticated:
         return auth_error if auth_error else jsonify({'error': 'Unauthorized'}), status_code if status_code else 401
     try:
@@ -248,14 +246,13 @@ def chat_completions():
 @app.route('/hf/v1/models', methods=['GET'])
 def list_models():
-    is_authenticated, auth_error, status_code = func.authenticate_request(PASSWORD, request)
     if not is_authenticated:
         return auth_error if auth_error else jsonify({'error': 'Unauthorized'}), status_code if status_code else 401
     response = {"object": "list", "data": GEMINI_MODELS}
     return jsonify(response)
 def keep_alive():
-    """ 定期向应用自身发送请求，保持活跃 """
     try:
         response = requests.get("http://127.0.0.1:7860/", timeout=10)
         response.raise_for_status()
@@ -265,8 +262,7 @@ def keep_alive():
 if __name__ == '__main__':
     scheduler = BackgroundScheduler()
-    # 设置定时任务，每 10 分钟执行一次 keep_alive 函数
     scheduler.add_job(keep_alive, 'interval', hours = 12)
-    # 启动调度器
     scheduler.start()
     app.run(debug=True, host='0.0.0.0', port=int(os.environ.get('PORT', 7860)))

 app.secret_key = os.urandom(24)
 formatter = logging.Formatter('%(message)s')
 logger = logging.getLogger(__name__)
 ]
 @app.route('/')
 def index():
+    main_content = "Moonfanz Gemini"
     html_template = """
 <!DOCTYPE html>
 <html>
 @app.route('/hf/v1/chat/completions', methods=['POST'])
 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': 'Unauthorized'}), status_code if status_code else 401
     try:
 @app.route('/hf/v1/models', methods=['GET'])
 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': 'Unauthorized'}), status_code if status_code else 401
     response = {"object": "list", "data": GEMINI_MODELS}
     return jsonify(response)
 def keep_alive():
     try:
         response = requests.get("http://127.0.0.1:7860/", timeout=10)
         response.raise_for_status()
 if __name__ == '__main__':
     scheduler = BackgroundScheduler()
     scheduler.add_job(keep_alive, 'interval', hours = 12)
     scheduler.start()
     app.run(debug=True, host='0.0.0.0', port=int(os.environ.get('PORT', 7860)))

func.py CHANGED Viewed

@@ -5,25 +5,27 @@ from flask import jsonify
 import logging
 import json
 import re
 logger = logging.getLogger(__name__)
-def authenticate_request(password, request):
     auth_header = request.headers.get('Authorization')
     if not auth_header:
-        return False, jsonify({'error': 'Authorization header is missing'}), 401
     try:
         auth_type, pass_word = auth_header.split(' ', 1)
     except ValueError:
-        return False, jsonify({'error': 'Invalid Authorization header format'}), 401
     if auth_type.lower() != 'bearer':
-        return False, jsonify({'error': 'Authorization type must be Bearer'}), 401
     if pass_word != password:
-        return False, jsonify({'error': 'Unauthorized'}), 401
     return True, None, None

 import logging
 import json
 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