 1package main
 2
 3import (
 4    "bytes"
 5    "encoding/json"
 6    "fmt"
 7    "io"
 8    "net/http"
 9    "time"
10)
11
12func pollForResult(pollURL, apiKey string) (interface{}, error) {
13    for {
14        req, err := http.NewRequest("GET", pollURL, nil)
15        if err != nil {
16            return nil, err
17        }
18        req.Header.Set("Authorization", "Bearer "+apiKey)
19
20        client := &http.Client{}
21        resp, err := client.Do(req)
22        if err != nil {
23            return nil, err
24        }
25        defer resp.Body.Close()
26
27        body, err := io.ReadAll(resp.Body)
28        if err != nil {
29            return nil, err
30        }
31
32        var result map[string]interface{}
33        json.Unmarshal(body, &result)
34
35        status := result["status"].(string)
36        if status == "COMPLETED" {
37            var outputs interface{}
38            json.Unmarshal([]byte(result["output"].(string)), &outputs)
39            return outputs, nil
40        } else if status == "FAILED" {
41            return nil, fmt.Errorf("generation failed")
42        }
43
44        time.Sleep(7 * time.Second)
45    }
46}
47
48func main() {
49    url := "https://api.segmind.com/workflows/66bb3d7d364d0f944f366acc-v1"
50    apiKey := "YOUR_API_KEY"
51
52    data := []byte(`{
53  "news_script": "Write a concise and informative news lead that effectively summarizes the most important aspects of a breaking story from the blog link given,Don't mention any unwanted stuff or your an AI or you're writing a summary just be very precise and present the story like a script, no unwanted details: https://www.msn.com/en-in/travel/airtravel/qatar-airways-launches-world-s-first-ai-powered-flight-attendant-sama-2-0/vi-BB1kKZ4U?ocid=msedgntp&pc=ASTS&cvid=2d6ec7118efc4602b1b94c7379075cfc&ei=99",
54  "text_prompt_newsanchor": "a head shot of a female news reporter against a white background, realistic"
55}`)
56
57    req, err := http.NewRequest("POST", url, bytes.NewBuffer(data))
58    if err != nil {
59        panic(err)
60    }
61
62    req.Header.Set("Authorization", "Bearer "+apiKey)
63    req.Header.Set("Content-Type", "application/json")
64
65    client := &http.Client{}
66    resp, err := client.Do(req)
67    if err != nil {
68        panic(err)
69    }
70    defer resp.Body.Close()
71
72    body, err := io.ReadAll(resp.Body)
73    if err != nil {
74        panic(err)
75    }
76
77    var result map[string]interface{}
78    json.Unmarshal(body, &result)
79
80    fmt.Println("Request queued:", result)
81
82    pollURL := result["poll_url"].(string)
83    outputs, err := pollForResult(pollURL, apiKey)
84    if err != nil {
85        panic(err)
86    }
87
88    fmt.Println("Generation complete:", outputs)
89}