Upload 10 files

app.py

@@ -6,13 +6,16 @@ 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
-config_name = os.environ.get('FLASK_ENV', 'default')
-app.config.from_object(config[config_name])
+app.config['SECRET_KEY'] = 'your-secret-key-change-this-in-production'
+# 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
 
 # Khởi tạo extensions
 db.init_app(app)
@@ -29,7 +32,7 @@ def load_user(user_id):
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
 # === Load tokenizer & model ===
-MODEL_REPO = app.config['MODEL_REPO']
+MODEL_REPO = "Ptul2x5/Student_Feedback_Sentiment"  # 🔹 Repo Hugging Face của bạn
 
 print("🔄 Đang tải tokenizer và model từ Hugging Face...")
 tokenizer = AutoTokenizer.from_pretrained(MODEL_REPO, use_fast=False)
@@ -158,8 +161,8 @@ def predict():
             topic = torch.argmax(logits_topic, dim=1).item()
 
         # Mapping
-        SENTIMENT_MAP = app.config['SENTIMENT_MAP']
-        TOPIC_MAP = app.config['TOPIC_MAP']
+        SENTIMENT_MAP = {0: "negative", 1: "neutral", 2: "positive"}
+        TOPIC_MAP = {0: "lecturer", 1: "training_program", 2: "facility", 3: "others"}
         
         sentiment = SENTIMENT_MAP[sent]
         topic_result = TOPIC_MAP[topic]
@@ -201,11 +204,5 @@ with app.app_context():
 
 if __name__ == "__main__":
     # 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)
+    debug = os.environ.get("DEBUG", "False").lower() == "true"
+    app.run(host="0.0.0.0", port=7860, debug=debug)