Update controller.py

controller.py

@@ -31,13 +31,19 @@ def handle_submission(user_message, selected_tools, url_endpoint, document, imag
     log_response("Agent Response\n {}".format(response))
 
     return response
-
+    
 def handle_submission_chat(user_message, response):
-    #os.environ['HUGGINGFACEHUB_API_TOKEN'] = os.environ['HF_token']
-    agent_chat_bot =get_conversation_chain()
-    text = agent_chat_bot.predict(input=user_message+response)
+    # os.environ['HUGGINGFACEHUB_API_TOKEN'] = os.environ['HF_token']
+    agent_chat_bot = get_conversation_chain()
+
+    if response is not None:
+        text = agent_chat_bot.predict(input=user_message + response)
+    else:
+        text = agent_chat_bot.predict(input=user_message)
+
     print(text)
     return text
+
  
 def get_conversation_chain( ):
     """