Update controller.py

controller.py

@@ -2,14 +2,14 @@ import os
 from logger import log_response
 from custom_agent import CustomHfAgent 
 image = []
-def handle_submission(user_message, selected_tools, url_endpoint, document, image):
+def handle_submission(user_message, selected_tools, url_endpoint, document, image, context):
     
     log_response("User input \n {}".format(user_message))
     log_response("selected_tools \n {}".format(selected_tools))
     log_response("url_endpoint \n {}".format(url_endpoint))
-    log_response("url_endpoint \n {}".format(document))
-    log_response("url_endpoint \n {}".format(image))
-    image = image
+    log_response("document \n {}".format(document))
+    log_response("image \n {}".format(image))
+    log_response("context \n {}".format(context))
     
     agent = CustomHfAgent(
         url_endpoint=url_endpoint,
@@ -18,7 +18,7 @@ def handle_submission(user_message, selected_tools, url_endpoint, document, imag
         input_params={"max_new_tokens": 192},
     )
 
-    response = agent.chat(user_message,document=document,image=image)
+    response = agent.chat(user_message,document=document,image=image, context = context)
 
     log_response("Agent Response\n {}".format(response))