Update app.py

app.py

@@ -108,8 +108,13 @@ def chat(message, history, state):
 tokenizer = AutoTokenizer.from_pretrained("cardiffnlp/twitter-roberta-base-sentiment")
 sentiment_model = AutoModelForSequenceClassification.from_pretrained("cardiffnlp/twitter-roberta-base-sentiment")
 
+# Function for sentiment analysis
 # Function for sentiment analysis
 def analyze_sentiment(text, state):
+    # Initialize state if it's None
+    if state is None:
+        state = {'step': 1}
+    
     inputs = tokenizer(text, return_tensors="pt")
     with torch.no_grad():
         outputs = sentiment_model(**inputs)
@@ -224,9 +229,10 @@ def get_all_places(query, location, radius, api_key):
     return []
 
 # Gradio UI components
+# Function to create the UI with state initialization
 def create_ui():
     with gr.Blocks() as demo:
-        state = gr.State()
+        state = gr.State({'step': 1})  # Initialize state with a default value
         chatbot = gr.Chatbot(elem_id="chatbot", label="Mental Health Chatbot")
         message_input = gr.Textbox(placeholder="Ask me something...", label="Enter your message")
         sentiment_output = gr.Textbox(placeholder="Sentiment result", label="Sentiment")