From d32e4000f08e71194f2176b33c13768e71ac3d20 Mon Sep 17 00:00:00 2001
From: Baivab Mukhopadhyay <101830709+itsBaivab@users.noreply.github.com>
Date: Wed, 24 Jul 2024 23:32:56 +0530
Subject: [PATCH] feat: Improve the response format

---
 cmd/searchCmd.go | 33 ++++++++++++++++++++++++++++-----
 1 file changed, 28 insertions(+), 5 deletions(-)

diff --git a/cmd/searchCmd.go b/cmd/searchCmd.go
index 1aef8d9..2b5de92 100644
--- a/cmd/searchCmd.go
+++ b/cmd/searchCmd.go
@@ -5,6 +5,7 @@ import (
 	"fmt"
 	"log"
 	"os"
+	"regexp"
 	"strconv"
 	"strings"
 
@@ -20,19 +21,17 @@ var searchCmd = &cobra.Command{
 	Short: "Ask a question and get a response",
 	Args:  cobra.MinimumNArgs(1),
 	Run: func(cmd *cobra.Command, args []string) {
-		res := getApiRespone(args)
+		res := getApiResponse(args)
 		fmt.Println(res)
 	},
 }
 
-func getApiRespone(args []string) string {
-
+func getApiResponse(args []string) string {
 	userArgs := strings.Join(args[0:], " ")
 
 	ctx := context.Background()
 	client, err := genai.NewClient(ctx, option.WithAPIKey(os.Getenv("GEMINI_API_KEY")))
 	checkNilError(err)
-
 	defer client.Close()
 
 	// Validate user input is a number
@@ -47,7 +46,31 @@ func getApiRespone(args []string) string {
 
 	finalResponse := resp.Candidates[0].Content.Parts[0]
 
-	return fmt.Sprint(finalResponse)
+	return formatAsPlainText(fmt.Sprint(finalResponse))
+}
+
+func formatAsPlainText(input string) string {
+	// Remove Markdown headings
+	re := regexp.MustCompile(`^#{1,6}\s+(.*)`)
+	input = re.ReplaceAllString(input, "$1")
+
+	// Remove horizontal rules
+	re = regexp.MustCompile(`\n---\n`)
+	input = re.ReplaceAllString(input, "\n")
+
+	// Remove italic formatting
+	re = regexp.MustCompile(`_([^_]+)_`)
+	input = re.ReplaceAllString(input, "$1")
+
+	// Remove bold formatting
+	re = regexp.MustCompile(`\*\*([^*]+)\*\*`)
+	input = re.ReplaceAllString(input, "$1")
+
+	// Remove bullet points
+	re = regexp.MustCompile(`\n\* (.*)`)
+	input = re.ReplaceAllString(input, "\n- $1")
+
+	return input
 }
 
 func init() {