File size: 1,906 Bytes
6220346 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 |
import os
from managers.file_manager import FileManager
from tools.chess_board_recognition_tool import ChessBoardRecognitionTool
from tools.convert_audio_to_text_tool import ConvertAudioToTextTool
from tools.convert_image_to_text_tool import ConvertImageToTextTool
from tools.fetch_url_content_tool import FetchURLContentTool
def test_fetch_url():
print("Test FetchURLContentTool...")
tool = FetchURLContentTool()
url = "https://upload.wikimedia.org/wikipedia/commons/3/3c/Shaki_waterfall.jpg" # immagine piccola
path = tool.forward(url)
print(f"Downloaded in: {path}")
FileManager.cleanup_file(path)
def test_transcribe_audio():
print("Test TranscribeAudioTool...")
tool = ConvertAudioToTextTool()
sample_audio = "data/sample_audio.mp3"
if not os.path.exists(sample_audio):
print("File not found: data/sample_audio.mp3")
return
text = tool.forward(sample_audio)
print(f"Result:\n{text}")
def test_transcribe_image():
print("Test TranscribeImageTool...")
tool = ConvertImageToTextTool()
sample_audio = "data/sample_image.jpg"
if not os.path.exists(sample_audio):
print("File not found: data/sample_image.jpg")
return
text = tool.forward(sample_audio)
print(f"Result:\n{text}")
def test_chess_board_recognition_image():
print("Test CaptionImageTool...")
tool = ChessBoardRecognitionTool()
sample_image = "data/sample_image.jpg"
if not os.path.exists(sample_image):
print("File not found: data/sample_image.jpg")
return
caption = tool.forward(sample_image)
print(f"Result:\n{caption}")
def run_all_tests():
print("\n--- START TEST ---\n")
test_fetch_url()
test_transcribe_audio()
test_transcribe_image()
# test_chess_board_recognition_image()
print("\n--- ALL TESTS COMPLETED ---\n")
if __name__ == "__main__":
run_all_tests()
|