/bs_from_notes working perfectly

app/api.py

@@ -328,7 +328,7 @@ async def bs_from_notes(file: UploadFile = File(...)):
     if os.getenv("OPENROUTER_API_KEY"):
         env["OPENROUTER_API_KEY"] = os.getenv("OPENROUTER_API_KEY")
     env["INPUT_FILE"] = "clean_financial_data_bs.json"
-    cwd = "C:/SAHIL/NOTES"
+    cwd = "C:/SAHIL/production_notes"
 
     # Run sircodebs.py
     run_subprocess("pnlbs/sircodebs.py", [input_excel_path], env, cwd)
@@ -341,13 +341,18 @@ async def bs_from_notes(file: UploadFile = File(...)):
     # Run bl_llm.py
     result = run_subprocess("pnlbs/bl_llm.py", [], env, cwd)
     output_file = extract_output_file(result.stdout)
-    if not output_file or not os.path.exists(output_file):
+    # If output_file is not absolute, resolve relative to cwd
+    if output_file and not os.path.isabs(output_file):
+        output_file_path = os.path.join(cwd, output_file)
+    else:
+        output_file_path = output_file
+    if not output_file or not os.path.exists(output_file_path):
         debug_msg = f"\nSTDOUT:\n{result.stdout}\nSTDERR:\n{result.stderr}"
         logger.error(f"Could not determine output file from bl_llm.py output.{debug_msg}")
         raise HTTPException(status_code=500, detail=f"Could not determine output file from bl_llm.py output.{debug_msg}")
 
-    logger.info(f"Pipeline completed. Output file: {output_file}")
-    return {"message": "Balance Sheet generated successfully.", "file": output_file}
+    logger.info(f"Pipeline completed. Output file: {output_file_path}")
+    return {"message": "Balance Sheet generated successfully.", "file": output_file_path}
 
 
 

pnlbs/bl_llm.py

@@ -1,3 +1,7 @@
+"""
+Enhanced Balance Sheet Generator
+Production-ready script for extracting, analyzing, and generating Excel balance sheets from financial JSON data.
+"""
 import os
 import json
 import re
@@ -14,7 +18,11 @@ from pydantic_settings import BaseSettings
 load_dotenv()
 
 # Configure logging
-logging.basicConfig(level=logging.INFO)
+logging.basicConfig(
+    level=logging.INFO,
+    format='%(asctime)s %(levelname)s %(name)s: %(message)s',
+    datefmt='%Y-%m-%d %H:%M:%S'
+)
 logger = logging.getLogger(__name__)
 
 class Settings(BaseSettings):
@@ -745,6 +753,7 @@ Return ONLY this JSON format:
         output_file = os.path.join(output_dir, f"balance_sheet_{timestamp}.xlsx")
         wb.save(output_file)
         logger.info(f"Output file: {output_file}")
+        print(f"Output file: {os.path.abspath(output_file)}")  # For API subprocess parsing
         return output_file
 
     def process(self, input_file: str, output_dir: str = "output") -> Optional[str]:
@@ -816,20 +825,40 @@ Return ONLY this JSON format:
             return None
 
 def main() -> None:
-    """Main function for running the balance sheet generator."""
-    logger.info("ENHANCED BALANCE SHEET GENERATOR v2.0")
+    """
+    Main function for running the balance sheet generator.
+    Accepts input file and output directory from command-line arguments or environment variables.
+    Handles errors gracefully and logs all major events.
+    """
+    logger.info("ENHANCED BALANCE SHEET GENERATOR v2.0 started.")
+    import sys
     api_key = settings.api_key
+    input_file = settings.input_file
+    output_dir = settings.output_dir
+    if len(sys.argv) > 1:
+        input_file = sys.argv[1]
+    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
-    input_file = settings.input_file
     if not os.path.exists(input_file):
         logger.error(f"Input file not found: {input_file}. Please ensure your JSON data file exists.")
         return
     generator = EnhancedBalanceSheetGenerator(api_key)
-    result = generator.process(input_file, settings.output_dir)
-    if result:
-        logger.info(f"COMPLETED SUCCESSFULLY! Output file: {result}")
-        logger.info("Open the Excel file to view your balance sheet")
-    else:
-        logger.error("PROCESSING FAILED. Please check the error messages above and try again.")
+    try:
+        result = generator.process(input_file, output_dir)
+        if result:
+            abs_path = os.path.abspath(result)
+            if os.path.exists(abs_path):
+                logger.info(f"COMPLETED SUCCESSFULLY! Output file: {abs_path}")
+                print(f"Output file: {abs_path}")  # For API subprocess parsing
+            else:
+                logger.error(f"PROCESSING FAILED. Output file not created: {abs_path}")
+        else:
+            logger.error("PROCESSING FAILED. Please check the error messages above and try again.")
+    except Exception as e:
+        logger.error(f"Fatal error: {e}", exc_info=True)
+
+if __name__ == "__main__":
+    main()

pnlbs/sircodebs.py

@@ -49,9 +49,15 @@ def main() -> None:
     """
     Main function to extract notes from Excel and export as CSVs.
     """
-    logger.info("Loading Excel file: %s", settings.excel_file_path)
+    # Use command-line argument for Excel file path if provided
+    excel_path = settings.excel_file_path
+    if len(sys.argv) > 1:
+        excel_path = sys.argv[1]
+        logger.info(f"Excel file path from argument: {excel_path}")
+    else:
+        logger.info(f"Excel file path from settings: {excel_path}")
     global xls
-    xls = pd.ExcelFile(settings.excel_file_path)
+    xls = pd.ExcelFile(excel_path)
 
     # Clean each sheet
     note_2_8_df = clean_note(settings.note_2_8_sheet, settings.skiprows)