Update controller.py

controller.py

@@ -31,7 +31,16 @@ def handle_submission(user_message, selected_tools, url_endpoint, document, imag
     log_response("Agent Response\n {}".format(response))
 
     return response
-    
+
+def cut_text_after_keyword(text, keyword):
+    index = text.find(keyword)
+    if index != -1:
+        return text[:index].strip()
+    return text
+
+
+
+
 def handle_submission_chat(user_message, response):
     # os.environ['HUGGINGFACEHUB_API_TOKEN'] = os.environ['HF_token']
     agent_chat_bot = get_conversation_chain()
@@ -41,8 +50,10 @@ def handle_submission_chat(user_message, response):
     else:
         text = agent_chat_bot.predict(input=user_message)
 
-    print(text)
-    return text
+    result = cut_text_after_keyword(text, "Human:")
+    print(result)
+
+    return result
 
  
 def get_conversation_chain( ):