diff --git a/.env b/.env index 1664e9d..42413d2 100644 --- a/.env +++ b/.env @@ -1,3 +1,4 @@ -GEMINI_API_KEY= -GEMINI_API_URL= -GEMINI_API_MODEL= +API_KEY= +API_URL= +API_MODEL= +ANNOUNCEMENT_TEXT= \ No newline at end of file diff --git a/README.md b/README.md index c46ea71..1cd37bd 100644 --- a/README.md +++ b/README.md @@ -14,9 +14,9 @@ docker run -d \ --name stock-scanner \ -p 8888:8888 \ - -e GEMINI_API_KEY=替换为你的key \ - -e GEMINI_API_URL=替换为你的api地址 \ - -e GEMINI_API_MODEL=替换为你的模型 \ + -e API_KEY=替换为你的key \ + -e API_URL=替换为你的api地址 \ + -e API_MODEL=替换为你的模型 \ lanzhihong/stock-scanner:latest ``` 默认8888端口,部署完成后访问 http://127.0.0.1:8888 即可使用。 diff --git a/docker-compose.yml b/docker-compose.yml index 0b5b3c1..a89085b 100644 --- a/docker-compose.yml +++ b/docker-compose.yml @@ -4,9 +4,11 @@ services: stock-analyzer: build: . ports: - - "50684:8080" + - "8888:8888" environment: - - GEMINI_API_KEY=${GEMINI_API_KEY} + - API_KEY=${API_KEY} + - API_URL=${API_URL} + - API_MODEL=${API_MODEL} volumes: - .:/app restart: unless-stopped diff --git a/stock_analyzer.py b/stock_analyzer.py index 867a52b..815f4b2 100644 --- a/stock_analyzer.py +++ b/stock_analyzer.py @@ -18,8 +18,8 @@ class StockAnalyzer: load_dotenv() # 设置 Gemini API - self.gemini_api_url = os.getenv('GEMINI_API_URL') - self.gemini_api_key = os.getenv('GEMINI_API_KEY') + self.API_URL = os.getenv('API_URL') + self.API_KEY = os.getenv('API_KEY') # 配置参数 self.params = { @@ -243,17 +243,17 @@ class StockAnalyzer: """ headers = { - "Authorization": f"Bearer {self.gemini_api_key}", + "Authorization": f"Bearer {self.API_KEY}", "Content-Type": "application/json" } data = { - "model": os.getenv('GEMINI_API_MODEL'), + "model": os.getenv('API_MODEL'), "messages": [{"role": "user", "content": prompt}] } response = requests.post( - f"{self.gemini_api_url}/v1/chat/completions", + f"{self.API_URL}/v1/chat/completions", headers=headers, json=data, timeout=30 diff --git a/templates/index.html b/templates/index.html index 4b0b060..aee282d 100644 --- a/templates/index.html +++ b/templates/index.html @@ -10,6 +10,26 @@
+ {{ announcement }} +
+