Add Pydantic models and update endpoints to use them

app/schemas.py

@@ -0,0 +1,41 @@
+from pydantic import BaseModel
+from typing import Optional
+from datetime import datetime
+
+class CustomerBase(BaseModel):
+    name: str
+    company_name: str
+    email: str
+
+class CustomerCreate(CustomerBase):
+    pass
+
+class Customer(CustomerBase):
+    id: int
+    api_key: str
+    is_active: bool
+    created_at: datetime
+    updated_at: datetime
+
+    class Config:
+        from_attributes = True
+
+class CallRecordBase(BaseModel):
+    caller_number: str
+    called_number: str
+    file_path: str
+    transcription: str
+    summary: str
+    sentiment: str
+
+class CallRecordCreate(CallRecordBase):
+    pass
+
+class CallRecord(CallRecordBase):
+    id: str
+    customer_id: int
+    created_at: datetime
+    updated_at: datetime
+
+    class Config:
+        from_attributes = True 

main.py

@@ -14,7 +14,8 @@ from app.settings import (
     ADMIN_USERNAME, ADMIN_PASSWORD, WHISPER_MODEL,
     SUMMARIZER_MODEL, SENTIMENT_MODEL
 )
-from app.models import Customer, CallRecord
+from app.models import Customer as CustomerModel, CallRecord as CallRecordModel
+from app.schemas import Customer, CustomerCreate, CallRecord
 from app.auth import verify_api_key, get_current_admin, create_access_token, router as auth_router
 from app.database import get_db
 from sqlalchemy.orm import sessionmaker
@@ -49,9 +50,9 @@ app.include_router(auth_router, prefix="/api/v1", tags=["auth"])
 
 async def get_customer_by_api_key(api_key: str = Header(...), db: Session = Depends(get_db)):
     """Get customer by API key"""
-    customer = db.query(Customer).filter(
-        Customer.api_key == api_key,
-        Customer.is_active == True
+    customer = db.query(CustomerModel).filter(
+        CustomerModel.api_key == api_key,
+        CustomerModel.is_active == True
     ).first()
     if not customer:
         raise HTTPException(
@@ -65,7 +66,7 @@ async def process_call(
     file: UploadFile = File(...),
     caller_number: str = Form(...),
     called_number: str = Form(...),
-    customer: Customer = Depends(verify_api_key),
+    customer: CustomerModel = Depends(verify_api_key),
     db: Session = Depends(get_db)
 ):
     """
@@ -104,7 +105,7 @@ async def process_call(
         
         try:
             # Create call record
-            call_record = CallRecord(
+            call_record = CallRecordModel(
                 id=process_id,
                 customer_id=customer.id,
                 caller_number=caller_number,