Update app.py

app.py

@@ -31,18 +31,23 @@ def generate_diary(emotion, num_samples=1, max_length=100, temperature=0.7):
     return [tokenizer.decode(output_sequence, skip_special_tokens=True) for output_sequence in output]
 
 def main():
+    print("감정 입력을 시작합니다.")
     while True:
         # 사용자로부터 감정 입력 받기
         try:
-            emotion = input("오늘의 감정을 입력하세요 (happy, sad, angry 등): ")
+            emotion = input("오늘의 감정을 입력하세요 (happy, sad, angry 등): ").strip()  # 입력 문자열의 앞뒤 공백 제거
             if emotion.lower() in ['happy', 'sad', 'angry']:
                 break
             else:
                 print("입력된 감정이 잘못되었습니다. 다시 입력해주세요.")
-        except EOFError:
-            print("\n사용자 입력이 종료되었습니다. 프로그램을 종료합니다.")
+        except KeyboardInterrupt:
+            print("\n사용자가 입력을 취소했습니다. 프로그램을 종료합니다.")
             return
         
+        # 빈 문자열이 입력되었을 때
+        if not emotion:
+            print("감정을 입력해주세요.")
+
     # 일기 생성
     diary_entries = generate_diary(emotion)
     # 생성된 일기 출력