Update app.py

app.py

@@ -1,8 +1,9 @@
 import streamlit as st
 import requests
-
+import os
+SECRET_TOKEN = os.getenv("SECRET_TOKEN")
 API_URL = "https://api-inference.huggingface.co/models/ProsusAI/finbert"
-headers = {"Authorization": "Bearer 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"}
+headers = {"Authorization": f:"Bearer {SECRET_TOKEN}"}
 
 def show_plot(bot_response):
     scores = {label_info['label']: label_info['score'] for label_info in bot_response}
@@ -17,10 +18,14 @@ def display_chat_history(chat_history):
         if entry["speaker"] == "You":
             st.text(f"You: {entry['message']}")
         elif entry["speaker"] == "Chatbot":
-            st.text(f"Chatbot:")
-            # for label_info in entry['message'][0]:
-            #     st.write(f"Label: {label_info['label']}, Score: {label_info['score']}")
-            show_plot(entry['message'][0])
+            try:
+                st.text(f"Chatbot:")
+                # for label_info in entry['message'][0]:
+                #     st.write(f"Label: {label_info['label']}, Score: {label_info['score']}")
+                print(entry['message'])
+                show_plot(entry['message'][0])
+            except:
+                st.text(f"Something went wrong.")
 
 
 def display_bot_response(bot_response):
@@ -60,4 +65,4 @@ def main():
     #     display_bot_response(chat_history[-1]["message"])
 
 if __name__ == "__main__":
-    main()
+    main()