| import requests
|
| import os
|
|
|
| def test_vision_api(image_path, prompt, api_url, output_json_path):
|
| with open(image_path, "rb") as img_file:
|
| files = {"image": img_file}
|
| data = {"prompt": prompt}
|
| response = requests.post(api_url, files=files, data=data)
|
| try:
|
| result = response.json()
|
| except Exception as e:
|
| print("Non-JSON response from API:")
|
| print(response.text)
|
| result = {"error": str(e), "raw_response": response.text}
|
| with open(output_json_path, "w") as f:
|
| import json
|
| json.dump({"frame": os.path.basename(image_path), "result": result}, f, indent=2)
|
| print(f"Logged response to {output_json_path}")
|
|
|
| if __name__ == "__main__":
|
|
|
| test_vision_api(
|
| image_path="frames/sample_task_0001.png",
|
| prompt="Describe the user's task, app, actions, and the final goal.",
|
| api_url="http://localhost:8000/vision",
|
| output_json_path="annotations/test_vision_api_response.json"
|
| )
|
|
|
