Update bs/balance_sheet_generator.py

bs/balance_sheet_generator.py

@@ -29,10 +29,13 @@ logging.basicConfig(
 logger = logging.getLogger(__name__)
 
 class Settings(BaseSettings):
-    """Application settings loaded from environment variables or .env file."""
-    api_key: str = Field(default_factory=lambda: os.getenv("OPENROUTER_API_KEY", ""), env="OPENROUTER_API_KEY")
-    input_file: str = Field(default="data/clean_financial_data_bs.json", env="INPUT_FILE")
-    output_dir: str = Field(default="data/output", env="BL_OUTPUT_DIR")
+    """Application settings loaded from environment variables."""
+    input_file: str = Field(default="data/clean_financial_data_bs.json")
+    output_dir: str = Field(default="data/output")
+    
+    model_config = {
+        "extra": "ignore"
+    }
 
 settings = Settings()
 
@@ -691,7 +694,7 @@ Return ONLY this JSON format:
 def main() -> None:
     """
     Main function for running the balance sheet generator.
-    Accepts input file and output directory from command-line arguments or environment variables.
+    Accepts input file, output directory, and API key from command-line arguments.
     Handles errors gracefully and logs all major events.
     """
     logger.info(" ENHANCED BALANCE SHEET GENERATOR v3.0 (Template-Based) started.")
@@ -704,7 +707,7 @@ def main() -> None:
     if len(sys.argv) > 2:
         output_dir = sys.argv[2]
     if not api_key:
-        logger.error(" Missing OpenRouter API key. Please provide a valid OPENROUTER_API_KEY.")
+        logger.error(" Missing OpenRouter API key. Please provide your API key via the 'user_api_key' parameter in your request.")
         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.")