Update app.py

app.py

@@ -151,43 +151,47 @@ def main():
     user_input = st.text_area("Enter a short sentence about your current mood:", "")
     
     if user_input:
-        # Clean the input text (stripping unnecessary spaces, lowercasing)
-        clean_input = user_input.strip().lower()
+        # Display Enter button only after user has entered input
+        enter_button = st.button("Enter")
         
-        # Use the model to predict emotion
-        try:
-            result = emotion_classifier(clean_input)
-            st.write(f"Raw Model Result: {result}")  # Debug output to see raw result
+        if enter_button:
+            # Clean the input text (stripping unnecessary spaces, lowercasing)
+            clean_input = user_input.strip().lower()
             
-            emotion = result[0]['label'].lower()
-            
-            st.subheader(f"Emotion Detected: {emotion.capitalize()}")
-            
-            # Get well-being suggestions based on emotion
-            suggestions = get_well_being_suggestions(emotion)
-            
-            # Display text suggestions
-            st.write(suggestions["text"])
+            # Use the model to predict emotion
+            try:
+                result = emotion_classifier(clean_input)
+                st.write(f"Raw Model Result: {result}")  # Debug output to see raw result
+                
+                emotion = result[0]['label'].lower()
+                
+                st.subheader(f"Emotion Detected: {emotion.capitalize()}")
+                
+                # Get well-being suggestions based on emotion
+                suggestions = get_well_being_suggestions(emotion)
+                
+                # Display text suggestions
+                st.write(suggestions["text"])
 
-            # Display links
-            if suggestions["links"]:
-                st.write("Useful Resources:")
-                for link in suggestions["links"]:
-                    st.markdown(f"[{link}]({link})", unsafe_allow_html=True)
+                # Display links
+                if suggestions["links"]:
+                    st.write("Useful Resources:")
+                    for link in suggestions["links"]:
+                        st.markdown(f"[{link}]({link})", unsafe_allow_html=True)
 
-            # Display video links
-            if suggestions["videos"]:
-                st.write("Relaxation Videos:")
-                for video in suggestions["videos"]:
-                    st.markdown(f"[Watch here]({video})", unsafe_allow_html=True)
+                # Display video links
+                if suggestions["videos"]:
+                    st.write("Relaxation Videos:")
+                    for video in suggestions["videos"]:
+                        st.markdown(f"[Watch here]({video})", unsafe_allow_html=True)
             
-            # Add a button for a summary
-            if st.button('Summary'):
-                st.write(f"Emotion detected: {emotion.capitalize()}. Here are your well-being suggestions to enhance your mood.")
-                st.write("Explore the links and videos to improve your emotional health!")
+                # Add a button for a summary
+                if st.button('Summary'):
+                    st.write(f"Emotion detected: {emotion.capitalize()}. Here are your well-being suggestions to enhance your mood.")
+                    st.write("Explore the links and videos to improve your emotional health!")
             
-        except Exception as e:
-            st.error(f"Error predicting emotion: {str(e)}")
+            except Exception as e:
+                st.error(f"Error predicting emotion: {str(e)}")
 
 # Run the Streamlit app
 if __name__ == "__main__":