Automatically identify whether the AI prompt content is a string or a code

[ddkwork]

Describe the feature

In addition, we need to use MD rendering and highlight syntax when the input prompt content is a block of code, and we need a scanning algorithm to automatically identify which programming language it is in order to set the highlight color

Relevant code

No response

[kkoreilly]

Yes, I will implement good MD code highlighting and rendering soon.

[ddkwork]

package main

import (
	"fmt"
	"regexp"
)

func extractCodeBlocks(text string) {
	patterns := map[string]string{
		"Go":     `\bfunc\b|\bpackage\b|\bimport\b`,
		"C++":    `\bclass\b|\bint\b|\bvoid\b`,
		"Python": `\bdef\b|\bimport\b|\bfrom\b`,
		"Rust":   `\bfn\b|\buse\b|\bmod\b`,
	}

	for lang, pattern := range patterns {
		re := regexp.MustCompile(pattern)
		matches := re.FindAllStringIndex(text, -1)
		if len(matches) > 0 {
			fmt.Printf("%s code blocks found:\n", lang)
			for _, match := range matches {
				fmt.Printf("Start line: %d, End line: %d\n", getLineNumber(text, match[0]), getLineNumber(text, match[1]))
			}
		}
	}
}

func getLineNumber(text string, index int) int {
	lines := 1
	for i := 0; i < index; i++ {
		if text[i] == '\n' {
			lines++
		}
	}
	return lines
}

func main() {
	text := `
        package main

        import (
            "fmt"
        )

        func main() {
            fmt.Println("Hello, world!")
        }

        // C++ code block
        class MyClass {
            int val;
        };

        # Python code block
        def hello():
            print("Hello, world!")

        // Rust code block
        pub fn hello() {
            println!("Hello, world!");
        }
    `
	extractCodeBlocks(text)
}

[kkoreilly]

Just to clarify, the issue is not that we need to magically detect the language of the code blocks. It tells us the language of the code blocks, and I just need to connect our markdown renderer with our syntax highlighter to achieve the appropriate highlighting.