modified: _pycache__/create_app.cpython-312.pyc

modified: __pycache__/database.cpython-312.pyc
modified: configs.py
modified: create_app.py
modified: database.py
modified: inference/__pycache__/__init__.cpython-312.pyc
modified: inference/__pycache__/infer_single.cpython-312.pyc
modified: models/__pycache__/__init__.cpython-312.pyc
modified: models/__pycache__/model.cpython-312.pyc
modified: utils/__pycache__/__init__.cpython-312.pyc
modified: utils/__pycache__/data_utils.cpython-312.pyc
modified: utils/__pycache__/prompts.cpython-312.pyc
modified: utils/__pycache__/util_func.cpython-312.pyc
modified: views/__pycache__/__init__.cpython-312.pyc
modified: views/__pycache__/about.cpython-312.pyc
modified: views/__pycache__/auth.cpython-312.pyc
modified: views/__pycache__/dashboard.cpython-312.pyc
modified: views/__pycache__/error.cpython-312.pyc
modified: views/__p_ycache__/infer.cpython-312.pyc
modified: views/auth.py

configs.py

@@ -1,2 +1,3 @@
 SECRET_KEY = 'secret_key_here'
+SQLALCHEMY_DATABASE_URI = 'sqlite:///users.db'
 SQLALCHEMY_TRACK_MODIFICATIONS = False

create_app.py

@@ -5,7 +5,7 @@ from flask import render_template
 from transformers import LongformerTokenizer, AutoTokenizer, AutoModelForCausalLM, LongformerConfig
 from models.model import *
 from utils.util_func import *
-import os
+from safetensors.torch import load_file
 
 db = SQLAlchemy()
 login_manager = LoginManager()
@@ -20,67 +20,44 @@ MODEL_SESSION = None
 def load_models():
     global MODELS_LOADED, LONGFORMER_TOKENIZER, LONGFORMER_MODEL, QWEN_TOKENIZER, QWEN_MODEL
     device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
-    print("DEIVCE=", device)
     if not MODELS_LOADED:
-        print("START TO GET QWEN")
-        model_name = 'Qwen/Qwen3-0.6B'
-        QWEN_TOKENIZER = AutoTokenizer.from_pretrained(model_name, device='auto')
-        QWEN_TOKENIZER.pad_token_id = QWEN_TOKENIZER.eos_token_id
-        print("QWEN TOKENIZER LOADED")
-        try:
-            QWEN_MODEL = AutoModelForCausalLM.from_pretrained(model_name, device_map="auto")
-            QWEN_MODEL = QWEN_MODEL.to(device)
-            print("QWEN MODEL LOADED", flush=True)
-        except Exception as e:
-            print(f"ERROR LOADING QWEN MODEL: {str(e)}")
-            raise  # Re-raise to see full traceback
-        # QWEN_MODEL = AutoModelForCausalLM.from_pretrained(model_name).half()
-        # QWEN_MODEL = QWEN_MODEL.to(device)
-        print(QWEN_MODEL)
-        print("QWEN MODEL LOADED", flush=True)
         LONGFORMER_TOKENIZER = LongformerTokenizer.from_pretrained('allenai/longformer-base-4096', device='auto')
         config = LongformerConfig.from_json_file("Longformer_checkpoint/config.json")
-        print("LONGFORMER TOKENIZER LOADED")
         LONGFORMER_MODEL = CustomLongformerForSequenceClassification(config).from_pretrained('SFM2001/LongFormerScorer')
         LONGFORMER_MODEL = LONGFORMER_MODEL.to(device)
-        print(LONGFORMER_MODEL)
-        print("LONGFORMER MODEL LOADED", flush=True)
         LONGFORMER_MODEL.eval()
