Create controller.py

controller.py

@@ -0,0 +1,23 @@
+from logger import log_response
+from custom_agent import CustomHfAgent 
+
+def handle_submission(user_message, selected_tools, url_endpoint):
+    
+    log_response("User input \n {}".format(user_message))
+    log_response("selected_tools \n {}".format(selected_tools))
+    log_response("url_endpoint \n {}".format(url_endpoint))
+    
+    agent = CustomHfAgent(
+        url_endpoint=url_endpoint,
+        token=os.environ['HF_token'],
+        additional_tools=selected_tools,
+        input_params={"max_new_tokens": 192},
+    )
+
+    response = agent.run(user_message)
+
+    log_response("Agent Response\n {}".format(response))
+
+    return response
+
+