Add database port field and fix relationship between Customer and CallRecord

app/auth.py

@@ -100,7 +100,7 @@ async def create_customer(
     """Create a new customer"""
     try:
         # Validate required fields
-        required_fields = ["name", "company_name", "email", "db_host", "db_user", "db_password", "db_name"]
+        required_fields = ["name", "company_name", "email", "db_host", "db_port", "db_user", "db_password", "db_name"]
         missing_fields = [field for field in required_fields if field not in customer_data]
         if missing_fields:
             raise HTTPException(
@@ -123,6 +123,7 @@ async def create_customer(
             email=customer_data["email"],
             api_key=str(secrets.token_urlsafe(32)),
             db_host=customer_data["db_host"],
+            db_port=customer_data["db_port"],
             db_user=customer_data["db_user"],
             db_password=customer_data["db_password"],
             db_name=customer_data["db_name"]

app/models.py

@@ -15,122 +15,94 @@ class Customer(Base):
     __tablename__ = "customers"
 
     id = Column(Integer, primary_key=True, index=True)
-    name = Column(String(255), nullable=False)
-    company_name = Column(String(255), nullable=False)
-    email = Column(String(255), unique=True, nullable=False)
-    api_key = Column(String(255), unique=True, nullable=False)
+    name = Column(String(100), nullable=False)
+    company_name = Column(String(100), nullable=False)
+    email = Column(String(100), unique=True, nullable=False)
+    api_key = Column(String(100), unique=True, nullable=False)
     is_active = Column(Boolean, default=True)
     created_at = Column(DateTime, default=datetime.utcnow)
     updated_at = Column(DateTime, default=datetime.utcnow, onupdate=datetime.utcnow)
     
     # Database credentials
-    db_host = Column(String(255), nullable=False)
-    db_user = Column(String(255), nullable=False)
-    db_password = Column(String(255), nullable=False)
-    db_name = Column(String(255), nullable=False)
+    db_host = Column(String(100), nullable=False)
+    db_port = Column(Integer, nullable=False, default=3306)  # Default MySQL port
+    db_user = Column(String(100), nullable=False)
+    db_password = Column(String(100), nullable=False)
+    db_name = Column(String(100), nullable=False)
 
     # Relationship with call records
-    call_records = relationship("CallRecord", back_populates="customer")
+    call_records = relationship("CallRecord", back_populates="customer", cascade="all, delete-orphan")
 
     def get_db_engine(self):
-        """Get SQLAlchemy engine for customer's database"""
+        """Create SQLAlchemy engine for customer's database"""
         try:
-            # Construct database URL
-            db_url = f"mysql+pymysql://{self.db_user}:{self.db_password}@{self.db_host}/{self.db_name}"
-            
-            # Create engine
+            db_url = f"mysql+pymysql://{self.db_user}:{self.db_password}@{self.db_host}:{self.db_port}/{self.db_name}"
             engine = create_engine(db_url)
-            
             # Test connection
             with engine.connect() as conn:
                 conn.execute("SELECT 1")
-            
             return engine
         except Exception as e:
-            print(f"Error connecting to customer database: {str(e)}")
+            print(f"Error creating database engine: {str(e)}")
             return None
 
     def create_tables(self):
         """Create necessary tables in customer's database"""
-        engine = self.get_db_engine()
-        if not engine:
-            return False
-        
         try:
-            # Create tables
+            engine = self.get_db_engine()
+            if not engine:
+                return False
             Base.metadata.create_all(engine)
             return True
         except Exception as e:
             print(f"Error creating tables: {str(e)}")
             return False
 
-    def get_call_records(self, start_date=None, end_date=None):
-        """Get call records for this customer"""
+    def get_call_records(self):
+        """Get all call records for this customer"""
         engine = self.get_db_engine()
         if not engine:
             return []
-        
+        Session = sessionmaker(bind=engine)
+        session = Session()
         try:
-            Session = sessionmaker(bind=engine)
-            session = Session()
-            
-            query = session.query(CallRecord).filter(CallRecord.customer_id == self.id)
-            
-            if start_date:
-                query = query.filter(CallRecord.created_at >= start_date)
-            if end_date:
-                query = query.filter(CallRecord.created_at <= end_date)
-            
-            records = query.all()
-            session.close()
+            records = session.query(CallRecord).filter(CallRecord.customer_id == self.id).all()
             return records
-        except Exception as e:
-            print(f"Error getting call records: {str(e)}")
-            return []
+        finally:
+            session.close()
 
-    def get_call_details(self, call_id):
-        """Get details of a specific call"""
+    def get_call_details(self, call_id: int):
+        """Get detailed call information"""
         engine = self.get_db_engine()
         if not engine:
             return None
-        
+        Session = sessionmaker(bind=engine)
+        session = Session()
         try:
-            Session = sessionmaker(bind=engine)
-            session = Session()
-            
             record = session.query(CallRecord).filter(
                 CallRecord.id == call_id,
                 CallRecord.customer_id == self.id
             ).first()
-            
-            session.close()
             return record
-        except Exception as e:
-            print(f"Error getting call details: {str(e)}")
-            return None
+        finally:
+            session.close()
 
-    def search_calls(self, query):
-        """Search calls with custom query"""
+    def search_calls(self, query: str):
+        """Search calls by transcription or summary"""
         engine = self.get_db_engine()
         if not engine:
             return []
-        
+        Session = sessionmaker(bind=engine)
+        session = Session()
         try:
-            Session = sessionmaker(bind=engine)
-            session = Session()
-            
-            # Implement search logic here
-            # This is a basic example
             records = session.query(CallRecord).filter(
                 CallRecord.customer_id == self.id,
-                CallRecord.transcription.ilike(f"%{query}%")
+                (CallRecord.transcription.ilike(f"%{query}%") | 
+                 CallRecord.summary.ilike(f"%{query}%"))
             ).all()
-            
-            session.close()
             return records
-        except Exception as e:
-            print(f"Error searching calls: {str(e)}")
-            return []
+        finally:
+            session.close()
 
     def get_db_url(self) -> Optional[str]:
         """Get database URL if credentials are configured"""
@@ -192,10 +164,10 @@ class Customer(Base):
 class CallRecord(Base):
     __tablename__ = "call_records"
 
-    id = Column(String(255), primary_key=True)
-    customer_id = Column(Integer, nullable=False)
-    caller_number = Column(String(255), nullable=False)
-    called_number = Column(String(255), nullable=False)
+    id = Column(Integer, primary_key=True, index=True)
+    customer_id = Column(Integer, ForeignKey("customers.id"), nullable=False)
+    caller_number = Column(String(20), nullable=False)
+    called_number = Column(String(20), nullable=False)
     file_path = Column(String(255), nullable=False)
     transcription = Column(String(10000))
     summary = Column(String(1000))
@@ -203,6 +175,7 @@ class CallRecord(Base):
     created_at = Column(DateTime, default=datetime.utcnow)
     updated_at = Column(DateTime, default=datetime.utcnow, onupdate=datetime.utcnow)
 
+    # Relationship with customer
     customer = relationship("Customer", back_populates="call_records")
 
     @classmethod

app/static/admin.html

@@ -58,6 +58,10 @@
                             <label for="db_host" class="form-label">Database Host</label>
                             <input type="text" class="form-control" id="db_host" required>
                         </div>
+                        <div class="mb-3">
+                            <label for="db_port" class="form-label">Database Port</label>
+                            <input type="number" class="form-control" id="db_port" value="3306" required>
+                        </div>
                         <div class="mb-3">
                             <label for="db_user" class="form-label">Database Username</label>
                             <input type="text" class="form-control" id="db_user" required>
@@ -150,6 +154,7 @@
                 company_name: document.getElementById('company_name').value,
                 email: document.getElementById('email').value,
                 db_host: document.getElementById('db_host').value,
+                db_port: parseInt(document.getElementById('db_port').value),
                 db_user: document.getElementById('db_user').value,
                 db_password: document.getElementById('db_password').value,
                 db_name: document.getElementById('db_name').value