Update app.py

app.py

@@ -4,19 +4,23 @@ import requests
 API_URL = "https://api-inference.huggingface.co/models/ProsusAI/finbert"
 headers = {"Authorization": "Bearer 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"}
 
+def show_plot(bot_response):
+    scores = {label_info['label']: label_info['score'] for label_info in bot_response}
+    st.bar_chart(scores)
+
 def query(payload):
     response = requests.post(API_URL, headers=headers, json=payload)
     return response.json()
 
 def display_chat_history(chat_history):
-    for entry in chat_history:
+    for entry in chat_history[::-1]:
         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']}")
+            # for label_info in entry['message'][0]:
+            #     st.write(f"Label: {label_info['label']}, Score: {label_info['score']}")
+            show_plot(entry['message'][0])
 
 
 def display_bot_response(bot_response):
@@ -41,8 +45,9 @@ def main():
             })
 
             # Update the chat history
-            chat_history.append({"speaker": "You", "message": user_input})
+            
             chat_history.append({"speaker": "Chatbot", "message": bot_response})
+            chat_history.append({"speaker": "You", "message": user_input})
 
             # Save the chat history in session state
             st.session_state.chat_history = chat_history