added more logs and inital message prompt

app.py

@@ -4,6 +4,11 @@ import os
  
 client = OpenAI(api_key=os.environ['openAIToken'])
 assistantId = os.environ['assistantId']
+initial_message = os.environ['initialMessage']
+
+# client = OpenAI(api_key="sk-proj-4GPZmEnOFsZx0fZeQ9B3T3BlbkFJkXHmGCuW0rGmK0RAavtY")
+# assistantId = "asst_Rce5MptxfyEwp4NjImSDvXGD"
+# initial_message = "Hello. Let's start defining new facility. Also provide options."
 
 def createThread(sessionStorage):
   streaming_thread = client.beta.threads.create()
@@ -11,6 +16,7 @@ def createThread(sessionStorage):
   return sessionStorage
 
 def addMessage(text, threadId):
+  print("User message: ", text)
   return client.beta.threads.messages.create(
     thread_id=threadId,
     role="user",
@@ -24,7 +30,7 @@ def handle_suggestions(string_of_suggestions):
   local_message = None
   parts = string_of_suggestions.split('#s#')
   list_of_suggestions = [x.strip('"') for x in parts[0].strip('][').split('", ') if x.strip('"')]
-  print(list_of_suggestions)
+  list_of_suggestions = [ x for x in list_of_suggestions if x]
   if len(parts) > 1:
     local_message = parts[1]
   return list_of_suggestions, local_message
@@ -45,7 +51,7 @@ CSS ="""
 
 def initiate_chatting(storage):
   threadId = storage[0]
-  addMessage("Hi", threadId)
+  addMessage(initial_message, threadId)
   global list_of_suggestions
   list_of_suggestions = []
   string_of_suggestions = ""
@@ -57,7 +63,6 @@ def initiate_chatting(storage):
     assistant_id=assistantId,
   ) as stream:
     for text in stream.text_deltas:
-      print(text, end="")
       local_message = None
       if "[" in text and is_it_first_response:
         is_loading_suggestions = True
@@ -80,7 +85,6 @@ def initiate_chatting(storage):
 def createDemo():
   with gr.Blocks(css=CSS, fill_height=True) as demo:
     storage = gr.State([])
-    inital_response_message = ''
     chatbot = gr.Chatbot(label="Facility managment bot", elem_id="chatbot")
     with gr.Row():
       for i in range(6):
@@ -96,11 +100,10 @@ def createDemo():
         return "", history + [[user_message, None]]
 
     def respond(chat_history, storage):
-      print("Responding")
       global btn_list
       message = chat_history[-1][0]
       threadId = storage[0]
-      print("THREAD_ID:", threadId)
+      print("Responding for threadId: ", threadId)
       chat_history[-1][1] = ""
       addMessage(message, threadId)
       global list_of_suggestions
@@ -108,30 +111,35 @@ def createDemo():
       string_of_suggestions = ""
       is_loading_suggestions = False
       is_it_first_response = True
-      with client.beta.threads.runs.stream(
-        thread_id=threadId,
-        assistant_id=assistantId,
-      ) as stream:
-        for text in stream.text_deltas:
-          print(text, end="")
-          local_message = None
-          if "[" in text and is_it_first_response:
-            is_loading_suggestions = True
-
-          is_it_first_response = False
-
-          if is_loading_suggestions != True:
-            local_message = text
-          else:
-            string_of_suggestions = string_of_suggestions + text
-            if "#s#" in string_of_suggestions:
-              is_loading_suggestions = False
-              list_of_suggestions, local_message = handle_suggestions(string_of_suggestions)
-          if local_message is not None:
-            chat_history[-1][1] += local_message 
-            yield {chatbot: chat_history}
-
-        stream.until_done()
+      try:
+        with client.beta.threads.runs.stream(
+          thread_id=threadId,
+          assistant_id=assistantId,
+        ) as stream:
+          for text in stream.text_deltas:
+            print(text, end="")
+            local_message = None
+            if "[" in text and is_it_first_response:
+              is_loading_suggestions = True
+
+            is_it_first_response = False
+
+            if is_loading_suggestions != True:
+              local_message = text
+            else:
+              string_of_suggestions = string_of_suggestions + text
+              if "#s#" in string_of_suggestions:
+                is_loading_suggestions = False
+                list_of_suggestions, local_message = handle_suggestions(string_of_suggestions)
+            if local_message is not None:
+              chat_history[-1][1] += local_message 
+              yield {chatbot: chat_history}
+          stream.until_done()
+        print("")
+      except:
+        list_of_suggestions = []
+        chat_history[-1][1] = "Error occured during processing your message. Please try again"
+        yield {chatbot: chat_history}
 
     def update_suggestions():
       global list_of_suggestions