Update bs/balance_sheet_generator.py

bs/balance_sheet_generator.py

@@ -696,7 +696,7 @@ def main() -> None:
     """
     logger.info(" ENHANCED BALANCE SHEET GENERATOR v3.0 (Template-Based) started.")
     import sys
-    api_key = settings.api_key
+    api_key = os.environ.get("OPENROUTER_API_KEY")
     input_file = settings.input_file
     output_dir = settings.output_dir
     if len(sys.argv) > 1:
@@ -704,11 +704,11 @@ def main() -> None:
     if len(sys.argv) > 2:
         output_dir = sys.argv[2]
     if not api_key:
-        logger.error(" Missing OPENROUTER_API_KEY environment variable. Please set your OpenRouter API key in the .env file.")
-        return
+        logger.error(" Missing OpenRouter API key. Please provide a valid OPENROUTER_API_KEY.")
+        sys.exit(1)
     if not os.path.exists(input_file):
         logger.error(f" Input file not found: {input_file}. Please ensure your JSON data file exists.")
-        return
+        sys.exit(1)
     
     generator = EnhancedBalanceSheetGenerator(api_key)
     try:
@@ -720,10 +720,13 @@ def main() -> None:
                 print(f"Output file: {abs_path}")  # For API subprocess parsing
             else:
                 logger.error(f" PROCESSING FAILED. Output file not created: {abs_path}")
+                sys.exit(1)
         else:
             logger.error(" PROCESSING FAILED. Please check the error messages above and try again.")
+            sys.exit(1)
     except Exception as e:
         logger.error(f" Fatal error: {e}", exc_info=True)
+        sys.exit(1)
 
 if __name__ == "__main__":
     main()