fix: auth failed

pyproject.toml

@@ -2,8 +2,20 @@
 build-backend = "poetry-core.masonry.api"
 requires = ["poetry-core"]
 
+[tool.coverage.report]
+exclude_also = [
+  "raise AssertionError",
+  "raise NotImplementedError",
+  "if __name__ == .__main__.:",
+  "if TYPE_CHECKING:",
+  "except Exception as e"
+]
+
 [tool.coverage.run]
-omit = ["yuisub"]
+omit = [
+  "./yuisub/__main__.py",
+  "./yuisub/__init__.py"
+]
 
 [tool.mypy]
 disable_error_code = "attr-defined"

yuisub/__main__.py

@@ -8,11 +8,12 @@
 
 parser = argparse.ArgumentParser()
 parser.description = "Generate bilingual SRT files from audio or SRT input."
-# srt
+# input
 parser.add_argument("-a", "--AUDIO", type=str, help="Path to the audio file", required=False)
 parser.add_argument("-s", "--SRT", type=str, help="Path to the input SRT file", required=False)
-parser.add_argument("-oz", "--OUTPUT_ZH", type=str, help="Path to save the Chinese SRT file", required=True)
-parser.add_argument("-ob", "--OUTPUT_BILINGUAL", type=str, help="Path to save the bilingual SRT file", required=True)
+# srt output
+parser.add_argument("-oz", "--OUTPUT_ZH", type=str, help="Path to save the Chinese SRT file", required=False)
+parser.add_argument("-ob", "--OUTPUT_BILINGUAL", type=str, help="Path to save the bilingual SRT file", required=False)
 # openai gpt
 parser.add_argument("-om", "--OPENAI_MODEL", type=str, help="Openai model name", required=True)
 parser.add_argument("-api", "--OPENAI_API_KEY", type=str, help="Openai API key", required=True)
@@ -30,6 +31,9 @@ def main() -> None:
     if args.AUDIO and args.SRT:
         raise ValueError("Please provide only one input file, either audio or SRT.")
 
+    if not args.OUTPUT_ZH and not args.OUTPUT_BILINGUAL:
+        raise ValueError("Please provide output paths for the subtitles.")
+
     if args.AUDIO:
         import torch
 

yuisub/llm.py

@@ -38,6 +38,10 @@ async def ask(self, question: ORIGIN) -> ZH:
             content = json.loads(response.choices[0].message.content)
             zh = ZH(**content)
 
+        except openai.AuthenticationError as e:
+            print(f"Authentication Error: {e} retrying...")
+            raise e
+
         except openai.APIConnectionError as e:
             print(f"Connection Error: {e} retrying...")
             raise e