test fix

app/__init__.py

@@ -67,46 +67,56 @@ def create_app(config_class=None):
     
     # Register test blueprint first (this should always work)
     logger.info("Registering test blueprint...")
-    from .routes.test import test_bp
-    app.register_blueprint(test_bp)
+    try:
+        from .routes.test import test_bp
+        app.register_blueprint(test_bp)
+        logger.info("Test blueprint registered successfully")
+    except Exception as e:
+        logger.error(f"Error registering test blueprint: {e}")
     
     # Register other blueprints
     logger.info("Registering other blueprints...")
+    
+    # Try to register each blueprint individually
     try:
-        with app.app_context():
-            logger.info("Importing auth blueprint...")
-            from .routes.auth import auth_bp
-            
-            logger.info("Importing dashboard blueprint...")
-            from .routes.dashboard import dashboard_bp
-            
-            logger.info("Importing API blueprint...")
-            from .routes.api import api_bp
-            
-            logger.info("Importing compliance blueprint...")
-            from .routes.compliance import compliance_bp
-            
-            logger.info("Importing Google Ads blueprint...")
-            from .routes.google_ads import google_ads_bp
-            
-            logger.info("Registering auth blueprint...")
-            app.register_blueprint(auth_bp)
-            
-            logger.info("Registering dashboard blueprint...")
-            app.register_blueprint(dashboard_bp)
-            
-            logger.info("Registering API blueprint...")
-            app.register_blueprint(api_bp)
-            
-            logger.info("Registering compliance blueprint...")
-            app.register_blueprint(compliance_bp)
-            
-            logger.info("Registering Google Ads blueprint...")
-            app.register_blueprint(google_ads_bp)
+        logger.info("Importing and registering dashboard blueprint...")
+        from .routes.dashboard import dashboard_bp
+        app.register_blueprint(dashboard_bp)
+        logger.info("Dashboard blueprint registered successfully")
     except Exception as e:
-        logger.error(f"Error registering blueprints: {e}")
-        # Continue without all blueprints for debugging
-        pass
+        logger.error(f"Error registering dashboard blueprint: {e}")
+    
+    try:
+        logger.info("Importing and registering API blueprint...")
+        from .routes.api import api_bp
+        app.register_blueprint(api_bp)
+        logger.info("API blueprint registered successfully")
+    except Exception as e:
+        logger.error(f"Error registering API blueprint: {e}")
+    
+    try:
+        logger.info("Importing and registering compliance blueprint...")
+        from .routes.compliance import compliance_bp
+        app.register_blueprint(compliance_bp)
+        logger.info("Compliance blueprint registered successfully")
+    except Exception as e:
+        logger.error(f"Error registering compliance blueprint: {e}")
+    
+    try:
+        logger.info("Importing and registering Google Ads blueprint...")
+        from .routes.google_ads import google_ads_bp
+        app.register_blueprint(google_ads_bp)
+        logger.info("Google Ads blueprint registered successfully")
+    except Exception as e:
+        logger.error(f"Error registering Google Ads blueprint: {e}")
+    
+    try:
+        logger.info("Importing and registering auth blueprint...")
+        from .routes.auth import auth_bp
+        app.register_blueprint(auth_bp)
+        logger.info("Auth blueprint registered successfully")
+    except Exception as e:
+        logger.error(f"Error registering auth blueprint: {e}")
     
     # Add a simple route directly to the app
     @app.route('/')

app/models/__init__.py

@@ -1,2 +1,38 @@
 # app/models/__init__.py
-# This file makes the models directory a Python package 
+# This file makes the models directory a Python package 
+
+from app import db
+from datetime import datetime
+from flask_login import UserMixin
+from werkzeug.security import generate_password_hash, check_password_hash
+
+class User(UserMixin, db.Model):
+    id = db.Column(db.Integer, primary_key=True)
+    email = db.Column(db.String(120), unique=True, nullable=False)
+    password = db.Column(db.String(128), nullable=False)
+    role = db.Column(db.String(20), default="viewer")
+    created_at = db.Column(db.DateTime, default=datetime.utcnow)
+    updated_at = db.Column(db.DateTime, default=datetime.utcnow, onupdate=datetime.utcnow)
+
+    def __repr__(self):
+        return f'<User {self.email}>'
+    
+    def set_password(self, password):
+        self.password = generate_password_hash(password)
+    
+    def check_password(self, password):
+        return check_password_hash(self.password, password)
+
+class Ad(db.Model):
+    id = db.Column(db.String(36), primary_key=True)
+    content = db.Column(db.Text)
+    sentiment = db.Column(db.JSON)
+    ad_metadata = db.Column(db.JSON)
+    user_id = db.Column(db.Integer, db.ForeignKey('user.id'))
+    versions = db.relationship('AdVersion', backref='ad')
+
+class AdVersion(db.Model):
+    id = db.Column(db.Integer, primary_key=True)
+    ad_id = db.Column(db.String(36), db.ForeignKey('ad.id'))
+    content = db.Column(db.Text)
+    timestamp = db.Column(db.DateTime, default=datetime.utcnow) 

app/routes/auth.py

@@ -2,6 +2,7 @@ from flask import Blueprint, render_template, redirect, url_for, flash
 from flask_login import login_user, logout_user, login_required
 from ..models import User
 from ..forms import LoginForm, RegistrationForm
+from app import db
 
 auth_bp = Blueprint('auth', __name__)