ai base
This commit is contained in:
		
							
								
								
									
										61
									
								
								ros_web_api_bellande_ai_system/CMakeLists.txt
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										61
									
								
								ros_web_api_bellande_ai_system/CMakeLists.txt
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,61 @@ | ||||
| # Copyright (C) 2024 Bellande Robotics Sensors Research Innovation Center, Ronaldson Bellande | ||||
| # | ||||
| # This program is free software: you can redistribute it and/or modify | ||||
| # it under the terms of the GNU General Public License as published by | ||||
| # the Free Software Foundation, either version 3 of the License, or | ||||
| # (at your option) any later version. | ||||
| # | ||||
| # This program is distributed in the hope that it will be useful, | ||||
| # but WITHOUT ANY WARRANTY; without even the implied warranty of | ||||
| # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the | ||||
| # GNU General Public License for more details. | ||||
| # | ||||
| # You should have received a copy of the GNU General Public License | ||||
| # along with this program.  If not, see <https://www.gnu.org/licenses/>. | ||||
|  | ||||
| cmake_minimum_required(VERSION 3.8) | ||||
| project(ros_web_api_bellande_ai_system) | ||||
|  | ||||
| # Find ROS | ||||
| if($ENV{ROS_VERSION} EQUAL 1) | ||||
|   find_package(catkin REQUIRED COMPONENTS | ||||
|     roscpp | ||||
|     rospy | ||||
|   ) | ||||
| else() | ||||
|   find_package(ament_cmake REQUIRED COMPONENTS | ||||
|     rclcpp | ||||
|     rclpy | ||||
|   ) | ||||
| endif() | ||||
|  | ||||
| if($ENV{ROS_VERSION} EQUAL 1) | ||||
|   catkin_package( | ||||
|     INCLUDE_DIRS | ||||
|     LIBRARIES ${PROJECT_NAME} | ||||
|     CATKIN_DEPENDS | ||||
|     roscpp | ||||
|     rospy | ||||
|   ) | ||||
| endif() | ||||
|  | ||||
| # Install Python scripts for both ROS 1 | ||||
| if($ENV{ROS_VERSION} EQUAL 1) | ||||
|   catkin_install_python( | ||||
|     PROGRAMS | ||||
|     src/bellande_ai_system_base.py | ||||
|     DESTINATION ${CATKIN_PACKAGE_BIN_DESTINATION} | ||||
|   ) | ||||
| endif() | ||||
|  | ||||
| # Install Python scripts, configuration files, and launch files | ||||
| if($ENV{ROS_VERSION} EQUAL "1") | ||||
|     install(PROGRAMS src/bellande_ai_system_base.py DESTINATION ${CATKIN_PACKAGE_LIB_DESTINATION}) | ||||
|     install(DIRECTORY config/ DESTINATION ${CATKIN_PACKAGE_SHARE_DESTINATION}/config) | ||||
|     install(DIRECTORY launch/ DESTINATION ${CATKIN_PACKAGE_SHARE_DESTINATION}/launch) | ||||
| elseif($ENV{ROS_VERSION} EQUAL "2") | ||||
|     install(PROGRAMS src/bellande_ai_system_base.py DESTINATION lib/${PROJECT_NAME}) | ||||
|     install(DIRECTORY config/ DESTINATION share/${PROJECT_NAME}/config) | ||||
|     install(DIRECTORY launch/ DESTINATION share/${PROJECT_NAME}/launch) | ||||
|     ament_package() | ||||
| endif() | ||||
							
								
								
									
										10
									
								
								ros_web_api_bellande_ai_system/README.md
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										10
									
								
								ros_web_api_bellande_ai_system/README.md
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,10 @@ | ||||
| # Bellande Web ROS/ROS2 API | ||||
|  | ||||
| # [Bellande Step](https://github.com/RonaldsonBellande/bellande_robots_step) | ||||
|  | ||||
| # ROS1 Launch  | ||||
| - python3 bellande_step_api_2d.launch.py x1:=0 y1:=0 x2:=5 y2:=5 limit:=3 | ||||
| - roslaunch ros_web_api_bellande_step bellande_step_api_2d.launch x1:=0 y1:=0 x2:=5 y2:=5 limit:=3  | ||||
|  | ||||
| # ROS2 Launch | ||||
| - ros2 launch ros_web_api_bellande_step bellande_step_api_2d.launch.py x1:=0 y1:=0 x2:=5 y2:=5 limit:=3 | ||||
							
								
								
									
										7
									
								
								ros_web_api_bellande_ai_system/config/configs.json
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										7
									
								
								ros_web_api_bellande_ai_system/config/configs.json
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,7 @@ | ||||
| { | ||||
|   "url": "https://bellanderoboticssensorsresearchinnovationcenter-kot42qxp.b4a.run", | ||||
|   "endpoint_path": { | ||||
|     "base": "/api/Bellande_AI_System/base" | ||||
|   }, | ||||
|   "Bellande_Framework_Access_Key": "bellande_web_api_opensource" | ||||
| } | ||||
| @@ -0,0 +1,67 @@ | ||||
| # Copyright (C) 2024 Bellande Robotics Sensors Research Innovation Center, Ronaldson Bellande | ||||
| # | ||||
| # This program is free software: you can redistribute it and/or modify | ||||
| # it under the terms of the GNU General Public License as published by | ||||
| # the Free Software Foundation, either version 3 of the License, or | ||||
| # (at your option) any later version. | ||||
| # | ||||
| # This program is distributed in the hope that it will be useful, | ||||
| # but WITHOUT ANY WARRANTY; without even the implied warranty of | ||||
| # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the | ||||
| # GNU General Public License for more details. | ||||
| # | ||||
| # You should have received a copy of the GNU General Public License | ||||
| # along with this program.  If not, see <https://www.gnu.org/licenses/>. | ||||
|  | ||||
| import os | ||||
| import sys | ||||
| import subprocess | ||||
| from launch import LaunchDescription | ||||
| from launch_ros.actions import Node | ||||
| from launch.actions import DeclareLaunchArgument | ||||
| from launch.substitutions import LaunchConfiguration | ||||
|  | ||||
| def ros1_launch_description(): | ||||
|     # Get command-line arguments | ||||
|     args = sys.argv[1:] | ||||
|     # Construct the ROS 1 launch command | ||||
|     roslaunch_command = ["roslaunch", "ros_web_api_bellande_ai_system", "bellande_ai_system_base.launch"] + args | ||||
|     # Execute the launch command | ||||
|     subprocess.call(roslaunch_command) | ||||
|  | ||||
| def ros2_launch_description(): | ||||
|     # Declare launch arguments | ||||
|     config_file_arg = DeclareLaunchArgument( | ||||
|         'config_file', | ||||
|         default_value='$(find ros_web_api_bellande_ai_system)/config/configs.json', | ||||
|         description='Path to the AI system configuration file' | ||||
|     ) | ||||
|  | ||||
|     # Create a list to hold all nodes to be launched | ||||
|     nodes_to_launch = [] | ||||
|  | ||||
|     # ROS2 specific configurations | ||||
|     ros_launch_arguments = [config_file_arg] | ||||
|  | ||||
|     nodes_to_launch.append(Node( | ||||
|         package='ros_web_api_bellande_ai_system', | ||||
|         executable='bellande_ai_system_base.py', | ||||
|         name='bellande_ai_system_base_node', | ||||
|         output='screen', | ||||
|         parameters=[ | ||||
|             {'config_file': LaunchConfiguration('config_file')}, | ||||
|         ], | ||||
|     )) | ||||
|  | ||||
|     # Return the LaunchDescription containing all nodes and arguments | ||||
|     return LaunchDescription(ros_launch_arguments + nodes_to_launch) | ||||
|  | ||||
| if __name__ == "__main__": | ||||
|     ros_version = os.getenv("ROS_VERSION") | ||||
|     if ros_version == "1": | ||||
|         ros1_launch_description() | ||||
|     elif ros_version == "2": | ||||
|         ros2_launch_description() | ||||
|     else: | ||||
|         print("Unsupported ROS version. Please set the ROS_VERSION environment variable to '1' for ROS 1 or '2' for ROS 2.") | ||||
|         sys.exit(1) | ||||
| @@ -0,0 +1,29 @@ | ||||
| <?xml version="1.0"?> | ||||
| <!-- | ||||
| Copyright (C) 2024 Bellande Robotics Sensors Research Innovation Center, Ronaldson Bellande | ||||
|  | ||||
| Licensed under the Apache License, Version 2.0 (the "License"); you may not | ||||
| use this file except in compliance with the License. You may obtain a copy of | ||||
| the License at | ||||
|  | ||||
| http://www.apache.org/licenses/LICENSE-2.0 | ||||
|  | ||||
| Unless required by applicable law or agreed to in writing, software | ||||
| distributed under the License is distributed on an "AS IS" BASIS, WITHOUT | ||||
| WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the | ||||
| License for the specific language governing permissions and limitations under | ||||
| the License. | ||||
| --> | ||||
| <?xml version="1.0"?> | ||||
| <launch> | ||||
|   <!-- Load configuration file --> | ||||
|   <arg name="config_file" default="$(find ros_web_api_bellande_ai_system)/config/configs.json" /> | ||||
|   <param name="ai_system_config" textfile="$(arg config_file)" /> | ||||
|  | ||||
|   <!-- Launch the AI System base node --> | ||||
|   <node name="bellande_ai_system_base_node" pkg="ros_web_api_bellande_ai_system" | ||||
|     type="bellande_ai_system_base.py" output="screen"> | ||||
|     <param name="config_file" value="$(arg config_file)" /> | ||||
|   </node> | ||||
|  | ||||
| </launch> | ||||
							
								
								
									
										53
									
								
								ros_web_api_bellande_ai_system/package.xml
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										53
									
								
								ros_web_api_bellande_ai_system/package.xml
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,53 @@ | ||||
| <?xml version="1.0"?> | ||||
| <!-- | ||||
| Copyright (C) 2024 Bellande Robotics Sensors Research Innovation Center, Ronaldson Bellande | ||||
|  | ||||
| This program is free software: you can redistribute it and/or modify | ||||
| it under the terms of the GNU General Public License as published by | ||||
| the Free Software Foundation, either version 3 of the License, or | ||||
| (at your option) any later version. | ||||
|  | ||||
| This program is distributed in the hope that it will be useful, | ||||
| but WITHOUT ANY WARRANTY; without even the implied warranty of | ||||
| MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the | ||||
| GNU General Public License for more details. | ||||
|  | ||||
| You should have received a copy of the GNU General Public License | ||||
| along with this program.  If not, see <https://www.gnu.org/licenses/>. | ||||
| --> | ||||
| <package format="3"> | ||||
|   <name>ros_web_api_bellande_ai_system</name> | ||||
|   <version>0.0.1</version> | ||||
|   <description>Bellande ROS/ROS2 package with a JSON config file for making HTTP requests</description> | ||||
|   <author email="ronaldsonbellande@gmail.com">Ronaldson Bellande</author> | ||||
|   <maintainer email="ronaldsonbellande@gmail.com">Ronaldson Bellande</maintainer> | ||||
|   <license>GNU General Public License v3.0</license> | ||||
|  | ||||
|  | ||||
|   <buildtool_depend condition="$ROS_VERSION == 1">catkin</buildtool_depend> | ||||
|  | ||||
|   <build_depend condition="$ROS_VERSION == 1">roscpp</build_depend> | ||||
|   <build_depend condition="$ROS_VERSION == 1">rospy</build_depend> | ||||
|  | ||||
|   <build_export_depend condition="$ROS_VERSION == 1">roscpp</build_export_depend> | ||||
|   <build_export_depend condition="$ROS_VERSION == 1">rospy</build_export_depend> | ||||
|  | ||||
|   <exec_depend condition="$ROS_VERSION == 1">roscpp</exec_depend> | ||||
|   <exec_depend condition="$ROS_VERSION == 1">rospy</exec_depend> | ||||
|  | ||||
|  | ||||
|   <buildtool_depend condition="$ROS_VERSION == 2">ament_cmake</buildtool_depend> | ||||
|  | ||||
|   <build_depend condition="$ROS_VERSION == 2">rclcpp</build_depend> | ||||
|   <build_depend condition="$ROS_VERSION == 2">rclpy</build_depend> | ||||
|  | ||||
|   <build_export_depend condition="$ROS_VERSION == 2">rclcpp</build_export_depend> | ||||
|   <build_export_depend condition="$ROS_VERSION == 2">rclpy</build_export_depend> | ||||
|  | ||||
|   <exec_depend condition="$ROS_VERSION == 2">rclcpp</exec_depend> | ||||
|   <exec_depend condition="$ROS_VERSION == 2">rclpy</exec_depend> | ||||
|  | ||||
|  | ||||
|   <export> | ||||
|   </export> | ||||
| </package> | ||||
							
								
								
									
										11
									
								
								ros_web_api_bellande_ai_system/setup.py
									
									
									
									
									
										Executable file
									
								
							
							
						
						
									
										11
									
								
								ros_web_api_bellande_ai_system/setup.py
									
									
									
									
									
										Executable file
									
								
							| @@ -0,0 +1,11 @@ | ||||
