Update

app.py

@@ -1,7 +1,7 @@
 import os
 import torch
 from transformers import AutoTokenizer
-from flask import Flask, request, jsonify, render_template, redirect, url_for, flash, session
+from flask import Flask, request, jsonify, render_template, redirect, url_for, flash
 from flask_login import LoginManager, login_user, logout_user, login_required, current_user
 from models import db, User, Feedback
 from forms import RegistrationForm, LoginForm
@@ -11,15 +11,8 @@ app = Flask(__name__)
 
 # Cấu hình
 app.config['SECRET_KEY'] = 'your-secret-key-change-this-in-production'
-# Tắt CSRF protection để test
-app.config['WTF_CSRF_ENABLED'] = False
-# Cấu hình session cho Hugging Face Spaces
-app.config['SESSION_COOKIE_SECURE'] = False  # Cho HTTP
-app.config['SESSION_COOKIE_HTTPONLY'] = True
-app.config['SESSION_COOKIE_SAMESITE'] = 'Lax'
-# Sử dụng database persistent cho Hugging Face Spaces
-# Sử dụng thư mục workspace để lưu database (persistent across restarts)
-db_path = os.path.join(os.getcwd(), 'feedback_analysis.db')
+# Sử dụng đường dẫn database phù hợp với Hugging Face Spaces
+db_path = os.path.join(os.getcwd(), 'instance', 'feedback_analysis.db')
 os.makedirs(os.path.dirname(db_path), exist_ok=True)
 app.config['SQLALCHEMY_DATABASE_URI'] = f'sqlite:///{db_path}'
 app.config['SQLALCHEMY_TRACK_MODIFICATIONS'] = False
@@ -59,131 +52,40 @@ print("✅ Model đã sẵn sàng!")
 @app.route("/", methods=["GET"])
 @login_required
 def home():
-    print("=" * 40)
-    print("🏠 HOME DEBUG")
-    print("=" * 40)
-    print(f"Current user authenticated: {current_user.is_authenticated}")
-    print(f"Current user ID: {current_user.id if current_user.is_authenticated else 'None'}")
-    print(f"Current user username: {current_user.username if current_user.is_authenticated else 'None'}")
-    print(f"Session data: {dict(session)}")
-    print("=" * 40)
-    print("🏁 HOME DEBUG END")
-    print("=" * 40)
     return render_template("index.html")
 
-@app.route("/debug", methods=["GET"])
-def debug():
-    """Debug route để kiểm tra session và database"""
-    try:
-        user_count = User.query.count()
-        users = User.query.all()
-        
-        return jsonify({
-            "current_user_authenticated": current_user.is_authenticated,
-            "current_user_id": current_user.id if current_user.is_authenticated else None,
-            "current_user_username": current_user.username if current_user.is_authenticated else None,
-            "session_data": dict(session),
-            "database_path": db_path,
-            "database_exists": os.path.exists(db_path),
-            "user_count": user_count,
-            "users": [{"id": u.id, "username": u.username} for u in users]
-        })
-    except Exception as e:
-        return jsonify({"error": str(e)})
-
 @app.route("/register", methods=["GET", "POST"])
 def register():
-    print("=" * 40)
-    print("📝 REGISTER DEBUG")
-    print("=" * 40)
-    print(f"Request method: {request.method}")
-    print(f"Current user authenticated: {current_user.is_authenticated}")
-    
     if current_user.is_authenticated:
-        print("User already authenticated, redirecting to home")
         return redirect(url_for('home'))
     
     form = RegistrationForm()
-    print(f"Form created: {form}")
-    
     if form.validate_on_submit():
-        print(f"Form validated successfully")
-        print(f"Username: {form.username.data}")
-        print(f"Password: {form.password.data}")
-        
-        try:
-            user = User(username=form.username.data)
-            user.set_password(form.password.data)
-            db.session.add(user)
-            db.session.commit()
-            print("✅ User created successfully")
-            flash('Đăng ký thành công! Vui lòng đăng nhập.', 'success')
-            print("Flash message set: Đăng ký thành công!")
-            return redirect(url_for('login'))
-        except Exception as e:
-            print(f"❌ Error creating user: {e}")
-            flash('Có lỗi xảy ra khi đăng ký. Vui lòng thử lại.', 'danger')
-            print("Flash message set: Có lỗi xảy ra khi đăng ký.")
-    else:
-        print(f"Form validation failed: {form.errors}")
+        user = User(username=form.username.data)
+        user.set_password(form.password.data)
+        db.session.add(user)
+        db.session.commit()
+        flash('Đăng ký thành công! Vui lòng đăng nhập.', 'success')
+        return redirect(url_for('login'))
     
-    print("=" * 40)
-    print("🏁 REGISTER DEBUG END")
-    print("=" * 40)
     return render_template('register.html', form=form)
 
 @app.route("/login", methods=["GET", "POST"])
 def login():
-    print("=" * 40)
-    print("🔐 LOGIN DEBUG")
-    print("=" * 40)
-    print(f"Request method: {request.method}")
-    print(f"Current user authenticated: {current_user.is_authenticated}")
-    
     if current_user.is_authenticated:
-        print("User already authenticated, redirecting to home")
         return redirect(url_for('home'))
     
     form = LoginForm()
-    print(f"Form created: {form}")
-    
     if form.validate_on_submit():
