Update app/main.py
Browse files- app/main.py +11 -2
app/main.py
CHANGED
|
@@ -10,6 +10,8 @@ from datetime import datetime
|
|
| 10 |
from io import BytesIO
|
| 11 |
from urllib.parse import quote_plus
|
| 12 |
from jinja2 import Template
|
|
|
|
|
|
|
| 13 |
from openai import OpenAI
|
| 14 |
|
| 15 |
# —— 环境变量读取 —— #
|
|
@@ -17,8 +19,15 @@ BAIDU_API_KEY = os.getenv("BAIDU_API_KEY", "")
|
|
| 17 |
BAIDU_SECRET_KEY = os.getenv("BAIDU_SECRET_KEY", "")
|
| 18 |
OPENAI_API_KEY = os.getenv("OPENAI_API_KEY", "")
|
| 19 |
|
| 20 |
-
# ——
|
| 21 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 22 |
|
| 23 |
# —— Baidu OCR —— #
|
| 24 |
def get_access_token(api_key, secret_key):
|
|
|
|
| 10 |
from io import BytesIO
|
| 11 |
from urllib.parse import quote_plus
|
| 12 |
from jinja2 import Template
|
| 13 |
+
|
| 14 |
+
import openai
|
| 15 |
from openai import OpenAI
|
| 16 |
|
| 17 |
# —— 环境变量读取 —— #
|
|
|
|
| 19 |
BAIDU_SECRET_KEY = os.getenv("BAIDU_SECRET_KEY", "")
|
| 20 |
OPENAI_API_KEY = os.getenv("OPENAI_API_KEY", "")
|
| 21 |
|
| 22 |
+
# —— 指定代理站 Base URL —— #
|
| 23 |
+
# 设置全局 module 级别(旧接口兼容)
|
| 24 |
+
openai.api_base = "https://free.v36.cm/v1/"
|
| 25 |
+
|
| 26 |
+
# 初始化 v1 客户端,并显式传入 api_base
|
| 27 |
+
client = OpenAI(
|
| 28 |
+
api_key=OPENAI_API_KEY,
|
| 29 |
+
api_base="https://free.v36.cm/v1/"
|
| 30 |
+
)
|
| 31 |
|
| 32 |
# —— Baidu OCR —— #
|
| 33 |
def get_access_token(api_key, secret_key):
|