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()
 
@@ -24,17 +24,17 @@ class TokenCache:
 
 cache = TokenCache()
 
-# --- 2. 核心后端逻辑 (带自动刷新) ---
+# --- 2. 核心后端逻辑 ---
 
 def fetch_account():
-    """注册新账号并初始化缓存"""
+    """注册新账号"""
     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()
         
@@ -45,13 +45,11 @@ def fetch_account():
         return None
 
 def get_valid_token():
-    """复用缓存或静默刷新"""
     if cache.token and cache.balance > 0:
         return cache.token
     return fetch_account()
 
 def sync_balance():
-    """查询并更新当前 Token 的余额"""
     if not cache.token: return 0
     url = "https://level2labs-prod--adm-agent-helper-user-get-credits.modal.run/"
     try:
@@ -62,19 +60,25 @@ def sync_balance():
         cache.balance = 0
         return 0
 
-# --- 3. API 解析辅助函数 ---
+# --- 3. 稳健的 SSE 解析器 ---
 
-def clean_sse_line(line):
-    """解析 upstream 返回的每一行，处理可能存在的 data: 前缀"""
-    line = line.decode("utf-8").strip()
+def parse_upstream_line(line):
+    """解析 upstream 返回的每一行，兼容 data: 前缀和纯 JSON"""
     if not line: return None
-    # 移除 data: 前缀（如果存在）
-    if line.startswith("data: "):
-        line = line[6:]
-    if line == "[DONE]":
-        return None
+    decoded_line = line.decode("utf-8").strip()
+    if not decoded_line: return None
+    
+    # 处理 SSE 常见的 data: 前缀
+    json_str = decoded_line
+    if decoded_line.startswith("data: "):
+        json_str = decoded_line[6:].strip()
+    
+    if json_str == "[DONE]": return None
+    
     try:
-        return json.loads(line)
+        data_obj = json.loads(json_str)
+        # 根据之前的日志，内容通常在 "data" 字段
+        return data_obj.get("data", "")
     except:
         return None
 
@@ -82,7 +86,7 @@ def clean_sse_line(line):
 
 async def verify_api(auth: HTTPAuthorizationCredentials = Depends(security)):
     if auth.credentials != ACCESS_TOKEN:
-        raise HTTPException(status_code=401, detail="Unauthorized")
+        raise HTTPException(status_code=401, detail="Invalid Access Token")
     return auth.credentials
 
 @app.get("/v1/models")
@@ -97,26 +101,25 @@ async def chat_completions(request: Request, t: str = Depends(verify_api)):
     content = body.get("messages", [])[-1]["content"] if body.get("messages") else ""
     
     jwt = get_valid_token()
-    if not jwt: raise HTTPException(status_code=500, detail="Provider Error")
+    if not jwt: raise HTTPException(status_code=500, detail="Failed to get provider token")
 
     url = "https://level2labs-prod--adm-agent-send-chat-message.modal.run/"
-    headers = {"Authorization": f"Bearer {jwt}"}
     data = {
         "content": (None, content), "model": (None, model),
         "conversation_id": (None, f"c-{uuid.uuid4()}"),
         "message_id": (None, f"m-{uuid.uuid4()}"), "max_mode": (None, "true")
     }
 
-    target_resp = requests.post(url, files=data, headers=headers, stream=True, timeout=60)
+    target_resp = requests.post(url, files=data, headers={"Authorization": f"Bearer {jwt}"}, stream=True, timeout=60)
 
     def generate():
         cid = f"chatcmpl-{uuid.uuid4()}"
         for line in target_resp.iter_lines():
-            item = clean_sse_line(line)
-            if item and "data" in item:
+            text_chunk = parse_upstream_line(line)
+            if text_chunk:
                 chunk = {
                     "id": cid, "object": "chat.completion.chunk", "created": int(time.time()),
-                    "model": model, "choices": [{"index": 0, "delta": {"content": item["data"]}, "finish_reason": None}]
+                    "model": model, "choices": [{"index": 0, "delta": {"content": text_chunk}, "finish_reason": None}]
                 }
                 yield f"data: {json.dumps(chunk)}\n\n"
         yield "data: [DONE]\n\n"
@@ -127,84 +130,85 @@ async def chat_completions(request: Request, t: str = Depends(verify_api)):
     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"]
