Update app.py

app.py

@@ -1,28 +1,21 @@
-# app.py
+import os
+if not os.path.exists('emotion_transformer_model.pth'):
+    from src.train_transformer import train
+    train()
 
+# 2) Streamlit UI
+import streamlit as st
 from textblob import TextBlob
 from src.inference import predict
-from src.responses import get_response
+from src.responses import get_response, correct_spelling, neg_inputs
 
-print("EmotiBot 🌿: Hi! How are you feeling today? (Type 'exit' to quit)")
-
-while True:
-    user = input("You: ").strip()
-    if user.lower() in ['exit','quit']:
-        print("EmotiBot 🌿: Take care! I’m here whenever you want to talk.")
-        break
-    # optional spelling fix:
-    user = str(TextBlob(user).correct())
-
-    # if any negative phrase → force sadness,
-    # else run model inference
-    if any(phrase in user.lower() for phrase in negative_inputs):
-        emotion = "sadness"
-    else:
-        emotion = predict(user)
-
-    reply, done = get_response(emotion, user)
-    print(f"EmotiBot 🌿: {reply}")
+st.title("🌿 EmotiBot")
+user = st.text_input("You:")
+if st.button("Send") and user:
+    txt = correct_spelling(user)
+    emo = 'sadness' if any(p in txt.lower() for p in neg_inputs) else predict(txt)
+    reply, done = get_response(emo, txt)
+    st.markdown(f"**Detected emotion:** {emo}")
+    st.write(reply)
     if done:
-        break
-
+        st.write("Take care!")