Skip to content
Snippets Groups Projects
Commit 9ed140f4 authored by Sivert2101's avatar Sivert2101
Browse files

Made a new main file that runs automatic, and added a prompt list

parent 2ecca693
No related branches found
No related tags found
No related merge requests found
package main
import (
"fmt"
"io"
"llama/compiler"
displayindicator "llama/display-indicator"
"llama/extraction"
ollamaimplementation "llama/ollama-implementation"
"os"
"strings"
)
func main_automatic() {
//reader := bufio.NewReader(os.Stdin)
var conversationContext []int // Variable to store conversation context
//fmt.Print("Enter your prompt (or type 'exit' to quit): ")
//userPrompt, _ := reader.ReadString('\n')
//userPrompt = strings.TrimSpace(userPrompt)
promptText, err := os.Open("promptList.txt")
if err != nil {
fmt.Println("Error opening file", err)
}
text, err := io.ReadAll(promptText)
if err != nil {
fmt.Println("Error reading file:", err)
return
}
prompt := string(text)
promptList := strings.Split(prompt, "\n")
var userPrompt = ""
for i := range promptList {
fmt.Println(promptList[i])
userPrompt = promptList[i]
}
for {
//if userPrompt == "exit" {
// fmt.Println("Exiting the program.")
// break
//}
var modifiedPrompt = userPrompt + extraction.GoPrompt
fmt.Println("Prompt received. Generating response...")
// Start a go routine to display a waiting indicator while the response is being generated
done := make(chan bool)
go displayindicator.DisplayLoadingIndicator(done)
// Generate response using Ollama API, passing the context
response, updatedContext, err := ollamaimplementation.GetOllamaResponse(modifiedPrompt, conversationContext)
// Signal the waiting indicator to stop
done <- true
if err != nil {
fmt.Println("Error generating response:", err)
continue
}
// Update the conversation context with the response
//conversationContext = updatedContext
fmt.Println(updatedContext)
generatedCode, _ := extraction.Extract(response) // Handle error with string
fmt.Println("Ollama's response:", generatedCode)
output, err := compiler.CompileStringToGo(generatedCode)
if err != nil {
userPrompt = output + "\nFollowing are the errors, please fix the code. Write it again, and write only source code along with same test cases with no further explanation. The format should be ```rust <yourcode + testcases> ```"
} else {
fmt.Println("Compiled successfully. Here is the output: %v", output)
userPrompt = "exit"
}
}
}
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment