api errors solved

.gitignore

@@ -16,6 +16,9 @@ __pycache__/
 input/
 output*/
 csv_notes_pnl/
+csv_notes_bs/
+clean_financial_data_bs.json
+clean_financial_data_pnl.json
 generated_notes*/
 balancesheet_excel/
 cashflow_excel/

Dockerfile

@@ -1,4 +1,4 @@
-# Use Python 3.11 as base image (you can change to 3.9 if you want)
+# Use Python 3.11 as base image
 FROM python:3.11-slim
 
 # -------------------------------
@@ -14,15 +14,12 @@ RUN apt-get update && apt-get install -y \
     && rm -rf /var/lib/apt/lists/*
 
 # -------------------------------
-# Copy requirements file
+# Copy and install Python dependencies
 COPY requirements.txt .
-
-# -------------------------------
-# Install Python dependencies
 RUN pip install --no-cache-dir -r requirements.txt
 
 # -------------------------------
-# Create necessary directories (customize as needed)
+# Optional: Create necessary directories if not bind-mounted
 RUN mkdir -p /app/input \
     /app/output1 \
     /app/generated_notes \
@@ -35,19 +32,15 @@ RUN mkdir -p /app/input \
     /app/cashflow_excel \
     && chmod -R 777 /app/input /app/output1 /app/generated_notes /app/output2 /app/output3 /app/csv_notes_bs /app/csv_notes_pnl /app/balancesheet_excel /app/pnl_excel /app/cashflow_excel
 
-# -------------------------------
-# Copy the application code
-COPY . /app/
-
 # -------------------------------
 # Set environment variables
 ENV PYTHONPATH=/app
 ENV PYTHONUNBUFFERED=1
 
 # -------------------------------
-# Expose the port the app runs on
+# Expose the port
 EXPOSE 8000
 
 # -------------------------------
-# Command to run the FastAPI app
-CMD ["uvicorn", "app.main:app", "--host", "0.0.0.0", "--port", "8000"]
+# Start FastAPI with reload enabled
+CMD ["uvicorn", "app.main:app", "--host", "0.0.0.0", "--port", "8000", "--reload"]

app/api.py

@@ -1,5 +1,5 @@
 from fastapi import APIRouter, UploadFile, File, Form, HTTPException
-from fastapi.responses import JSONResponse, PlainTextResponse
+from fastapi.responses import JSONResponse, PlainTextResponse, FileResponse
 from typing import Optional, Dict, Any
 from app.notes import generate_notes
 from app.utils import clean_value
@@ -147,44 +147,41 @@ async def llm_generate_and_excel(
     wrapped_json_path = "generated_notes/notes_wrapped.json"
 
     if note_number:
-        # Support multiple note numbers (comma-separated)
+        # ...existing code for note_number...
         note_numbers = [n.strip() for n in note_number.split(",")]
         all_notes = []
         for n in note_numbers:
             success = generator.generate_note(n, trial_balance_path=output_json)
             if success:
-                # Read the just-generated note
                 with open("generated_notes/notes.json", "r", encoding="utf-8") as f:
                     note_json = json.load(f)
                 all_notes.append(note_json)
-        # Now write all notes together
         with open("generated_notes/notes.json", "w", encoding="utf-8") as f:
             json.dump({"notes": all_notes}, f, indent=2, ensure_ascii=False)
-        # --- Normalize all notes ---
         wrapped = normalize_llm_notes_json({"notes": all_notes})
         with open(wrapped_json_path, "w", encoding="utf-8") as f2:
             json.dump(wrapped, f2, ensure_ascii=False, indent=2)
-        # --------------------------
         excel_path = "generated_notes_excel/notes.xlsx"
         json_to_xlsx(wrapped_json_path, excel_path)
-        return {"message": f"Notes {', '.join(note_numbers)} generated. Excel saved at {excel_path}."}
     else:
-        # Generate all notes
+        # ...existing code for all notes...
         results = generator.generate_all_notes(trial_balance_path=output_json)
         if not any(results.values()):
             logger.error("Failed to generate any notes. LLM API may be down or unreachable.")
             raise HTTPException(status_code=500, detail="Failed to generate any notes. LLM API may be down or unreachable.")
-        # Read all notes.json
         with open("generated_notes/notes.json", "r", encoding="utf-8") as f:
             notes_json = json.load(f)
-        # --- Normalize all notes ---
         wrapped = normalize_llm_notes_json(notes_json)
         with open(wrapped_json_path, "w", encoding="utf-8") as f2:
             json.dump(wrapped, f2, ensure_ascii=False, indent=2)
-        # --------------------------
         excel_path = "generated_notes_excel/notes.xlsx"
         json_to_xlsx(wrapped_json_path, excel_path)
-        return {"message": f"All notes generated. Excel saved at {excel_path}."}
+    # Return the Excel file as a downloadable response
+    return FileResponse(
+        excel_path,
+        filename=os.path.basename(excel_path),
+        media_type="application/vnd.openxmlformats-officedocument.spreadsheetml.sheet"
+    )
         
 
 
@@ -264,7 +261,11 @@ async def run_full_pipeline(
         logger.error(f"json_xlsx.json_to_xlsx failed: {e}")
         raise HTTPException(status_code=500, detail=f"json_xlsx.json_to_xlsx failed: {e}")
 
-    return {"message": "Pipeline completed successfully. Excel file saved in output3."}
+    return FileResponse(
+        output3_xlsx,
+        filename=os.path.basename(output3_xlsx),
+        media_type="application/vnd.openxmlformats-officedocument.spreadsheetml.sheet"
+    )
 
 def run_subprocess(
     script_path: str,
@@ -352,7 +353,11 @@ async def bs_from_notes(file: UploadFile = File(...)):
         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_path}")
-    return {"message": "Balance Sheet generated successfully.", "file": output_file_path}
+    return FileResponse(
+        output_file_path,
+        filename=os.path.basename(output_file_path),
+        media_type="application/vnd.openxmlformats-officedocument.spreadsheetml.sheet"
+    )
 
 
 
@@ -400,4 +405,8 @@ async def pnl_from_notes(file: UploadFile = File(...)):
         raise HTTPException(status_code=500, detail=f"Could not determine output file from pnl_note.py output.{debug_msg}")
 
     logger.info(f"Pipeline completed. Output file: {output_file_path}")
-    return {"message": "Profit and Loss statement generated successfully.", "file": output_file_path}
+    return FileResponse(
+        output_file_path,
+        filename=os.path.basename(output_file_path),
+        media_type="application/vnd.openxmlformats-officedocument.spreadsheetml.sheet"
+    )

docker-compose.yml

@@ -1,23 +1,11 @@
-version: '3.9'
-
 services:
-  fastapi-api:
+  app:
     build: .
-    container_name: financial-notes-api
+    container_name: Financial-Statements
+    volumes:
+      - .:/app
     ports:
       - "8000:8000"
-    volumes:
-      - ./input:/app/input
-      - ./output1:/app/output1
-      - ./output2:/app/output2
-      - ./output3:/app/output3
-      - ./generated_notes:/app/generated_notes
-      - ./csv_notes_bs:/app/csv_notes_bs
-      - ./csv_notes_pnl:/app/csv_notes_pnl
-      - ./balancesheet_excel:/app/balancesheet_excel
-      - ./pnl_excel:/app/pnl_excel
-      - ./cashflow_excel:/app/cashflow_excel
-      - ./.env:/app/.env
     environment:
       - PYTHONUNBUFFERED=1
       - PORT=8000