Update src/Captioner.py

src/Captioner.py

@@ -8,25 +8,25 @@ import json
 class Captioner:
     def __init__(self, api_key_path = None, proxy=None, api_base="https://api.lingyiwanwu.com/v1"):
         
-        if api_key_path is None:
-            # try find datas/01_key.txt and ../datas/01_key.txt
-            cand_paths = ['datas/01_key.txt', '../datas/01_key.txt']
-            flag = False
-            for path in cand_paths:
-                if os.path.exists(path):
-                    api_key_path = path
-                    flag = True
-                    break
+        # if api_key_path is None:
+        #     # try find datas/01_key.txt and ../datas/01_key.txt
+        #     cand_paths = ['datas/01_key.txt', '../datas/01_key.txt']
+        #     flag = False
+        #     for path in cand_paths:
+        #         if os.path.exists(path):
+        #             api_key_path = path
+        #             flag = True
+        #             break
 
-            if not flag:
-                raise ValueError("Please provide the path to the API key file.")
+        #     if not flag:
+        #         raise ValueError("Please provide the path to the API key file.")
 
 
-        self.api_key = self.load_access_token(api_key_path)
+        self.api_key = os.getenv('YI_VL_KEY')
         self.api_base = api_base
-        if proxy:
-            os.environ['HTTP_PROXY'] = proxy
-            os.environ['HTTPS_PROXY'] = proxy
+        # if proxy:
+        #     os.environ['HTTP_PROXY'] = proxy
+        #     os.environ['HTTPS_PROXY'] = proxy
         self.client = OpenAI(
             api_key=self.api_key,
             base_url=self.api_base