-        print(f"Form validated successfully")
-        print(f"Username: {form.username.data}")
-        print(f"Password: {form.password.data}")
-        
         user = User.query.filter_by(username=form.username.data).first()
-        print(f"User found: {user}")
-        
-        if user:
-            print(f"User exists, checking password...")
-            password_check = user.check_password(form.password.data)
-            print(f"Password check result: {password_check}")
-            
-            if password_check:
-                print("Password correct, logging in user")
-                login_user(user, remember=True)
-                print(f"User authenticated after login: {current_user.is_authenticated}")
-                print(f"User ID: {current_user.id}")
-                print(f"Session data: {dict(session)}")
-                flash('Đăng nhập thành công!', 'success')
-                print("Flash message set: Đăng nhập thành công!")
-                next_page = request.args.get('next')
-                return redirect(next_page) if next_page else redirect(url_for('home'))
-            else:
-                print("Password incorrect")
-                flash('Tên đăng nhập hoặc mật khẩu không đúng.', 'danger')
-                print("Flash message set: Tên đăng nhập hoặc mật khẩu không đúng.")
+        if user and user.check_password(form.password.data):
+            login_user(user, remember=True)
+            flash('Đăng nhập thành công!', 'success')
+            next_page = request.args.get('next')
+            return redirect(next_page) if next_page else redirect(url_for('home'))
         else:
-            print("User not found")
             flash('Tên đăng nhập hoặc mật khẩu không đúng.', 'danger')
-            print("Flash message set: Tên đăng nhập hoặc mật khẩu không đúng.")
-    else:
-        print(f"Form validation failed: {form.errors}")
     
-    print("=" * 40)
-    print("🏁 LOGIN DEBUG END")
-    print("=" * 40)
     return render_template('login.html', form=form)
 
 @app.route("/logout")
@@ -197,26 +99,6 @@ def logout():
 def health():
     return jsonify({"status": "healthy", "message": "✅ PhoBERT MultiTask API is running!"})
 
-@app.route("/api/users", methods=["GET"])
-def get_users():
-    """API để xem danh sách users (cho admin)"""
-    try:
-        users = User.query.all()
-        user_list = []
-        for user in users:
-            user_list.append({
-                'id': user.id,
-                'username': user.username,
-                'created_at': user.created_at.isoformat(),
-                'feedback_count': len(user.feedbacks)
-            })
-        return jsonify({
-            'users': user_list,
-            'total': len(user_list)
-        })
-    except Exception as e:
-        return jsonify({"error": f"Có lỗi xảy ra: {str(e)}"}), 500
-
 @app.route("/api/feedback-history", methods=["GET"])
 @login_required
 def get_feedback_history():
@@ -317,71 +199,8 @@ def predict():
 
 # Khởi tạo database
 with app.app_context():
-    print("=" * 60)
-    print("🔍 DEBUG DATABASE - HUGGING FACE SPACE")
-    print("=" * 60)
-    
-    # Debug: Kiểm tra đường dẫn database
-    print(f"📁 Current working directory: {os.getcwd()}")
-    print(f"📁 Database path: {db_path}")
-    print(f"📁 Database exists: {os.path.exists(db_path)}")
-    print(f"📁 Instance directory exists: {os.path.exists(os.path.dirname(db_path))}")
-    
-    # Debug: Kiểm tra quyền ghi file
-    try:
-        test_file = os.path.join(os.getcwd(), 'test_write.txt')
-        with open(test_file, 'w') as f:
-            f.write("test")
-        os.remove(test_file)
-        print("✅ Có quyền ghi file")
-    except Exception as e:
-        print(f"❌ Không có quyền ghi file: {e}")
-    
-    # Tạo database
     db.create_all()
     print("✅ Database đã sẵn sàng!")
-    
-    # Debug: Kiểm tra database sau khi tạo
-    try:
-        user_count = User.query.count()
-        print(f"📊 Số lượng users trong database: {user_count}")
-        
-        if user_count > 0:
-            users = User.query.all()
-            print("👥 Danh sách users:")
-            for user in users:
-                print(f"  - ID: {user.id}, Username: {user.username}, Created: {user.created_at}")
-        else:
-            print("👥 Chưa có user nào, tạo admin user mặc định...")
-            admin_user = User(username='admin')
-            admin_user.set_password('admin123')
-            db.session.add(admin_user)
-            db.session.commit()
-            print("✅ Admin user đã được tạo: username=admin, password=admin123")
-            
-            # Tạo thêm demo user
-            demo_user = User(username='demo')
-            demo_user.set_password('demo123')
-            db.session.add(demo_user)
-            db.session.commit()
-            print("✅ Demo user đã được tạo: username=demo, password=demo123")
-            
-        # Debug: Kiểm tra file database
-        if os.path.exists(db_path):
-            file_size = os.path.getsize(db_path)
-            print(f"📁 Database file size: {file_size} bytes")
-            print(f"📁 Database file path: {db_path}")
-        else:
-            print("❌ Database file không tồn tại!")
-            
-    except Exception as e:
-        print(f"❌ Lỗi database: {e}")
-        import traceback
-        traceback.print_exc()
-    
-    print("=" * 60)
-    print("🏁 DATABASE DEBUG HOÀN THÀNH")
-    print("=" * 60)
 
 if __name__ == "__main__":
     # Hugging Face Spaces configuration