#include "openai.h"
#include "markdown.h"
#include "plugin.h"
#include "line.h"
#include <locale.h>
#include <stdio.h>
char * get_prompt_from_args(int c, char **argv){
char * prompt = malloc(1024*1024 + 1);
prompt[0] = 0;
for(int i = 1; i < c; i++){
if(argv[i][0] == '-')
break;
strncat(prompt, argv[i], 1024*1024);
if(i < c - 1){
strncat(prompt, " ", 1024*1024);
}else{
strncat(prompt, ".", 1024*1024);
}
}
if(!*prompt){
free(prompt);
return NULL;
}
return prompt;
}
bool try_prompt(int argc,char*argv[]){
char * prompt = get_prompt_from_args(argc, argv);
if(prompt != NULL){
char * response = openai_chat("user",prompt);
parse_markdown_to_ansi(response);
printf("\n");
free(response);
free(prompt);
return true;
}
return false;
}
void help();
void render(char *);
void serve(){
render("Starting server. *Put executables in a dir named cgi-bin and they will behave as webpages.*");
int res = system("python3 -m http.server --cgi");
// Thanks tsoding!
(void)res;
}
void render(char * content){
parse_markdown_to_ansi(content);
printf("\n\n");
}
void repl(){
line_init();
setbuf(stdout, NULL);
char *line;
char *previous_line = NULL;
while((line = line_read("> "))){
if(!line || !*line){
line = previous_line;
}
if(!line || !*line)
continue;
previous_line = line;
if(line[0] == '!'){
plugin_run(line + 1);
continue;
}
if(!strncmp(line,"exit", 4)){
exit(0);
}
if(!strncmp(line,"help",4)){
help();
continue;
}
if(!strncmp(line,"serve",5)){
serve();
}
if(!strncmp(line,"spar ",5)){
char * response = line+5;
while(true){
render(response);
sleep(2);
//line = line_read("> ");
//if(!*line)
response = openai_chat("user",response);
}
}
if(!strncmp(line,"ls",2) || !strncmp(line,"list",4)){
int offset = 2;
if(!strncmp(line,"list",4)){
offset = 4;
}
char * command = (char *)malloc(strlen(line) + 42);
command[0] = 0;
strcpy(command, "ls ");
strcat(command, line + offset);
int res = system(command);
(void)res;
free(command);
continue;
}
line_add_history(line);
char * response = openai_chat("user", line);
render(response);
free(response);
}
}
void help(){
char help_text[1024*1024] = {0};
char * template = "# Help\n"
"Written by retoor@molodetz.nl.\n\n"
"## Features\n"
" - navigate trough history using `arrows`.\n"
" - navigate trough history with **recursive search** using `ctrl+r`.\n"
" - **inception with python** for *incomming* and *outgoing* content.\n"
" - markdown and **syntax highlighting**.\n"
" - **execute python commands** with prefixing `!`\n"
" - list files of current workdirectory using `ls`.\n"
" - type `serve` to start a webserver with directory listing. Easy for network transfers.\n\n"
"## Configuration\n"
" - model temperature is %f.\n"
" - model name is %s.\n"
" - max tokens is %d.\n\n"
"## In development\n"
" - **google search** and actions with those results.\n"
" - **reminders**.\n"
" - predefined **templates** for **reviewing** / **refactoring** so you can personalize.\n";
sprintf(help_text,template,prompt_temperature,prompt_model,prompt_max_tokens);
render(help_text);
}
void openai_include(char * path){
FILE * file = fopen(path,"r");
if(file == NULL){
return;
}
fseek(file, 0, SEEK_END);
long size = ftell(file);
fseek(file, 0, SEEK_SET);
char * buffer = (char *)malloc(size);
size_t read = fread(buffer,1,size,file);
if(read == 0){
return;
}
fclose(file);
buffer[read] = 0;
openai_system(buffer);
free(buffer);
}
void init(){
line_init();
const char *locale = setlocale(LC_ALL, NULL);
char payload[4096] = {0};
sprintf(payload, "User locale is %s. User lang is %s.\n"
"You are Retoor. Use a lot of markdown in response.\n"
"Be confident and short in answers.\n"
"You divide things by zero if you have to."
, locale, locale);
printf("%s","Loading...");
openai_system(payload);
openai_include("context.txt");
printf("%s", "\rLoaded! Type help for feautures.\n");
}
int main(int argc, char *argv[]){
init();
if(try_prompt(argc,argv))
return 0;
repl();
return 0;
}