Add OpenAI-powered content parsing improvement feature

Author: paskal

backend/extractor/readability.go

@@ -178,6 +178,114 @@ func (f *UReadability) extractWithRules(ctx context.Context, reqURL string, rule
 	return rb, nil
 }
 
+// ContentParsedWrong handles the logic for when content is parsed incorrectly
+func (f *UReadability) ContentParsedWrong(ctx context.Context, urlStr string) (string, error) {
+	// Extract content using the current method
+	originalContent, err := f.Extract(ctx, urlStr)
+	if err != nil {
+		return "", fmt.Errorf("failed to extract content: %v", err)
+	}
+
+	// Get CSS selector from ChatGPT
+	selector, err := f.getChatGPTSelector(ctx, urlStr)
+	if err != nil {
+		return "", fmt.Errorf("failed to get CSS selector: %v", err)
+	}
+
+	// Get the HTML body
+	body, err := f.getHTMLBody(urlStr)
+	if err != nil {
+		return "", fmt.Errorf("failed to get HTML body: %v", err)
+	}
+
+	// Extract content using the new selector
+	newContent, err := f.extractContentWithSelector(body, selector)
+	if err != nil {
+		return "", fmt.Errorf("failed to extract content with new selector: %v", err)
+	}
+
+	// Compare original and new content
+	if strings.TrimSpace(originalContent.Content) != strings.TrimSpace(newContent) {
+		// Contents are different, create a new rule
+		rule := datastore.Rule{
+			Author:   "",
+			Domain:   f.extractDomain(urlStr),
+			Content:  selector,
+			TestURLs: []string{urlStr},
+			Enabled:  true,
+		}
+
+		_, err = f.Rules.Save(ctx, rule)
+		if err != nil {
+			return "", fmt.Errorf("failed to save new rule: %v", err)
+		}
+
+		return fmt.Sprintf("new custom rule with DOM %s created", selector), nil
+	}
+
+	return "default rule is good, no need to create the custom one", nil
+}
+
+func (f *UReadability) getChatGPTSelector(ctx context.Context, urlStr string) (string, error) {
+	client := openai.NewClient(f.OpenAIKey)
+	resp, err := client.CreateChatCompletion(
+		ctx,
+		openai.ChatCompletionRequest{
+			Model: openai.GPT4o,
+			Messages: []openai.ChatCompletionMessage{
+				{
+					Role:    openai.ChatMessageRoleSystem,
+					Content: "You are a helpful assistant that provides CSS selectors for extracting main content from web pages.",
+				},
+				{
+					Role:    openai.ChatMessageRoleUser,
+					Content: fmt.Sprintf("Given the URL %s, identify the CSS selector that can be used to extract the main content of the article. This typically includes elements like 'article', 'main', or specific classes. Return only this selector and nothing else.", urlStr),
+				},
+			},
+		},
+	)
+
+	if err != nil {
+		return "", err
+	}
+
+	return resp.Choices[0].Message.Content, nil
+}
+
+func (f *UReadability) getHTMLBody(urlStr string) (string, error) {
+	//nolint:gosec
+	resp, err := http.Get(urlStr)
+	if err != nil {
+		return "", err
+	}
+	defer resp.Body.Close()
+
+	body, err := io.ReadAll(resp.Body)
+	if err != nil {
+		return "", err
+	}
+
+	return string(body), nil
+}
+
+func (f *UReadability) extractContentWithSelector(body, selector string) (string, error) {
+	doc, err := goquery.NewDocumentFromReader(strings.NewReader(body))
+	if err != nil {
+		return "", err
+	}
+
+	content := doc.Find(selector).Text()
+	return content, nil
+}
+
+func (f *UReadability) extractDomain(urlStr string) string {
+	u, err := url.Parse(urlStr)
+	if err != nil {
+		return ""
+	}
+	return u.Hostname()
+}
+
 // getContent retrieves content from raw body string, both content (text only) and rich (with html tags)
 // if rule is provided, it uses custom rule, otherwise tries to retrieve one from the storage,
 // and at last tries to use general readability parser

backend/rest/server.go

@@ -82,6 +82,7 @@ func (s *Server) routes(frontendDir string) chi.Router {
 		r.Get("/content/v1/parser", s.extractArticleEmulateReadability)
 		r.Post("/extract", s.extractArticle)
 		r.Post("/auth", s.authFake)
+		r.Get("/content-parsed-wrong", s.contentParsedWrong)
 
 		r.Group(func(protected chi.Router) {
 			protected.Use(basicAuth("ureadability", s.Credentials))
@@ -391,6 +392,30 @@ func (s *Server) authFake(w http.ResponseWriter, r *http.Request) {
 	render.JSON(w, r, JSON{"pong": t.Format("20060102150405")})
 }
 
+func (s *Server) contentParsedWrong(w http.ResponseWriter, r *http.Request) {
+	if s.Readability.OpenAIKey == "" {
+		render.Status(r, http.StatusBadRequest)
+		render.JSON(w, r, JSON{"error": "OpenAI key is not set"})
+		return
+	}
+
+	exampleURL := r.URL.Query().Get("url")
+	if exampleURL == "" {
+		render.Status(r, http.StatusBadRequest)
+		render.JSON(w, r, JSON{"error": "url parameter is required"})
+		return
+	}
+
+	message, err := s.Readability.ContentParsedWrong(r.Context(), exampleURL)
+	if err != nil {
+		render.Status(r, http.StatusInternalServerError)
+		render.JSON(w, r, JSON{"error": err.Error()})
+		return
+	}
+
+	render.JSON(w, r, JSON{"message": message})
+}
+
 func getBid(id string) primitive.ObjectID {
 	bid, err := primitive.ObjectIDFromHex(id)
 	if err != nil {