Update main.py

main.py

@@ -57,7 +57,7 @@ if all([
     st.session_state.untrue_detector_probability = 1
     
     st.session_state.bert_disinfo_result = [{'label': '', 'score': 1}]
-        
+    st.session_state.emotions_result = [{'label': '', 'score': 1}]    
 
 content = load_content()
 if 'loaded' not in st.session_state:
@@ -102,7 +102,7 @@ if st.session_state.agree:
             st.session_state.untrue_detector_probability = max(st.session_state.untrue_detector_probability[0], st.session_state.untrue_detector_probability[1]) 
 
             st.session_state.bert_disinfo_result = st.session_state.bert_disinfo(user_input)
-            st.session_state.bert_disinfo_result = st.session_state.emotions(user_input)
+            st.session_state.emotions_result = st.session_state.emotions(user_input)
 
             
 
@@ -133,18 +133,18 @@ if st.session_state.agree:
                        str(round(st.session_state.bert_disinfo_result[0]['score'] * 100, 2)) +
                        content['bert_no_2'][st.session_state.lang], icon="✅")
 
-        if st.session_state.emotions[0]['score'] < 0.97:
+        if st.session_state.emotions_result[0]['score'] < 0.97:
             st.warning(content['emotions_label_1'][st.session_state.lang] + 
-                       str(st.session_state.emotions[0]['label']) + 
+                       str(st.session_state.emotions_result[0]['label']) + 
                        content['emotions_label_2'][st.session_state.lang] 
-                       str(round(st.session_state.emotions[0]['score'] * 100, 2)) + 
+                       str(round(st.session_state.emotions_result[0]['score'] * 100, 2)) + 
                        content['emotions_label_3'][st.session_state.lang] +
                        content['emotions_label_4'][st.session_state.lang], icon = "⚠️")
         else:
             st.success(content['emotions_label_1'][st.session_state.lang] + 
-                       str(st.session_state.emotions[0]['label']) + 
+                       str(st.session_state.emotions_result[0]['label']) + 
                        content['emotions_label_2'][st.session_state.lang] 
-                       str(round(st.session_state.emotions[0]['score'] * 100, 2)) + 
+                       str(round(st.session_state.emotions_result[0]['score'] * 100, 2)) + 
                        content['emotions_label_3'][st.session_state.lang], icon="✅")