+            text_chunk = parse_upstream_line(line)
+            if text_chunk:
+                full_text += text_chunk
         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("")
+# --- 5. Gradio 管理面板 ---
 
-    # 登录门禁
+with gr.Blocks(title="AI Doc Maker 管理面板", theme=gr.themes.Soft()) as demo:
+    # 鉴权门禁
     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()
+        gr.Markdown("## 🔐 访问受限\n请输入系统配置的 `ACCESS_TOKEN` 以解锁界面。")
+        pwd = gr.Textbox(label="ACCESS_TOKEN", type="password")
+        login_btn = gr.Button("解锁系统", variant="primary")
+        login_err = gr.Markdown()
 
-    # 主面板
+    # 主操作区
     with gr.Column(visible=False) as main_view:
-        gr.Markdown("# 🚀 AI Doc Maker 管理面板")
+        gr.Markdown("# 🚀 AI Doc Maker 后台管理")
         
-        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.Row():
+            ui_token = gr.Textbox(label="当前 Token (缓存)", interactive=False)
+            ui_balance = gr.Textbox(label="剩余额度", interactive=False)
+            ui_refresh = gr.Button("🔄 刷新账号")
+
+        with gr.Row():
+            with gr.Column():
+                ui_input = gr.Textbox(label="输入内容", lines=5, placeholder="写个笑话...")
+                with gr.Row():
+                    ui_chat_btn = gr.Button("💬 发送对话", variant="primary")
+                    ui_tts_btn = gr.Button("🎙️ 生成语音", variant="secondary")
             
-            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):
+            with gr.Column():
+                ui_output = gr.Textbox(label="AI 回复", lines=8, interactive=False)
+                ui_audio = gr.Audio(label="语音预览", interactive=False)
+
+    # --- UI 交互逻辑 ---
+    def handle_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), "", "", "", "❌ 验证失败"
+            return gr.update(visible=False), gr.update(visible=True), jwt, str(bal), ""
+        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])
+    login_btn.click(handle_login, inputs=pwd, outputs=[login_view, main_view, ui_token, ui_balance, login_err])
 
-    def on_refresh():
+    def handle_refresh():
         jwt = fetch_account()
         bal = sync_balance()
-        return jwt, bal, jwt
+        return jwt, str(bal)
 
-    ui_refresh.click(on_refresh, outputs=[ui_token, ui_balance, current_jwt_state])
+    ui_refresh.click(handle_refresh, outputs=[ui_token, ui_balance])
 
-    def on_ui_chat(jwt, text):
-        if not jwt: yield "请先登录", ""; return
+    def handle_ui_chat(text):
+        jwt = get_valid_token()
+        if not jwt: yield "Token 获取失败"; return
+        
         url = "https://level2labs-prod--adm-agent-send-chat-message.modal.run/"
         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():
-            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()
+        try:
+            resp = requests.post(url, files=data, headers={"Authorization": f"Bearer {jwt}"}, stream=True, timeout=60)
+            for line in resp.iter_lines():
+                chunk = parse_upstream_line(line)
+                if chunk:
+                    full_text += chunk
+                    yield full_text
+            sync_balance()
+        except Exception as e:
+            yield f"请求出错: {e}"
 
-    ui_chat_btn.click(on_ui_chat, inputs=[current_jwt_state, ui_input], outputs=[ui_reply, ui_raw_json])
+    ui_chat_btn.click(handle_ui_chat, inputs=ui_input, outputs=ui_output)
 
-    def on_ui_tts(jwt, text):
-        if not jwt: return "请先登录", None
+    def handle_ui_tts(text):
+        jwt = get_valid_token()
+        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}"}
@@ -212,12 +216,13 @@ with gr.Blocks(title="AI Doc Maker 管理面板") as demo:
             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
+            return audio_url
+        except:
+            return None
 
-    ui_tts_btn.click(on_ui_tts, inputs=[current_jwt_state, ui_input], outputs=[ui_audio_json, ui_audio_play])
+    ui_tts_btn.click(handle_ui_tts, inputs=ui_input, outputs=ui_audio)
 
+# 挂载并启动
 app = gr.mount_gradio_app(app, demo, path="/")
 
 if __name__ == "__main__":