support for image from url

Author: yashschandra

run.go

@@ -12,14 +12,20 @@ func getRunCommand() *cobra.Command {
 		Short: "Upscayl single image using command line options",
 		Run: func(cmd *cobra.Command, args []string) {
 			image, _ := cmd.Flags().GetString("input")
-			outputPath, err := upscayl.Upscayl(upscayl.Input{ImagePath: image})
+			url, _ := cmd.Flags().GetString("url")
+			input := upscayl.Input{
+				ImagePath: image,
+				ImageURL:  url,
+			}
+			outputPath, err := upscayl.Upscayl(input)
 			if err != nil {
 				log.Fatal("error while upscayling", err.Error())
 			}
 			log.Println("output image at", outputPath)
 		},
 	}
 	cmd.Flags().StringP("input", "i", "", "Input image path (jpg/png/webp) or directory")
+	cmd.Flags().StringP("url", "u", "", "Input image url (jpg/png/webp)")
 	cmd.Flags().StringP("output", "o", "", "Output image path (jpg/png/webp) or directory")
 	cmd.Flags().IntP("model-scale", "z", 4, "Scale according to the model (can be 2, 3, 4)")
 	cmd.Flags().IntP("output-scale", "s", 4, "Custom output scale (can be 2, 3, 4)")
@@ -34,6 +40,5 @@ func getRunCommand() *cobra.Command {
 	cmd.Flags().BoolP("tta", "x", false, "Enable TTA mode")
 	cmd.Flags().StringP("format", "f", "ext/png", "Output image format (jpg/png/webp)")
 	cmd.Flags().BoolP("verbose", "v", false, "Verbose output")
-	_ = cmd.MarkFlagRequired("input")
 	return cmd
 }

upscayl/upscayl.go

@@ -1,13 +1,16 @@
 package upscayl
 
 import (
+	"errors"
 	"fmt"
 	"io"
 	"log"
 	"net/http"
+	"net/url"
 	"os"
 	"os/exec"
 	"os/user"
+	"path"
 	"path/filepath"
 	"runtime"
 	"strings"
@@ -106,6 +109,7 @@ func init() {
 
 type Input struct {
 	ImagePath      string `json:"imagePath"`
+	ImageURL       string `json:"imageUrl"`
 	OutputPath     string `json:"outputPath"`
 	ModelPath      string `json:"modelPath"`
 	Model          string `json:"model"`
@@ -121,6 +125,33 @@ type Input struct {
 }
 
 func Upscayl(input Input) (string, error) {
+	if input.ImagePath == "" && input.ImageURL == "" {
+		return "", errors.New("input path or url not set")
+	}
+	if input.ImagePath != "" && input.ImageURL != "" {
+		return "", errors.New("only 1 of the input allowed")
+	}
+	if input.ImageURL != "" {
+		parsedURL, err := url.Parse(input.ImageURL)
+		if err != nil {
+			return "", err
+		}
+		ext := path.Ext(parsedURL.Path)
+		if ext != ".png" && ext != ".jpg" && ext != ".jpeg" && ext != ".webp" {
+			return "", errors.New("invalid extension in url expected .png, .jpg, .jpeg, .webp")
+		}
+		tmpFilePath := filepath.Join(rootDir, "input"+ext)
+		err = download(input.ImageURL, tmpFilePath)
+		if err != nil {
+			return "", err
+		}
+		defer os.Remove(tmpFilePath)
+		input.ImagePath = tmpFilePath
+	}
+	return upscaylImage(input)
+}
+
+func upscaylImage(input Input) (string, error) {
 	args := make([]string, 0)
 	args = append(args, fmt.Sprintf("-i %s", input.ImagePath))
 	if input.OutputPath == "" {