| from distutils.core import setup | ||||
| from catkin_pkg.python_setup import generate_distutils_setup | ||||
|  | ||||
| # fetch values from package.xml | ||||
| setup_args = generate_distutils_setup( | ||||
|     scripts=['src/bellande_ai_system_base.py'], | ||||
|     packages=['ros_web_api_bellande_ai_system'], | ||||
|     package_dir={'': 'src'}, | ||||
| ) | ||||
|  | ||||
| setup(**setup_args) | ||||
							
								
								
									
										96
									
								
								ros_web_api_bellande_ai_system/src/bellande_ai_system_base.py
									
									
									
									
									
										Executable file
									
								
							
							
						
						
									
										96
									
								
								ros_web_api_bellande_ai_system/src/bellande_ai_system_base.py
									
									
									
									
									
										Executable file
									
								
							| @@ -0,0 +1,96 @@ | ||||
| # Copyright (C) 2024 Bellande Robotics Sensors Research Innovation Center, Ronaldson Bellande | ||||
| # | ||||
| # This program is free software: you can redistribute it and/or modify | ||||
| # it under the terms of the GNU General Public License as published by | ||||
| # the Free Software Foundation, either version 3 of the License, or | ||||
| # (at your option) any later version. | ||||
| # | ||||
| # This program is distributed in the hope that it will be useful, | ||||
| # but WITHOUT ANY WARRANTY; without even the implied warranty of | ||||
| # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the | ||||
| # GNU General Public License for more details. | ||||
| # | ||||
| # You should have received a copy of the GNU General Public License | ||||
| # along with this program.  If not, see <https://www.gnu.org/licenses/>. | ||||
|  | ||||
| import json | ||||
| import os | ||||
| import requests | ||||
| from std_msgs.msg import String | ||||
|  | ||||
| def get_ai_response(input_text): | ||||
|     payload = { | ||||
|         "input_text": input_text | ||||
|     } | ||||
|     headers = { | ||||
|         'accept': 'application/json', | ||||
|         'Content-Type': 'application/json', | ||||
|         "Authorization": f"Bearer {api_access_key}" | ||||
|     } | ||||
|     try: | ||||
|         response = requests.post( | ||||
|             api_url, | ||||
|             json=payload, | ||||
|             headers=headers | ||||
|         ) | ||||
|         response.raise_for_status() | ||||
|         data = response.json() | ||||
|         return String(f"AI Response: {data['response']}") | ||||
|     except requests.exceptions.RequestException as e: | ||||
|         print(f"Error: {e}") | ||||
|         return None | ||||
|  | ||||
| def input_callback(msg): | ||||
|     ai_response = get_ai_response(msg.data) | ||||
|     if ai_response: | ||||
|         pub.publish(ai_response) | ||||
|  | ||||
| def main(): | ||||
|     global api_url, api_access_key, pub | ||||
|     config_file_path = os.path.join(os.path.dirname(__file__), '../config/ai_system.json') | ||||
|      | ||||
|     if not os.path.exists(config_file_path): | ||||
|         print("Config file not found:", config_file_path) | ||||
|         return | ||||
|      | ||||
|     with open(config_file_path, 'r') as config_file: | ||||
|         config = json.load(config_file) | ||||
|         url = config['url'] | ||||
|         endpoint_path = config['endpoint_path']["base"] | ||||
|         api_access_key = config["Bellande_Framework_Access_Key"] | ||||
|      | ||||
|     # Initialize ROS node | ||||
|     if ros_version == "1": | ||||
|         rospy.init_node('ai_system_node', anonymous=True) | ||||
|         pub = rospy.Publisher('ai_response', String, queue_size=10) | ||||
|         sub = rospy.Subscriber('ai_input', String, input_callback) | ||||
|     elif ros_version == "2": | ||||
|         rclpy.init() | ||||
|         node = rclpy.create_node('ai_system_node') | ||||
|         pub = node.create_publisher(String, 'ai_response', 10) | ||||
|         sub = node.create_subscription(String, 'ai_input', input_callback, 10) | ||||
|      | ||||
|     api_url = f"{url}{endpoint_path}" | ||||
|      | ||||
|     try: | ||||
|         print("AI system node is running. Ctrl+C to exit.") | ||||
|         if ros_version == "1": | ||||
|             rospy.spin() | ||||
|         elif ros_version == "2": | ||||
|             rclpy.spin(node) | ||||
|     except KeyboardInterrupt: | ||||
|         print("Shutting down AI system node.") | ||||
|     except Exception as e: | ||||
|         print(f"An error occurred: {str(e)}") | ||||
|     finally: | ||||
|         if ros_version == "2": | ||||
|             node.destroy_node() | ||||
|             rclpy.shutdown() | ||||
|  | ||||
| if __name__ == '__main__': | ||||
|     ros_version = os.getenv("ROS_VERSION") | ||||
|     if ros_version == "1": | ||||
|         import rospy | ||||
|     elif ros_version == "2": | ||||
|         import rclpy | ||||
|     main() | ||||
		Reference in New Issue
	
	Block a user