+        
+        model_name = 'Qwen/Qwen3-1.7B'
+        QWEN_TOKENIZER = AutoTokenizer.from_pretrained(model_name, device='auto')
+        QWEN_TOKENIZER.pad_token_id = QWEN_TOKENIZER.eos_token_id
+        QWEN_MODEL = AutoModelForCausalLM.from_pretrained(model_name, torch_dtype=torch.float16).half()
         MODELS_LOADED = True
-        print("LOAD ENDED", flush=True)
-
 
 def create_app():
     set_seed(42)
-    app = Flask(__name__, instance_relative_config=True)
-    
-    # Configure database
-    app.config['SQLALCHEMY_DATABASE_URI'] = 'sqlite:///' + os.path.join('/instance', 'users.db')
-    app.config['SQLALCHEMY_TRACK_MODIFICATIONS'] = False
-    app.config['SECRET_KEY'] = 'your-very-secret-key-here'  # Must be set!
-    
-    # Ensure instance folder exists
-    try:
-        os.makedirs(app.instance_path)
-    except OSError:
-        pass
-
-    # Initialize extensions
+    load_models()
+    app = Flask(__name__)
+    app.config.from_pyfile('configs.py')
     db.init_app(app)
     login_manager.init_app(app)
     login_manager.login_view = 'auth.login'
+    @login_manager.user_loader
+    def load_user(user_id):
+        return User.query.get(int(user_id))
 
-    # Load models (after db initialization)
     with app.app_context():
-        load_models()
-        db.create_all()  # Create tables if they don't exist
-        
-        # Register blueprints
         from views import auth_bp, dashboard_bp, infer_bp, about_bp, error_bp
         app.register_blueprint(auth_bp)
         app.register_blueprint(dashboard_bp)
         app.register_blueprint(infer_bp)
         app.register_blueprint(about_bp)
         app.register_blueprint(error_bp)
+        @app.errorhandler(Exception)
+        def handle_all_exceptions(e):
+            code = getattr(e, 'code', 500)
+            error_message = str(e) if hasattr(e, 'description') else "Something went wrong."
+            return render_template('error.html', code=code, error_message=error_message), code
+        from database import User, History
+        db.create_all()
     
-    return app
+    return app

database.py

@@ -3,17 +3,19 @@ from werkzeug.security import generate_password_hash, check_password_hash
 from datetime import datetime
 from create_app import db
 
+
 class User(UserMixin, db.Model):
     id = db.Column(db.Integer, primary_key=True)
     email = db.Column(db.String(100), unique=True, nullable=False)
     nickname = db.Column(db.String(100), unique=True, nullable=False)
-    password_hash = db.Column(db.String(128), nullable=False)
+    password_hash = db.Column(db.String(200))
 
     def set_password(self, password):
         self.password_hash = generate_password_hash(password)
 
     def check_password(self, password):
         return check_password_hash(self.password_hash, password)
+
 class History(db.Model):
     id = db.Column(db.Integer, primary_key=True)
     user_id = db.Column(db.Integer, db.ForeignKey('user.id'), nullable=False)
@@ -24,4 +26,4 @@ class History(db.Model):
     score_cc = db.Column(db.Integer, nullable=False)
     score_lr = db.Column(db.Integer, nullable=False)
     score_gra = db.Column(db.Integer, nullable=False)
-    user = db.relationship('User', backref=db.backref('histories', lazy=True))
+    user = db.relationship('User', backref=db.backref('histories', lazy=True))

views/auth.py

@@ -34,9 +34,13 @@ def login():
     if request.method == 'POST':
         email = request.form['email']
         password = request.form['password']
+        print(email, password)
         user = User.query.filter_by(email=email).first()
+        print(user)
         if user and user.check_password(password):
+            print("LOGIN?")
             login_user(user)
+            
             return redirect(url_for('dashboard.dashboard'))
         else:
             return render_template('login.html', message="Invalid email or password.")
@@ -47,4 +51,4 @@ def login():
 @login_required
 def logout():
     logout_user()
-    return redirect(url_for('auth.login'))
+    return redirect(url_for('auth.login'))