add token

browser_qwen/background.js

@@ -1,7 +1,8 @@
-var database;
+var database_url;
 
 function send_data(msg){
     chrome.storage.local.get(['access_token'], function(result) {
+        console.log(result)
         if (result.access_token) {
             console.log('access_token currently is ' + result.access_token);
             database_url = "https://taohongxiu-qwen-agent.hf.space/endpoint";

browser_qwen/background_副本.js

@@ -56,4 +56,3 @@ chrome.runtime.onMessage.addListener(async (msg, sender) => {
 //    }
 //  }
 });
-Add to Qwen's Reading List

browser_qwen/src/content.js

@@ -29,7 +29,7 @@ button.style.position = 'fixed';
 button.style.top = '30px';
 button.style.right = '30px';
 button.style.zIndex = "9999";
-button.textContent = "Add to Qwen's Reading List";
+button.textContent = "Add to ChatGpt's Reading List";
 button.style.fontFamily = 'Arial, sans-serif';
 button.style.fontSize = '14px';
 button.style.width = '140px';

browser_qwen/src/popup.html

@@ -112,10 +112,6 @@
 <!--    <iframe src=$popup_url style="height: 550px"></iframe>-->
 <div id="iframe_area" style="height: 570px"></div>
 
-<h3 style="display: none;">Customize Address:</h3>
-<input type="text" id="addr" name="addr" class="input-text" style="display: none;">
-<button id="set_addr" class="upload_btn" style="display: none;">Change</button>
-
 <h3>Customize Token:</h3>
 <input type="text" id="access_token" name="access_token" class="input-text" placeholder="input and chick change button, then reopen this">
 <button id="set_access_token" class="upload_btn">Change</button>

browser_qwen/src/popup.js

@@ -41,26 +41,6 @@ document.addEventListener('DOMContentLoaded', function() {
         });
     }, 500);
 
-//    fetch('../config_host.json')
-//      .then(response => response.json())
-//      .then(data => {
-//        console.log(data);
-//        popup_url = "http://"+data.database_host+":"+data.app_in_browser_port+"/";
-//        console.log(popup_url);
-//    })
-//    .catch(error => console.error('Error:', error));
-})
-
-document.getElementById('set_addr').addEventListener('click', function() {
-    var addr = document.getElementById('addr').value;
-    // save config
-    chrome.storage.local.set({database_host: addr}, function() {
-      console.log('database_host is set to ' + addr);
-//      chrome.runtime.sendMessage({ data: addr , close: true , flag: 'set_addr'});
-      document.getElementById('addr').value = '';
-    });
-})
-
 document.getElementById('set_access_token').addEventListener('click', function() {
     var access_token = document.getElementById('access_token').value;
     // save config

qwen_server/database_server.py

@@ -15,9 +15,7 @@ from qwen_agent.log import logger
 from qwen_agent.utils.utils import get_local_ip
 from qwen_server.schema import GlobalConfig
 from qwen_server.utils import extract_and_cache_document
-from starlette.middleware.base import BaseHTTPMiddleware
-from typing import Callable, Coroutine, Any
-
+from starlette.middleware.sessions import SessionMiddleware
 
 # Read config
 with open(Path(__file__).resolve().parent / 'server_config.json', 'r') as f:
@@ -80,9 +78,10 @@ def change_checkbox_state(text, cache_file):
 @app.middleware("http")
 async def add_process_time_header(request: Request, call_next):
     print(f"Request URL path: {request.url}")
-    access_token: str = request.headers.get("Authorization") or request.query_params.get("access_token")
+    access_token: str = request.headers.get("Authorization") or request.query_params.get("access_token") or request.session.get("access_token")
     if not access_token or os.getenv("ACCESS_TOKEN") != access_token:
         return Response(status_code=401, content="the token is not valid")
+    request.session.setdefault("access_token", access_token)
     return await call_next(request)
 
 
@@ -132,7 +131,7 @@ from qwen_server.assistant_server import demo as assistant_app
 
 # app = gr.mount_gradio_app(app, workstation_app, path="/workstation")
 app = gr.mount_gradio_app(app, assistant_app, path="/")
-
+app.add_middleware(SessionMiddleware, secret_key='dddsadf453454f64554dsgdfghfg', max_age=25200)
 if __name__ == '__main__':
     uvicorn.run(app='database_server:app',
                 host=server_config.server.server_host,

requirements.txt

@@ -34,4 +34,5 @@ supervisor
 torch==2.0.1
 transformers_stream_generator
 einops
-accelerate
+accelerate
+itsdangerous