Update controller.py

controller.py

@@ -2,11 +2,13 @@ import os
 from logger import log_response
 from custom_agent import CustomHfAgent 
 image = []
-def handle_submission(user_message, selected_tools, url_endpoint):
+def handle_submission(user_message, selected_tools, url_endpoint, document, image):
     
     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))
     
     agent = CustomHfAgent(
         url_endpoint=url_endpoint,