Update app.py

app.py

@@ -1,11 +1,19 @@
+# ==============================
+# app.py
+# KoBERT 감정 분석 + 앵무새 반응 + 노래 추천
+# ==============================
+
 from flask import Flask, render_template, request, session
-from kobert_model import predict_emotion
-from spotify_api import search_spotify
+import torch
+from kobert_model import predict_emotion   # 파인튜닝한 KoBERT 모델
+from spotify_api import search_spotify      # Spotify 노래 추천 API
 
 app = Flask(__name__)
-app.secret_key = "your_secret_key"
+app.secret_key = "yangtal12345"  # 💡 변경 가능
 
-# 감정별 앵무새 이미지
+# ==============================
+# 앵무새 이미지 & 반응
+# ==============================
 parrot_images = {
     "행복": "parrot-happy.png",
     "슬픔": "parrot-sad.png",
@@ -16,21 +24,24 @@ parrot_images = {
     "놀람": "parrot-surprise.png",
 }
 
-# 감정별 앵무새 반응 문구
 parrot_responses = {
     "행복": "와! 당신이 행복해서 저도 기분 좋아요! 🎉",
     "슬픔": "슬프다니, 제가 옆에서 힘이 되어줄게요... 🥺",
-    "분노": "화났군요! 앵무새가 같이 심호흡 해볼까요? 😤",
+    "분노": "화났군요! 저랑 같이 심호흡 해볼까요? 😤",
     "중립": "괜찮아요, 평온한 하루도 필요하죠.",
     "혐오": "그런 감정도 있어요. 힘내요!",
     "공포": "겁나는 상황이군요. 제가 지켜드릴게요!",
     "놀람": "우와, 놀랐나요? 저도 깜짝 놀랐어요! 😲",
 }
 
+# ==============================
+# 라우트
+# ==============================
 @app.route("/")
 def index():
     return render_template("index.html")
 
+
 @app.route("/chat", methods=["GET", "POST"])
 def chat():
     if "chat_history" not in session:
@@ -44,7 +55,7 @@ def chat():
         parrot_img = parrot_images.get(emotion, "parrot-neutral.png")
         parrot_reply = parrot_responses.get(emotion, "제가 듣고 있어요.")
 
-        # 노래 추천 여부
+        # 노래 추천
         songs = None
         if "노래추천해줘" in user_message or "노래 추천" in user_message:
             songs = search_spotify(emotion)
@@ -66,10 +77,12 @@ def chat():
     else:
         return render_template("chat.html", chat_history=session.get("chat_history", []))
 
+
 @app.route("/reset")
 def reset():
     session.pop("chat_history", None)
     return render_template("index.html")
 
+
 if __name__ == "__main__":
     app.run(host="0.0.0.0", port=5000, debug=True)