Update admin interface and add database credentials

app/auth.py

@@ -111,9 +111,27 @@ async def create_customer(
         name=customer_data["name"],
         company_name=customer_data["company_name"],
         email=customer_data["email"],
-        api_key=str(secrets.token_urlsafe(32))
+        api_key=str(secrets.token_urlsafe(32)),
+        db_host=customer_data["db_host"],
+        db_user=customer_data["db_user"],
+        db_password=customer_data["db_password"],
+        db_name=customer_data["db_name"]
     )
     
+    # Test database connection
+    if not customer.get_db_engine():
+        raise HTTPException(
+            status_code=status.HTTP_400_BAD_REQUEST,
+            detail="Invalid database credentials or connection failed"
+        )
+    
+    # Create tables in customer's database
+    if not customer.create_tables():
+        raise HTTPException(
+            status_code=status.HTTP_500_INTERNAL_SERVER_ERROR,
+            detail="Failed to create database tables"
+        )
+    
     db.add(customer)
     db.commit()
     db.refresh(customer)
@@ -123,7 +141,8 @@ async def create_customer(
         "name": customer.name,
         "company_name": customer.company_name,
         "email": customer.email,
-        "api_key": customer.api_key
+        "api_key": customer.api_key,
+        "is_active": customer.is_active
     }
 
 @router.get("/customers/", response_model=List[CustomerSchema])

app/models.py

@@ -1,11 +1,12 @@
 from sqlalchemy import Column, Integer, String, DateTime, ForeignKey, Boolean, create_engine, Text
 from sqlalchemy.ext.declarative import declarative_base
-from sqlalchemy.orm import relationship
+from sqlalchemy.orm import relationship, sessionmaker
 from datetime import datetime
 import requests
 from typing import Optional, List, Dict
 from .settings import API_BASE_URL
 from .database import get_db
+import os
 
 # Create declarative base
 Base = declarative_base()
@@ -14,23 +15,123 @@ class Customer(Base):
     __tablename__ = "customers"
 
     id = Column(Integer, primary_key=True, index=True)
-    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(64), unique=True, nullable=False)
+    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)
     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 for customer's own database
-    db_host = Column(String(255))
-    db_port = Column(Integer)
-    db_name = Column(String(100))
-    db_user = Column(String(100))
-    db_password = Column(String(255))
+    
+    # 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)
 
     # Relationship with call records
     call_records = relationship("CallRecord", back_populates="customer")
 
+    def get_db_engine(self):
+        """Get 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
+            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)}")
+            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
+            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"""
+        engine = self.get_db_engine()
+        if not engine:
+            return []
+        
+        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()
+            return records
+        except Exception as e:
+            print(f"Error getting call records: {str(e)}")
+            return []
+
+    def get_call_details(self, call_id):
+        """Get details of a specific call"""
+        engine = self.get_db_engine()
+        if not engine:
+            return None
+        
+        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
+
+    def search_calls(self, query):
+        """Search calls with custom query"""
+        engine = self.get_db_engine()
+        if not engine:
+            return []
+        
+        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}%")
+            ).all()
+            
+            session.close()
+            return records
+        except Exception as e:
+            print(f"Error searching calls: {str(e)}")
+            return []
+
     def get_db_url(self) -> Optional[str]:
         """Get database URL if credentials are configured"""
         if all([self.db_host, self.db_port, self.db_name, self.db_user, self.db_password]):
@@ -91,15 +192,14 @@ class Customer(Base):
 class CallRecord(Base):
     __tablename__ = "call_records"
 
-    id = Column(String(36), primary_key=True)
-    customer_id = Column(Integer, ForeignKey("customers.id"))
-    caller_number = Column(String(20), nullable=False)
-    called_number = Column(String(20), nullable=False)
-    file_path = Column(String(255))
-    transcription = Column(Text)
-    summary = Column(Text)
-    sentiment = Column(String(20))
-    keywords = Column(Text)
+    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)
+    file_path = Column(String(255), nullable=False)
+    transcription = Column(String(10000))
+    summary = Column(String(1000))
+    sentiment = Column(String(50))
     created_at = Column(DateTime, default=datetime.utcnow)
     updated_at = Column(DateTime, default=datetime.utcnow, onupdate=datetime.utcnow)
 

app/static/admin.html

@@ -9,6 +9,7 @@
         .container { max-width: 800px; margin-top: 2rem; }
         .customer-list { margin-top: 2rem; }
         .token-section { margin-bottom: 2rem; }
+        .hidden { display: none; }
     </style>
 </head>
 <body>
@@ -16,7 +17,7 @@
         <h1 class="mb-4">vBot Admin Interface</h1>
         
         <!-- Login Section -->
