Update app.py

app.py

@@ -2,7 +2,6 @@ import streamlit as st
 import requests
 import os
 from dotenv import load_dotenv
-from requests.exceptions import RequestException
 from nltk.sentiment import SentimentIntensityAnalyzer
 import nltk
 
@@ -33,6 +32,10 @@ model_pre_instructions = {
 # Initialize sentiment analyzer
 sia = SentimentIntensityAnalyzer()
 
+# Initialize session state
+if 'messages' not in st.session_state:
+    st.session_state.messages = []
+
 # Function to predict diagnosis based on sentiment analysis and chat content
 def predict_diagnosis(messages):
     # Perform sentiment analysis
@@ -80,10 +83,14 @@ if st.session_state.get('selected_model') != selected_model:
     st.info(f"You have switched to {selected_model}.")
 st.sidebar.button('Reset Chat', on_click=reset_conversation)
 
-# Display diagnosis based on chat messages
-diagnosis = predict_diagnosis(st.session_state.messages)
-st.sidebar.subheader("User Diagnosis")
-st.sidebar.write(", ".join(diagnosis))
+# Display diagnosis based on chat messages if messages exist
+if 'messages' in st.session_state:
+    diagnosis = predict_diagnosis(st.session_state.messages)
+    st.sidebar.subheader("User Diagnosis")
+    st.sidebar.write(", ".join(diagnosis))
+else:
+    st.sidebar.subheader("User Diagnosis")
+    st.sidebar.write("No messages yet")
 
 # Add logo and text to the sidebar
 st.sidebar.image("https://assets.isu.pub/document-structure/221118065013-a6029cf3d563afaf9b946bb9497d45d4/v1/2841525b232adaef7bd0efe1da81a4c5.jpeg", width=200)