🐛 Fix: Persistent Company Profile via Database to resolve 404 in production

backend/app/routers/company.py

@@ -1,25 +1,63 @@
-from fastapi import APIRouter, HTTPException
+from fastapi import APIRouter, HTTPException, Depends
+from sqlalchemy.orm import Session
 from app.schemas.company import CompanyProfile
-from app.services.persistence import save_to_json, load_from_json
+from app.database import get_db
+from app.models.company import CompanyProfileModel
+import json
 
 router = APIRouter()
 
-# Load initial profile from disk
-_profiles = load_from_json(CompanyProfile, "company_profile.json")
-company_profile_cache: CompanyProfile | None = _profiles[0] if _profiles else None
-
-
 @router.post("/company-profile", response_model=CompanyProfile)
-def save_company_profile(profile: CompanyProfile):
-    global company_profile_cache
-    company_profile_cache = profile
-    # Persist to disk (as a list of one item for simplicity with current service)
-    save_to_json([profile], "company_profile.json")
+def save_company_profile(profile: CompanyProfile, db: Session = Depends(get_db)):
+    # Try to find existing profile (assume only one for now)
+    db_profile = db.query(CompanyProfileModel).first()
+    
+    if not db_profile:
+        db_profile = CompanyProfileModel()
+        db.add(db_profile)
+    
+    db_profile.name = profile.name
+    db_profile.industry = profile.industry
+    db_profile.services = json.dumps(profile.services)
+    db_profile.experience = profile.experience
+    db_profile.certifications = json.dumps(profile.certifications)
+    db_profile.regions = json.dumps(profile.regions)
+    db_profile.documents_available = json.dumps(profile.documents_available)
+    
+    db.commit()
+    db.refresh(db_profile)
     return profile
 
-
 @router.get("/company-profile", response_model=CompanyProfile)
-def get_company_profile():
-    if company_profile_cache is None:
+def get_company_profile(db: Session = Depends(get_db)):
+    db_profile = db.query(CompanyProfileModel).first()
+    if not db_profile:
         raise HTTPException(status_code=404, detail="No company profile saved")
-    return company_profile_cache
+    
+    # Handle list fields that are stored as JSON strings
+    services = []
+    certifications = []
+    regions = []
+    docs = []
+    
+    try: services = json.loads(db_profile.services) if db_profile.services else []
+    except: services = [db_profile.services] if db_profile.services else []
+    
+    try: certifications = json.loads(db_profile.certifications) if db_profile.certifications else []
+    except: certifications = []
+    
+    try: regions = json.loads(db_profile.regions) if db_profile.regions else []
+    except: regions = [db_profile.regions] if db_profile.regions else []
+    
+    try: docs = json.loads(db_profile.documents_available) if db_profile.documents_available else []
+    except: docs = [db_profile.documents_available] if db_profile.documents_available else []
+
+    return CompanyProfile(
+        name=db_profile.name,
+        industry=db_profile.industry,
+        services=services,
+        experience=db_profile.experience,
+        certifications=certifications,
+        regions=regions,
+        documents_available=docs
+    )