mdpg4 / http_client_ir.py
Mahiruoshi's picture
Init
689227c verified
import requests
import PC_CONFIG
class ImageRecognitionClient:
def __init__(self, host, port):
self.server_url = f"http://{host}:{port}"
def check_status(self):
"""Check the status of the server."""
response = requests.get(f"{self.server_url}/status")
if response.status_code == 200:
return response.json()
else:
print(f"Error checking status: {response.status_code}")
return None
def send_file(self, file_path, direction, task_type):
with open(file_path, 'rb') as f:
files = {'file': f}
data = {'direction': direction, 'task_type': task_type}
response = requests.post(f"{self.server_url}/upload", files=files, data=data)
if response.status_code == 200:
print("File sent to the server successfully.")
print(response.json())
return response.json()
else:
print("Failed to send file to the server.")
print(response.text)
return None
def display_stitched(self):
response = requests.post(f"{self.server_url}/display_stitched")
if response.status_code == 200:
print("Stitched request sent to the server successfully.")
print(response.json())
return response.json()
else:
print("Failed to send stitched request to the server.")
print(response.text)
return None
if __name__ == "__main__":
host = '192.168.100.199'
port = '4000'
client = ImageRecognitionClient(host, port)
#client.check_status()
client.send_file('/Users/xot/Documents/UniversityWork/Y4S1/MDP/ImgRecAaron/data/image-rec/sample_images/IMG_9606.jpg', 'north', 'TASK_1') # Adjust the file path and direction as needed
#client.display_stitched()