-        <div class="card token-section">
+        <div class="card token-section" id="loginSection">
             <div class="card-body">
                 <h5 class="card-title">Admin Login</h5>
                 <form id="loginForm">
@@ -34,46 +35,65 @@
             </div>
         </div>
 
-        <!-- Add Customer Section -->
-        <div class="card">
-            <div class="card-body">
-                <h5 class="card-title">Add New Customer</h5>
-                <form id="customerForm">
-                    <div class="mb-3">
-                        <label for="name" class="form-label">Name</label>
-                        <input type="text" class="form-control" id="name" required>
-                    </div>
-                    <div class="mb-3">
-                        <label for="company_name" class="form-label">Company Name</label>
-                        <input type="text" class="form-control" id="company_name" required>
-                    </div>
-                    <div class="mb-3">
-                        <label for="email" class="form-label">Email</label>
-                        <input type="email" class="form-control" id="email" required>
-                    </div>
-                    <button type="submit" class="btn btn-success">Add Customer</button>
-                </form>
-                <div id="customerStatus" class="mt-3"></div>
+        <!-- Main Content (Hidden until login) -->
+        <div id="mainContent" class="hidden">
+            <!-- Add Customer Section -->
+            <div class="card">
+                <div class="card-body">
+                    <h5 class="card-title">Add New Customer</h5>
+                    <form id="customerForm">
+                        <div class="mb-3">
+                            <label for="name" class="form-label">Name</label>
+                            <input type="text" class="form-control" id="name" required>
+                        </div>
+                        <div class="mb-3">
+                            <label for="company_name" class="form-label">Company Name</label>
+                            <input type="text" class="form-control" id="company_name" required>
+                        </div>
+                        <div class="mb-3">
+                            <label for="email" class="form-label">Email</label>
+                            <input type="email" class="form-control" id="email" required>
+                        </div>
+                        <div class="mb-3">
+                            <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_user" class="form-label">Database Username</label>
+                            <input type="text" class="form-control" id="db_user" required>
+                        </div>
+                        <div class="mb-3">
+                            <label for="db_password" class="form-label">Database Password</label>
+                            <input type="password" class="form-control" id="db_password" required>
+                        </div>
+                        <div class="mb-3">
+                            <label for="db_name" class="form-label">Database Name</label>
+                            <input type="text" class="form-control" id="db_name" required>
+                        </div>
+                        <button type="submit" class="btn btn-success">Add Customer</button>
+                    </form>
+                    <div id="customerStatus" class="mt-3"></div>
+                </div>
             </div>
-        </div>
 
-        <!-- Customer List Section -->
-        <div class="card customer-list">
-            <div class="card-body">
-                <h5 class="card-title">Customer List</h5>
-                <div id="customerList" class="table-responsive">
-                    <table class="table">
-                        <thead>
-                            <tr>
-                                <th>Name</th>
-                                <th>Company</th>
-                                <th>Email</th>
-                                <th>API Key</th>
-                                <th>Status</th>
-                            </tr>
-                        </thead>
-                        <tbody id="customerTableBody"></tbody>
-                    </table>
+            <!-- Customer List Section -->
+            <div class="card customer-list">
+                <div class="card-body">
+                    <h5 class="card-title">Customer List</h5>
+                    <div id="customerList" class="table-responsive">
+                        <table class="table">
+                            <thead>
+                                <tr>
+                                    <th>Name</th>
+                                    <th>Company</th>
+                                    <th>Email</th>
+                                    <th>API Key</th>
+                                    <th>Status</th>
+                                </tr>
+                            </thead>
+                            <tbody id="customerTableBody"></tbody>
+                        </table>
+                    </div>
                 </div>
             </div>
         </div>
@@ -102,6 +122,9 @@
                     accessToken = data.access_token;
                     document.getElementById('tokenStatus').innerHTML = 
                         '<div class="alert alert-success">Login successful!</div>';
+                    // Show main content and hide login section
+                    document.getElementById('loginSection').classList.add('hidden');
+                    document.getElementById('mainContent').classList.remove('hidden');
                     loadCustomers();
                 } else {
                     document.getElementById('tokenStatus').innerHTML = 
@@ -125,7 +148,11 @@
             const customerData = {
                 name: document.getElementById('name').value,
                 company_name: document.getElementById('company_name').value,
-                email: document.getElementById('email').value
+                email: document.getElementById('email').value,
+                db_host: document.getElementById('db_host').value,
+                db_user: document.getElementById('db_user').value,
+                db_password: document.getElementById('db_password').value,
+                db_name: document.getElementById('db_name').value
             };
 
             try {
@@ -209,11 +236,6 @@
                 `;
             }
         }
-
-        // Initial load if token exists
-        if (accessToken) {
-            loadCustomers();
-        }
     </script>
 </body>
 </html>