Upload 11 files

app.py

@@ -6,13 +6,13 @@ from flask_login import LoginManager, login_user, logout_user, login_required, c
 from models import db, User, Feedback
 from forms import RegistrationForm, LoginForm
 from PhoBERTMultiTask import PhoBERTMultiTask
+from config import config
 
 app = Flask(__name__)
 
 # Cấu hình
-app.config['SECRET_KEY'] = 'your-secret-key-change-this-in-production'
-app.config['SQLALCHEMY_DATABASE_URI'] = 'sqlite:///feedback_analysis.db'
-app.config['SQLALCHEMY_TRACK_MODIFICATIONS'] = False
+config_name = os.environ.get('FLASK_ENV', 'default')
+app.config.from_object(config[config_name])
 
 # Khởi tạo extensions
 db.init_app(app)
@@ -29,7 +29,7 @@ def load_user(user_id):
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
 # === Load tokenizer & model ===
-MODEL_REPO = "Ptul2x5/Student_Feedback_Sentiment"  # 🔹 Repo Hugging Face của bạn
+MODEL_REPO = app.config['MODEL_REPO']
 
 print("🔄 Đang tải tokenizer và model từ Hugging Face...")
 tokenizer = AutoTokenizer.from_pretrained(MODEL_REPO, use_fast=False)
@@ -158,8 +158,8 @@ def predict():
             topic = torch.argmax(logits_topic, dim=1).item()
 
         # Mapping
-        SENTIMENT_MAP = {0: "negative", 1: "neutral", 2: "positive"}
-        TOPIC_MAP = {0: "lecturer", 1: "training_program", 2: "facility", 3: "others"}
+        SENTIMENT_MAP = app.config['SENTIMENT_MAP']
+        TOPIC_MAP = app.config['TOPIC_MAP']
         
         sentiment = SENTIMENT_MAP[sent]
         topic_result = TOPIC_MAP[topic]
@@ -200,5 +200,12 @@ with app.app_context():
     print("✅ Database đã sẵn sàng!")
 
 if __name__ == "__main__":
-    # Hugging Face luôn yêu cầu port = 7860
-    app.run(host="0.0.0.0", port=7860, debug=True)
+    # Hugging Face Spaces configuration
+    port = int(os.environ.get("PORT", 7860))
+    debug = os.environ.get("FLASK_ENV") != "production"
+    
+    print(f"🚀 Starting Flask app on port {port}")
+    print(f"🔧 Debug mode: {debug}")
+    print(f"🌐 Environment: {os.environ.get('FLASK_ENV', 'default')}")
+    
+    app.run(host="0.0.0.0", port=port, debug=debug)

config.py

@@ -0,0 +1,31 @@
+import os
+
+class Config:
+    """Cấu hình cơ bản"""
+    SECRET_KEY = os.environ.get('SECRET_KEY') or 'your-secret-key-change-this-in-production'
+    SQLALCHEMY_DATABASE_URI = os.environ.get('DATABASE_URL') or 'sqlite:///feedback_analysis.db'
+    SQLALCHEMY_TRACK_MODIFICATIONS = False
+    
+    # Hugging Face Spaces configuration
+    MODEL_REPO = "Ptul2x5/Student_Feedback_Sentiment"
+    
+    # Sentiment và Topic mapping
+    SENTIMENT_MAP = {0: "negative", 1: "neutral", 2: "positive"}
+    TOPIC_MAP = {0: "lecturer", 1: "training_program", 2: "facility", 3: "others"}
+
+class ProductionConfig(Config):
+    """Cấu hình cho production"""
+    SECRET_KEY = os.environ.get('SECRET_KEY')
+    if not SECRET_KEY:
+        raise ValueError("SECRET_KEY environment variable must be set in production")
+
+class DevelopmentConfig(Config):
+    """Cấu hình cho development"""
+    DEBUG = True
+
+# Chọn config dựa trên environment
+config = {
+    'development': DevelopmentConfig,
+    'production': ProductionConfig,
+    'default': DevelopmentConfig
+}