Update app.py

app.py

@@ -15,6 +15,7 @@ import httpx
 
 API_BASE_URL = os.getenv("API_BASE_URL", "http://localhost:9999/v1")
 MODEL_NAME = os.getenv("MODEL_NAME", "Step-Audio-R1")
+SECRET = os.getenv("API_SECRET", "")
 
 def escape_html(text):
     """Escape HTML special characters to prevent XSS"""
@@ -298,6 +299,8 @@ def chat(system_prompt, user_text, audio_file, history, max_tokens, temperature,
                 "stream": True,
                 "repetition_penalty": 1.0,
                 "stop_token_ids": [151665]
+            }, headers={
+                "Authorization": f"Bearer {SECRET}",
             })
 
             if response.status_code != 200:
@@ -419,7 +422,6 @@ custom_css = """
     max-width: 100% !important;
     font-family: -apple-system, BlinkMacSystemFont, "Segoe UI", Roboto, "Helvetica Neue", Arial, sans-serif;
 }
-
 /* 标题样式 */
 .app-header {
     text-align: center;
@@ -431,7 +433,6 @@ custom_css = """
     margin-bottom: 1.5rem;
     box-shadow: 0 8px 24px rgba(102, 126, 234, 0.35);
 }
-
 /* 标题背景装饰 */
 .app-header::before {
     content: '';
@@ -443,12 +444,10 @@ custom_css = """
     background: radial-gradient(circle, rgba(255, 255, 255, 0.1) 0%, transparent 70%);
     animation: rotate 20s linear infinite;
 }
-
 @keyframes rotate {
     from { transform: rotate(0deg); }
     to { transform: rotate(360deg); }
 }
-
 .app-header h1 {
     margin: 0;
     font-size: 2.8rem;
@@ -459,7 +458,6 @@ custom_css = """
     position: relative;
     z-index: 1;
 }
-
 .app-header p {
     color: rgba(255, 255, 255, 0.95) !important;
     text-shadow: 0 2px 4px rgba(0, 0, 0, 0.2);
@@ -467,14 +465,12 @@ custom_css = """
     z-index: 1;
     line-height: 1.5;
 }
-
 /* 聊天框样式 */
 .chatbot-container {
     border-radius: 12px;
     box-shadow: 0 2px 8px rgba(0, 0, 0, 0.08);
     overflow: hidden;
 }
-
 /* 思考过程样式 - 模仿Claude/ChatGPT的风格 */
 .thinking-block {
     background: linear-gradient(135deg, #f5f7fa 0%, #eef2f7 100%);
@@ -484,7 +480,6 @@ custom_css = """
     border-radius: 8px;
     box-shadow: 0 2px 4px rgba(0, 0, 0, 0.05);
 }
-
 .thinking-header {
     display: flex;
     align-items: center;
@@ -493,7 +488,6 @@ custom_css = """
     margin-bottom: 10px;
     font-size: 0.95rem;
 }
-
 .thinking-content {
     background: #ffffff;
     padding: 12px 16px;
@@ -506,7 +500,6 @@ custom_css = """
     word-wrap: break-word;
     border: 1px solid #e5e7eb;
 }
-
 /* 回复分隔线 */
 .response-divider {
     border: none;
@@ -514,19 +507,16 @@ custom_css = """
     background: linear-gradient(to right, transparent, #e5e7eb, transparent);
     margin: 20px 0;
 }
-
 /* 按钮样式 */
 .primary-btn {
     background: linear-gradient(135deg, #667eea 0%, #764ba2 100%) !important;
     border: none !important;
     transition: all 0.3s ease !important;
 }
-
 .primary-btn:hover {
     transform: translateY(-2px);
     box-shadow: 0 4px 12px rgba(102, 126, 234, 0.4) !important;
 }
-
 /* 左侧面板样式 */
 .left-panel {
     background: #f9fafb;
@@ -534,49 +524,41 @@ custom_css = """
     padding: 1rem;
     height: 100%;
 }
-
 /* 输入框样式 */
 .input-box textarea {
     border-radius: 8px !important;
     border: 2px solid #e5e7eb !important;
     transition: border-color 0.3s ease !important;
 }
-
 .input-box textarea:focus {
     border-color: #667eea !important;
     box-shadow: 0 0 0 3px rgba(102, 126, 234, 0.1) !important;
 }
-
 /* 输入区域标题 */
 h3 {
     color: #374151;
     font-size: 1.1rem;
     margin: 1rem 0 0.5rem 0;
 }
-
 /* 聊天消息样式优化 */
 .message-wrap {
     padding: 1rem !important;
 }
-
 .message {
     padding: 1rem !important;
     border-radius: 12px !important;
     line-height: 1.6 !important;
 }
-
 /* 用户消息 */
 .message.user {
     background: linear-gradient(135deg, #667eea 0%, #764ba2 100%) !important;
     color: white !important;
 }
-
 /* 助手消息 */
 .message.bot {
     background: #f9fafb !important;
     border: 1px solid #e5e7eb !important;
 }
-
 /* 左侧面板整体样式 */
 .left-column {
     background: linear-gradient(to bottom, #ffffff 0%, #f9fafb 100%);
@@ -584,29 +566,24 @@ h3 {
     padding: 1rem;
     box-shadow: 0 2px 8px rgba(0, 0, 0, 0.05);
 }
-
 /* 按钮容器样式 */
 .button-row {
     margin-top: 1rem;
     gap: 0.5rem;
 }
-
 /* 滚动条美化 */
 ::-webkit-scrollbar {
     width: 8px;
     height: 8px;
 }
-
 ::-webkit-scrollbar-track {
     background: #f1f1f1;
     border-radius: 4px;
 }
-
 ::-webkit-scrollbar-thumb {
     background: #888;
     border-radius: 4px;
 }
-
 ::-webkit-scrollbar-thumb:hover {
     background: #555;
 }
@@ -751,4 +728,4 @@ if __name__ == "__main__":
     print(f"API地址: {API_BASE_URL}")
     print(f"模型: {MODEL_NAME}")
 
-    demo.launch(server_name=args.host, server_port=args.port, share=False)
+    demo.launch(server_name=args.host, server_port=args.port, share=False)