Update app.py

app.py

@@ -75,4 +75,27 @@ def main():
         
         # Map emotion label from the model to our suggestion database
         if emotion == "LABEL_0":
- 
+            emotion = "NEGATIVE"
+        elif emotion == "LABEL_1":
+            emotion = "POSITIVE"
+        else:
+            emotion = "NEUTRAL"
+        
+        st.write(f"Emotion detected: {emotion}")
+        
+        # Step 3: Suggest activities, articles, and videos
+        resources = suggest_activity({emotion: 1})
+        st.write("Suggestions:")
+        for suggestion in resources["suggestions"]:
+            st.write(f"- {suggestion}")
+        
+        st.write("Articles:")
+        for article in resources["articles"]:
+            st.write(f"- [{article['title']}]({article['url']})")
+        
+        st.write("Videos:")
+        for video in resources["videos"]:
+            st.write(f"- [{video['title']}]({video['url']})")
+
+if __name__ == "__main__":
+    main()