JAVA/IO - Задача 1

/*Да се напише Java програма која ќе прикаже колкава е просечната големина на датотеките со екстензија .txt во именик зададен како аргумент на командна линија.

        Напомена: Користете ја File класата за пристап до содржината на именикот.

        Решение: Решението од HW01_1.java фајлот го поставувате тука со copy-paste*/

import java.io.File;
import java.nio.file.Path;
import java.nio.file.Paths;
import java.util.Scanner;

public class Main {
    public static void main(String[] args) {



        Scanner scanner = new Scanner(System.in);

        String directory= scanner.nextLine();

        Path dir = Paths.get(directory);

        File folder = new File(dir.toString());

        File [] listOfFiles = folder.listFiles();


        double prosek = 0;
        int brojac =0;

        for (File f:listOfFiles)


        {

            if (f.toString().endsWith(".txt"))
            {
                brojac++;
                prosek+=f.length();

            }
        }

        System.out.println(prosek/brojac);

        //System.out.println(file.isDirectory());





    }
}