Update app.py

app.py

@@ -9,7 +9,7 @@ from fastapi import FastAPI, Request, HTTPException, Depends, status
 from fastapi.responses import StreamingResponse
 from fastapi.security import HTTPBearer, HTTPAuthorizationCredentials
 
-# --- 1. 配置与全局状态 ---
+# --- 1. 全局配置与状态缓存 ---
 app = FastAPI()
 security = HTTPBearer()
 
@@ -17,7 +17,6 @@ ACCESS_TOKEN = os.getenv("ACCESS_TOKEN", "sk-admin-123456")
 API_KEY = "AIzaSyAZaD22Mzi9HkTcW3ErNxRA_sNEFolLBCA"
 SUPPORTED_MODELS = ["auto", "gpt-5-mini", "gemini-3-flash", "pro"]
 
-# 全局缓存：存储当前可用的 Token 及其余额
 class TokenCache:
     def __init__(self):
         self.token = ""
@@ -25,18 +24,17 @@ class TokenCache:
 
 cache = TokenCache()
 
-# --- 2. 核心逻辑函数 ---
+# --- 2. 核心后端逻辑 (带自动刷新) ---
 
-def fetch_new_account():
+def fetch_account():
     """注册新账号并初始化缓存"""
-    reg_url = f"https://identitytoolkit.googleapis.com/v1/accounts:signUp?key={API_KEY}"
     try:
-        # 注册
+        reg_url = f"https://identitytoolkit.googleapis.com/v1/accounts:signUp?key={API_KEY}"
         reg_resp = requests.post(reg_url, json={"returnSecureToken": True}, headers={"origin": "https://www.aidocmaker.com"}, timeout=10).json()
         token = reg_resp.get("idToken")
         if not token: return None
         
-        # 查询初始余额
+        # 初始同步余额
         bal_url = "https://level2labs-prod--adm-agent-helper-user-get-credits.modal.run/"
         bal_resp = requests.post(bal_url, headers={"Authorization": f"Bearer {token}"}, timeout=10).json()
         
@@ -47,130 +45,179 @@ def fetch_new_account():
         return None
 
 def get_valid_token():
-    """获取有效 Token：如果有缓存且有余额则复用，否则获取新的"""
+    """复用缓存或静默刷新"""
     if cache.token and cache.balance > 0:
         return cache.token
-    return fetch_new_account()
+    return fetch_account()
 
 def sync_balance():
-    """同步当前 Token 的实际余额"""
+    """查询并更新当前 Token 的余额"""
     if not cache.token: return 0
     url = "https://level2labs-prod--adm-agent-helper-user-get-credits.modal.run/"
     try:
-        data = requests.post(url, headers={"Authorization": f"Bearer {cache.token}"}).json()
+        data = requests.post(url, headers={"Authorization": f"Bearer {cache.token}"}, timeout=10).json()
         cache.balance = data.get("premium_quota", 0)
         return cache.balance
     except:
         cache.balance = 0
         return 0
 
-# --- 3. OpenAI 兼容接口 ---
+# --- 3. API 解析辅助函数 ---
 
-async def verify_api_token(auth: HTTPAuthorizationCredentials = Depends(security)):
+def clean_sse_line(line):
+    """解析 upstream 返回的每一行，处理可能存在的 data: 前缀"""
+    line = line.decode("utf-8").strip()
+    if not line: return None
+    # 移除 data: 前缀（如果存在）
+    if line.startswith("data: "):
+        line = line[6:]
+    if line == "[DONE]":
+        return None
+    try:
+        return json.loads(line)
+    except:
+        return None
+
+# --- 4. OpenAI 兼容接口 ---
+
+async def verify_api(auth: HTTPAuthorizationCredentials = Depends(security)):
     if auth.credentials != ACCESS_TOKEN:
-        raise HTTPException(status_code=401, detail="Invalid API Key")
+        raise HTTPException(status_code=401, detail="Unauthorized")
     return auth.credentials
 
 @app.get("/v1/models")
-async def list_models(token: str = Depends(verify_api_token)):
+async def list_models(t: str = Depends(verify_api)):
     return {"object": "list", "data": [{"id": m, "object": "model"} for m in SUPPORTED_MODELS]}
 
 @app.post("/v1/chat/completions")
-async def chat_completions(request: Request, token: str = Depends(verify_api_token)):
+async def chat_completions(request: Request, t: str = Depends(verify_api)):
     body = await request.json()
     model = body.get("model", "auto")
+    stream = body.get("stream", False)
     content = body.get("messages", [])[-1]["content"] if body.get("messages") else ""
     
-    # 自动获取/刷新 Token
     jwt = get_valid_token()
     if not jwt: raise HTTPException(status_code=500, detail="Provider Error")
 
     url = "https://level2labs-prod--adm-agent-send-chat-message.modal.run/"
-    form_data = {
+    headers = {"Authorization": f"Bearer {jwt}"}
+    data = {
         "content": (None, content), "model": (None, model),
-        "conversation_id": (None, f"conv-{uuid.uuid4()}"),
-        "message_id": (None, f"msg-{uuid.uuid4()}"), "max_mode": (None, "true")
+        "conversation_id": (None, f"c-{uuid.uuid4()}"),
+        "message_id": (None, f"m-{uuid.uuid4()}"), "max_mode": (None, "true")
     }
 
-    target_resp = requests.post(url, files=form_data, headers={"Authorization": f"Bearer {jwt}"}, stream=True)
+    target_resp = requests.post(url, files=data, headers=headers, stream=True, timeout=60)
 
     def generate():
+        cid = f"chatcmpl-{uuid.uuid4()}"
         for line in target_resp.iter_lines():
-            if line:
-                try:
-                    raw = json.loads(line.decode("utf-8"))
-                    yield f"data: {json.dumps({'choices': [{'delta': {'content': raw.get('data', '')}}]})}\n\n"
-                except: continue
+            item = clean_sse_line(line)
+            if item and "data" in item:
+                chunk = {
+                    "id": cid, "object": "chat.completion.chunk", "created": int(time.time()),
+                    "model": model, "choices": [{"index": 0, "delta": {"content": item["data"]}, "finish_reason": None}]
+                }
+                yield f"data: {json.dumps(chunk)}\n\n"
         yield "data: [DONE]\n\n"
-        # 响应结束后异步同步一下余额（简单处理，实际可根据响应次数扣减）
         sync_balance()
 
-    return StreamingResponse(generate(), media_type="text/event-stream")
-
-# --- 4. Gradio UI (带门禁验证) ---
-
-with gr.Blocks(title="AI Doc Maker 管理系统") as demo:
-    # 登录界面
-    with gr.Column(visible=True) as login_panel:
-        gr.Markdown("## 🔐 访问受限\n请输入系统的 ACCESS_TOKEN 以解锁管理面板。")
-        pwd_input = gr.Textbox(label="ACCESS_TOKEN", type="password", placeholder="输入 sk-xxxx...")
-        login_btn = gr.Button("解锁系统", variant="primary")
-        login_msg = gr.Markdown()
-
-    # 主管理界面 (初始隐藏)
-    with gr.Column(visible=False) as main_panel:
-        gr.Markdown("# 🚀 AI Doc Maker 后台管理")
+    if stream:
+        return StreamingResponse(generate(), media_type="text/event-stream")
+    else:
+        full_text = ""
+        for line in target_resp.iter_lines():
+            item = clean_sse_line(line)
+            if item and "data" in item:
+                full_text += item["data"]
+        sync_balance()
+        return {
+            "id": f"chatcmpl-{uuid.uuid4()}", "object": "chat.completion", "created": int(time.time()),
+            "model": model, "choices": [{"index": 0, "message": {"role": "assistant", "content": full_text}, "finish_reason": "stop"}]
+        }
+
+# --- 5. Gradio UI 管理面板 ---
+
+with gr.Blocks(title="AI Doc Maker 管理面板") as demo:
+    # 状态存储
+    current_jwt_state = gr.State("")
+
+    # 登录门禁
+    with gr.Column(visible=True) as login_view:
+        gr.Markdown("## 🔐 访问受限\n请输入 `ACCESS_TOKEN` 以解锁管理面板。")
+        pwd = gr.Textbox(label="Token", type="password")
+        login_btn = gr.Button("验证并进入", variant="primary")
+        err_msg = gr.Markdown()
+
+    # 主面板
+    with gr.Column(visible=False) as main_view:
+        gr.Markdown("# 🚀 AI Doc Maker 管理面板")
         
-        with gr.Row():
-            with gr.Column():
-                token_view = gr.Textbox(label="当前活跃 Token (缓存中)", interactive=False)
-                balance_view = gr.Textbox(label="剩余可用额度", interactive=False)
-                refresh_btn = gr.Button("强制刷新账号/同步余额")
+        with gr.Tab("余额与账号"):
+            with gr.Row():
+                ui_token = gr.Textbox(label="当前活跃 Token", interactive=False)
+                ui_balance = gr.Textbox(label="剩余可用额度", interactive=False)
+            ui_refresh = gr.Button("手动刷新/注册新账号")
+
+        with gr.Tab("对话与语音"):
+            ui_input = gr.Textbox(label="输入内容", lines=3, placeholder="输入文字...")
+            with gr.Row():
+                ui_chat_btn = gr.Button("发送消息", variant="primary")
+                ui_tts_btn = gr.Button("✨ 生成语音", variant="secondary")
             
-            with gr.Column():
-                chat_input = gr.Textbox(label="测试对话", placeholder="输入消息...")
-                chat_btn = gr.Button("发送测试", variant="primary")
-                chat_output = gr.Textbox(label="AI 回复", interactive=False)
-
-    # --- UI 逻辑绑定 ---
-    def handle_login(entered_token):
-        if entered_token == ACCESS_TOKEN:
-            # 获取当前可用信息
-            t = get_valid_token()
-            b = sync_balance()
-            return gr.update(visible=False), gr.update(visible=True), t, b, ""
-        else:
-            return gr.update(visible=True), gr.update(visible=False), "", "", "❌ 密码错误，请重试！"
-
-    login_btn.click(
-        handle_login, 
-        inputs=pwd_input, 
-        outputs=[login_panel, main_panel, token_view, balance_view, login_msg]
-    )
-
-    def manual_refresh():
-        # 强制获取新账号
-        t = fetch_new_account()
-        b = sync_balance()
-        return t, b
-
-    refresh_btn.click(manual_refresh, outputs=[token_view, balance_view])
-
-    def ui_chat_test(msg):
-        jwt = get_valid_token()
+            ui_reply = gr.Textbox(label="AI 回复", interactive=False)
+            with gr.Row():
+                ui_raw_json = gr.Textbox(label="原始 JSON (聊天)", interactive=False)
+                ui_audio_json = gr.Textbox(label="音频 JSON", interactive=False)
+            ui_audio_play = gr.Audio(label="播放器", interactive=False)
+
+    # --- 逻辑处理 ---
+    def on_login(token):
+        if token == ACCESS_TOKEN:
+            jwt = get_valid_token()
+            bal = sync_balance()
+            return gr.update(visible=False), gr.update(visible=True), jwt, bal, jwt, ""
+        return gr.update(visible=True), gr.update(visible=False), "", "", "", "❌ 验证失败"
+
+    login_btn.click(on_login, inputs=pwd, outputs=[login_view, main_view, ui_token, ui_balance, current_jwt_state, err_msg])
+
+    def on_refresh():
+        jwt = fetch_account()
+        bal = sync_balance()
+        return jwt, bal, jwt
+
+    ui_refresh.click(on_refresh, outputs=[ui_token, ui_balance, current_jwt_state])
+
+    def on_ui_chat(jwt, text):
+        if not jwt: yield "请先登录", ""; return
         url = "https://level2labs-prod--adm-agent-send-chat-message.modal.run/"
-        data = {"content": (None, msg), "model": (None, "auto"), "conversation_id": (None, "ui-test"), "message_id": (None, "ui-test"), "max_mode": (None, "true")}
+        data = {"content": (None, text), "model": (None, "auto"), "conversation_id": (None, "ui"), "message_id": (None, "ui"), "max_mode": (None, "true")}
         full_text = ""
         resp = requests.post(url, files=data, headers={"Authorization": f"Bearer {jwt}"}, stream=True)
         for line in resp.iter_lines():
-            if line:
-                full_text += json.loads(line.decode("utf-8")).get("data", "")
-                yield full_text
-        sync_balance() # 更新余额显示
+            item = clean_sse_line(line)
+            if item and "data" in item:
+                full_text += item["data"]
+                yield full_text, json.dumps(item, ensure_ascii=False)
+        sync_balance()
+
+    ui_chat_btn.click(on_ui_chat, inputs=[current_jwt_state, ui_input], outputs=[ui_reply, ui_raw_json])
+
+    def on_ui_tts(jwt, text):
+        if not jwt: return "请先登录", None
+        try:
+            url_create = "https://level2labs-prod--adm-agent-audio-create-audio-with-assistant.modal.run/"
+            headers = {"Authorization": f"Bearer {jwt}"}
+            res = requests.post(url_create, files={"prompt": (None, text), "voice_id": (None, "clear"), "speed": (None, "1")}, headers=headers).json()
+            name = res.get("name")
+            url_get = f"https://level2labs-prod--adm-agent-audio-get-audio-playback-url.modal.run/?name={name}"
+            audio_url = requests.get(url_get, headers=headers).json().get("url")
+            return json.dumps(res, indent=2), audio_url
+        except Exception as e:
+            return f"生成失败: {e}", None
 
-    chat_btn.click(ui_chat_test, inputs=chat_input, outputs=chat_output)
+    ui_tts_btn.click(on_ui_tts, inputs=[current_jwt_state, ui_input], outputs=[ui_audio_json, ui_audio_play])
 
-# 挂载
 app = gr.mount_gradio_app(app, demo, path="/")
 
 if __name__ == "__main__":